2 files changed, 174 insertions, 0 deletions
diff --git a/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/DBRequestStore.java b/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/DBRequestStore.java
new file mode 100644
index 00000000..a1b88270
--- /dev/null
+++ b/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/DBRequestStore.java
@@ -0,0 +1,122 @@
+package at.gv.egiz.pdfas.web.store;
+
+import java.text.SimpleDateFormat;
+import java.util.Calendar;
+import java.util.Date;
+
+import org.hibernate.Query;
+import org.hibernate.Session;
+import org.hibernate.SessionFactory;
+import org.hibernate.Transaction;
+import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
+import org.hibernate.cfg.Configuration;
+import org.hibernate.service.ServiceRegistry;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import at.gv.egiz.pdfas.api.ws.PDFASSignRequest;
+import at.gv.egiz.pdfas.web.config.WebConfiguration;
+import at.gv.egiz.pdfas.web.store.IRequestStore;
+import at.gv.egiz.pdfas.web.store.db.Request;
+
+public class DBRequestStore implements IRequestStore {
+
+	private static final Logger logger = LoggerFactory
+			.getLogger(DBRequestStore.class);
+	
+	private SessionFactory sessions;
+	private ServiceRegistry serviceRegistry;
+	
+	public DBRequestStore() {
+		Configuration cfg = new Configuration();
+		cfg.addAnnotatedClass(Request.class);
+		cfg.setProperties(WebConfiguration.getHibernateProps());
+		
+		serviceRegistry = new StandardServiceRegistryBuilder().applySettings(
+				cfg.getProperties()).build();
+		
+		sessions = cfg.buildSessionFactory(serviceRegistry);
+	}
+	
+	private void cleanOldRequests() {
+		int seconds = WebConfiguration.getDBTimeout();
+		Calendar calendar = Calendar.getInstance();
+		calendar.add(Calendar.SECOND, (-1)* seconds);
+		Date date = calendar.getTime();
+		SimpleDateFormat dt = new SimpleDateFormat("yyyy-mm-dd hh:mm:ss"); 
+		logger.info("Clearing Entries before: " + dt.format(date));
+		Session session = null;
+		Transaction tx = null;
+		try {
+			session = sessions.openSession();
+			tx = session.beginTransaction();
+			Query query = session.createQuery("delete from Request as req" + 
+				" where req.created < :date");
+			query.setCalendar("date", calendar);
+			query.executeUpdate();
+		} catch(Throwable e) {
+			logger.error("Failed to save Request", e);
+			tx.rollback();
+		} finally {
+			if(session != null) {
+				session.close();
+			}
+		}
+	}
+	
+	public String createNewStoreEntry(PDFASSignRequest request) {
+		// Clean Old Requests
+		this.cleanOldRequests();
+		Session session = null;
+		Transaction tx = null;
+		try {
+			session = sessions.openSession();
+			tx = session.beginTransaction();
+			Request dbRequest = new Request();
+			dbRequest.setSignRequest(request);
+			dbRequest.setCreated(Calendar.getInstance().getTime());
+			session.save(dbRequest);
+		
+			tx.commit();
+			return dbRequest.getId();
+		} catch(Throwable e) {
+			logger.error("Failed to save Request", e);
+			tx.rollback();
+			return null;
+		} finally {
+			if(session != null) {
+				session.close();
+			}
+		}
+	}
+
+	public PDFASSignRequest fetchStoreEntry(String id) {
+		// Clean Old Requests
+		this.cleanOldRequests();
+	
+		Session session = null;
+		Transaction tx = null;
+		try {
+			session = sessions.openSession();
+			tx = session.beginTransaction();
+			Request dbRequest = (Request) session.get(Request.class, id);
+			
+			PDFASSignRequest request = dbRequest.getSignRequest();
+			
+			session.delete(dbRequest);
+		
+			tx.commit();
+			return request;
+		} catch(Throwable e) {
+			logger.error("Failed to fetch Request", e);
+			tx.rollback();
+			return null;
+		} finally {
+			if(session != null) {
+				session.close();
+			}
+		}
+		
+	}
+
+}
diff --git a/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/db/Request.java b/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/db/Request.java
new file mode 100644
index 00000000..d7377166
--- /dev/null
+++ b/pdf-as-web-db/src/main/java/at/gv/egiz/pdfas/web/store/db/Request.java
@@ -0,0 +1,52 @@
+package at.gv.egiz.pdfas.web.store.db;
+
+import java.util.Date;
+
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.Id;
+import javax.persistence.Table;
+
+import org.hibernate.annotations.GenericGenerator;
+
+import at.gv.egiz.pdfas.api.ws.PDFASSignRequest;
+
+@Entity
+@Table(name = "requests")
+public class Request {
+
+	private String uuid;	
+	private Date created;
+	private PDFASSignRequest signRequest;
+	
+	@Id
+	@GeneratedValue(generator = "uuid")
+	@GenericGenerator(name = "uuid", strategy = "uuid2")
+	@Column(name = "id", unique = true)
+	public String getId() {
+		return this.uuid;
+	}
+
+	public void setId(String uuid) {
+		this.uuid = uuid;
+	}
+	
+	@Column(name = "created", nullable = false)
+	public Date getCreated() {
+		return this.created;
+	}
+
+	public void setCreated(Date created) {
+		this.created = created;
+	}
+	
+	@Column(name = "signRequest", nullable = false, length = 52428800)
+	public PDFASSignRequest getSignRequest() {
+		return this.signRequest;
+	}
+
+	public void setSignRequest(PDFASSignRequest signRequest) {
+		this.signRequest = signRequest;
+	}
+}