sascha@13: package de.intevation.artifactdatabase;
sascha@13: 
sascha@14: import java.util.UUID;
sascha@14: 
sascha@14: import java.sql.Connection;
sascha@14: import java.sql.SQLException;
sascha@14: import java.sql.PreparedStatement;
sascha@14: import java.sql.Types;
sascha@14: import java.sql.ResultSet;
sascha@14: 
sascha@14: import javax.sql.DataSource;
sascha@14: 
sascha@14: import java.io.IOException;
sascha@15: import java.io.ByteArrayInputStream;
sascha@14: import java.io.ByteArrayOutputStream;
sascha@14: import java.io.ObjectOutputStream;
sascha@15: import java.io.ObjectInputStream;
sascha@14: 
sascha@14: import java.util.zip.GZIPOutputStream;
sascha@15: import java.util.zip.GZIPInputStream;
sascha@14: 
sascha@13: import de.intevation.artifacts.Artifact;
sascha@13: 
sascha@17: import org.apache.log4j.Logger;
sascha@17: 
sascha@13: /**
sascha@13:  *  @author Sascha L. Teichmann
sascha@13:  */
sascha@13: public class Backend
sascha@13: {
sascha@17:     private static Logger logger = Logger.getLogger(Backend.class);
sascha@17: 
sascha@14:     public static final String SQL_NEXT_ID =
sascha@14:         SQL.get("artifacts.id.nextval");
sascha@14: 
sascha@14:     public static final String SQL_INSERT =
sascha@14:         SQL.get("artifacts.insert");
sascha@14: 
sascha@14:     public static final String SQL_UPDATE =
sascha@14:         SQL.get("artifacts.update");
sascha@14: 
sascha@14:     public static final String SQL_TOUCH =
sascha@14:         SQL.get("artifacts.touch");
sascha@14: 
sascha@15:     public static final String SQL_LOAD_BY_GID =
sascha@15:         SQL.get("artifacts.select.gid");
sascha@15: 
sascha@30:     protected DatabaseCleaner cleaner;
sascha@30: 
sascha@32:     public final class PersistentArtifact
sascha@32:     extends            Id
sascha@14:     {
sascha@32:         private Artifact artifact;
sascha@14: 
sascha@32:         public PersistentArtifact(Artifact artifact, int id) {
sascha@32:             super(id);
sascha@32:             this.artifact = artifact;
sascha@14:         }
sascha@14: 
sascha@32:         public Artifact getArtifact() {
sascha@32:             return artifact;
sascha@14:         }
sascha@14: 
sascha@32:         public void store() {
sascha@38:             if (logger.isDebugEnabled()) {
sascha@38:                 logger.debug("storing artifact id = " + getId());
sascha@38:             }
sascha@32:             Backend.this.store(this);
sascha@14:         }
sascha@14: 
sascha@32:         public void touch() {
sascha@38:             if (logger.isDebugEnabled()) {
sascha@38:                 logger.debug("touching artifact id = " + getId());
sascha@38:             }
sascha@32:             Backend.this.touch(this);
sascha@14:         }
sascha@32:     } // class ArtifactWithId
sascha@14: 
sascha@13:     public Backend() {
sascha@13:     }
sascha@13: 
sascha@30:     public Backend(DatabaseCleaner cleaner) {
sascha@30:         this.cleaner = cleaner;
sascha@30:     }
sascha@30: 
sascha@32:     public void setCleaner(DatabaseCleaner cleaner) {
sascha@32:         this.cleaner = cleaner;
sascha@32:     }
sascha@32: 
sascha@32:     public String newIdentifier() {
sascha@32:         UUID uuid = UUID.randomUUID();
sascha@32:         // TODO: check database for collisions.
sascha@32:         return uuid.toString();
sascha@32:     }
sascha@32: 
sascha@32:     public PersistentArtifact storeInitially(
sascha@32:         Artifact artifact,
sascha@32:         Long     ttl
sascha@32:     ) 
sascha@32:     throws Exception
sascha@32:     {
sascha@32:         return new PersistentArtifact(
sascha@32:             artifact,
sascha@32:             insertDatabase(artifact, ttl));
sascha@32:     }
sascha@32:     
sascha@32:     public PersistentArtifact getArtifact(String identifer) {
sascha@15: 
sascha@15:         try {
sascha@32:             UUID.fromString(identifer);
sascha@15:         }
sascha@15:         catch (IllegalArgumentException iae) {
sascha@17:             logger.warn(iae.getLocalizedMessage());
sascha@15:             return null;
sascha@15:         }
sascha@15: 
sascha@15:         Connection        connection  = null;
sascha@15:         PreparedStatement stmnt_load  = null;
sascha@15:         ResultSet         load_result = null;
sascha@15: 
sascha@15:         DataSource dataSource = DBConnection.getDataSource();
sascha@15:         try {
sascha@15:             connection = dataSource.getConnection();
sascha@15:             stmnt_load = connection.prepareStatement(SQL_LOAD_BY_GID);
sascha@32:             stmnt_load.setString(1, identifer);
sascha@15: 
sascha@15:             load_result = stmnt_load.executeQuery();
sascha@15: 
sascha@15:             if (!load_result.next()) {
sascha@15:                 return null;
sascha@15:             }
sascha@15: 
sascha@15:             int  id  = load_result.getInt(1);
sascha@15:             long ttl = load_result.getLong(3);
sascha@15: 
sascha@15:             if (!load_result.wasNull()) { // real time to life
sascha@15:                 long last_access = load_result.getTimestamp(2).getTime();
tim@34:                 if (last_access + ttl < System.currentTimeMillis()) {
sascha@15:                     artifactOutdated(id);
sascha@15:                     return null;
sascha@15:                 }
sascha@15:             }
sascha@15: 
sascha@15:             byte [] bytes = load_result.getBytes(4);
sascha@15: 
sascha@32:             Artifact artifact = restoreArtifact(bytes);
sascha@15: 
sascha@32:             return artifact == null
sascha@32:                 ? null
sascha@32:                 : new PersistentArtifact(artifact, id);
sascha@15:         }
sascha@15:         catch (SQLException sqle) {
sascha@17:             logger.error(sqle.getLocalizedMessage(), sqle);
sascha@15:         }
sascha@15:         finally {
sascha@15:             if (load_result != null) {
sascha@15:                 try { load_result.close(); }
sascha@15:                 catch (SQLException sqle) {}
sascha@15:             }
sascha@15:             if (stmnt_load != null) {
sascha@15:                 try { load_result.close(); }
sascha@15:                 catch (SQLException sqle) {}
sascha@15:             }
sascha@15:             if (connection != null) {
sascha@15:                 try { connection.close(); } 
sascha@15:                 catch (SQLException sqle) {}
sascha@15:             }
sascha@15:         }
sascha@15:         return null;
sascha@15:     }
sascha@15: 
sascha@32:     public static byte [] toBytes(Artifact artifact) {
sascha@32:         try {
sascha@32:             ByteArrayOutputStream bos = new ByteArrayOutputStream();
sascha@32:             GZIPOutputStream      gos = new GZIPOutputStream(bos);
sascha@32:             ObjectOutputStream    oos = new ObjectOutputStream(gos);
sascha@32: 
sascha@32:             oos.writeObject(artifact);
sascha@32:             oos.flush();
sascha@32:             oos.close();
sascha@32: 
sascha@32:             return bos.toByteArray();
sascha@32:         }
sascha@32:         catch (IOException ioe) {
sascha@32:             logger.error(ioe.getLocalizedMessage(), ioe);
sascha@32:             throw new RuntimeException(ioe);
sascha@32:         }
sascha@32:     }
sascha@32: 
sascha@15:     public static Artifact restoreArtifact(byte [] bytes) {
sascha@15: 
sascha@30:         if (bytes == null) {
sascha@30:             return null;
sascha@30:         }
sascha@30: 
sascha@15:         ObjectInputStream ois = null;
sascha@15: 
sascha@15:         try {
sascha@15:             ByteArrayInputStream bis = new ByteArrayInputStream(bytes);
sascha@15:             GZIPInputStream      gis = new GZIPInputStream(bis);
sascha@15:                                  ois = new ObjectInputStream(gis);
sascha@15: 
sascha@15:             return (Artifact)ois.readObject();
sascha@15:         }
sascha@15:         catch (IOException ioe) {
sascha@17:             logger.error(ioe.getLocalizedMessage(), ioe);
sascha@15:         }
sascha@15:         catch (ClassNotFoundException cnfe) {
sascha@17:             logger.error(cnfe.getLocalizedMessage(), cnfe);
sascha@15:         }
sascha@15:         catch (ClassCastException cce) {
sascha@17:             logger.error(cce.getLocalizedMessage(), cce);
sascha@15:         }
sascha@15:         finally {
sascha@15:             if (ois != null) {
sascha@15:                 try { ois.close(); }
sascha@15:                 catch (IOException ioe) { }
sascha@15:             }
sascha@15:         }
sascha@15: 
sascha@15:         return null;
sascha@15:     }
sascha@15: 
sascha@15:     protected void artifactOutdated(int id) {
sascha@17:         logger.info("artifactOutdated: id = " + id);
sascha@30:         if (cleaner != null) {
sascha@30:             cleaner.wakeup();
sascha@30:         }
sascha@15:     }
sascha@15: 
sascha@32:     protected int insertDatabase(Artifact artifact, Long ttl) {
sascha@32: 
sascha@32:         String uuid = artifact.identifier();
sascha@32: 
sascha@14:         Connection        connection    = null;
sascha@14:         PreparedStatement stmnt_next_id = null;
sascha@14:         PreparedStatement stmnt_insert  = null;
sascha@14:         ResultSet         res_id        = null;
sascha@14: 
sascha@14:         DataSource dataSource = DBConnection.getDataSource();
sascha@14:         try {
sascha@14:             connection = dataSource.getConnection();
sascha@14:             try {
sascha@14:                 connection.setAutoCommit(false);
sascha@14: 
sascha@14:                 stmnt_next_id = connection.prepareStatement(SQL_NEXT_ID);
sascha@14:                 stmnt_insert  = connection.prepareStatement(SQL_INSERT);
sascha@14: 
sascha@14:                 res_id = stmnt_next_id.executeQuery();
sascha@14: 
sascha@14:                 if (!res_id.next()) {
sascha@14:                     throw new RuntimeException("No id generated");
sascha@14:                 }
sascha@14: 
sascha@14:                 int id = res_id.getInt(1);
sascha@14: 
sascha@14:                 stmnt_insert.setInt(1, id);
sascha@32:                 stmnt_insert.setString(2, uuid);
sascha@14:                 if (ttl == null) {
sascha@14:                     stmnt_insert.setNull(3, Types.BIGINT);
sascha@14:                 }
sascha@14:                 else {
sascha@14:                     stmnt_insert.setLong(3, ttl.longValue());
sascha@14:                 }
sascha@14: 
sascha@32:                 stmnt_insert.setBytes(4, toBytes(artifact));
sascha@32: 
sascha@14:                 stmnt_insert.execute();
sascha@14: 
sascha@14:                 connection.commit();
sascha@14: 
sascha@14:                 return id;
sascha@14:             }
sascha@14:             catch (SQLException sqle) {
sascha@14:                 connection.rollback();
sascha@14:                 throw sqle;
sascha@14:             }
sascha@14:         }
sascha@14:         catch (SQLException sqle) {
sascha@17:             logger.error(sqle.getLocalizedMessage(), sqle);
sascha@14:         }
sascha@14:         finally {
sascha@14:             if (res_id != null) {
sascha@14:                 try { res_id.close(); }
sascha@14:                 catch (SQLException sqle) {}
sascha@14:             }
sascha@14:             if (stmnt_insert != null) {
sascha@14:                 try { stmnt_insert.close(); }
sascha@14:                 catch (SQLException sqle) {}
sascha@14:             }
sascha@14:             if (stmnt_next_id != null) {
sascha@14:                 try { stmnt_next_id.close(); }
sascha@14:                 catch (SQLException sqle) {}
sascha@14:             }
sascha@14:             if (connection != null) {
sascha@14:                 try { connection.close(); }
sascha@14:                 catch (SQLException sqle) {}
sascha@14:             }
sascha@14:         }
sascha@14:         throw new RuntimeException("failed insert artifact into database");
sascha@14:     }
sascha@14: 
sascha@32:     public void touch(PersistentArtifact artifact) {
sascha@26: 
sascha@14:         try {
sascha@26:             Connection        connection  = null;
sascha@26:             PreparedStatement stmnt_touch = null;
sascha@26:             DataSource        dataSource  = DBConnection.getDataSource();
sascha@14:             try {
sascha@26:                 connection = dataSource.getConnection();
sascha@26:                 try {
sascha@26:                     connection.setAutoCommit(false);
sascha@26:                     stmnt_touch = connection.prepareStatement(SQL_UPDATE);
sascha@32:                     stmnt_touch.setInt(1, artifact.getId());
sascha@26:                     stmnt_touch.execute();
sascha@26:                     connection.commit();
sascha@26:                 }
sascha@26:                 catch (SQLException sqle) {
sascha@26:                     connection.rollback();
sascha@26:                 }
sascha@14:             }
sascha@14:             catch (SQLException sqle) {
sascha@26:                 logger.error(sqle.getLocalizedMessage(), sqle);
sascha@26:             }
sascha@26:             finally {
sascha@26:                 if (stmnt_touch != null) {
sascha@26:                     try { stmnt_touch.close(); }
sascha@26:                     catch (SQLException sqle) {}
sascha@26:                 }
sascha@26:                 if (connection != null) {
sascha@26:                     try { connection.close(); }
sascha@26:                     catch (SQLException sqle) {}
sascha@26:                 }
sascha@14:             }
sascha@14:         }
sascha@26:         catch (Exception e) {
sascha@26:             logger.error(e.getLocalizedMessage(), e);
sascha@14:         }
sascha@14:     }
sascha@14: 
sascha@32:     public void store(PersistentArtifact artifact) {
sascha@26: 
sascha@14:         try {
sascha@26:             Connection        connection   = null;
sascha@26:             PreparedStatement stmnt_update = null;
sascha@26:             DataSource        dataSource   = DBConnection.getDataSource();
sascha@14:             try {
sascha@26:                 connection = dataSource.getConnection();
sascha@26:                 try {
sascha@26:                     connection.setAutoCommit(false);
sascha@26:                     stmnt_update = connection.prepareStatement(SQL_UPDATE);
sascha@32:                     stmnt_update.setInt(2, artifact.getId());
sascha@26: 
sascha@32:                     byte [] bytes = toBytes(artifact.getArtifact());
sascha@26: 
sascha@26:                     stmnt_update.setBytes(1, bytes);
sascha@26:                     stmnt_update.execute();
sascha@26:                     connection.commit();
sascha@26:                 }
sascha@26:                 catch (SQLException sqle) {
sascha@26:                     connection.rollback();
sascha@26:                 }
sascha@14:             }
sascha@14:             catch (SQLException sqle) {
sascha@26:                 logger.error(sqle.getLocalizedMessage(), sqle);
sascha@26:             }
sascha@26:             finally {
sascha@26:                 if (stmnt_update != null) {
sascha@26:                     try { stmnt_update.close(); }
sascha@26:                     catch (SQLException sqle) {}
sascha@26:                 }
sascha@26:                 if (connection != null) {
sascha@26:                     try { connection.close(); }
sascha@26:                     catch (SQLException sqle) {}
sascha@26:                 }
sascha@14:             }
sascha@14:         }
sascha@26:         catch (Exception e) {
sascha@26:             logger.error(e.getLocalizedMessage(), e);
sascha@14:         }
sascha@14:     }
sascha@13: }
sascha@13: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8: