ingo@2813: package de.intevation.flys.importer;
ingo@2813: 
ingo@2817: import java.sql.SQLException;
ingo@2817: 
ingo@2813: import java.util.ArrayList;
ingo@2813: import java.util.List;
ingo@2813: 
ingo@2813: import org.apache.log4j.Logger;
ingo@2813: 
ingo@2817: import org.hibernate.Session;
ingo@2817: import org.hibernate.Query;
ingo@2817: import org.hibernate.exception.ConstraintViolationException;
ingo@2817: 
ingo@2813: import de.intevation.flys.model.River;
ingo@2813: import de.intevation.flys.model.SedimentDensity;
ingo@2813: 
ingo@2813: 
ingo@2813: public class ImportSedimentDensity {
ingo@2813: 
ingo@2813:     private static Logger log = Logger.getLogger(ImportSedimentDensity.class);
ingo@2813: 
ingo@2813: 
ingo@2813:     protected SedimentDensity peer;
ingo@2813: 
ingo@2813:     protected ImportDepth depth;
ingo@2813: 
ingo@2816:     protected ImportUnit unit;
ingo@2816: 
ingo@2817:     protected String description;
ingo@2817: 
ingo@2813:     protected List<ImportSedimentDensityValue> values;
ingo@2813: 
ingo@2813: 
ingo@2817:     public ImportSedimentDensity(String description) {
ingo@2817:         this.description = description;
ingo@2817:         this.values      = new ArrayList<ImportSedimentDensityValue>();
ingo@2817:     }
ingo@2817: 
ingo@2817: 
ingo@2817:     public String getDescription() {
ingo@2817:         return description;
ingo@2813:     }
ingo@2813: 
ingo@2813: 
ingo@2813:     public void setDepth(ImportDepth depth) {
ingo@2813:         this.depth = depth;
ingo@2813:     }
ingo@2813: 
ingo@2813: 
ingo@2816:     public void setUnit(ImportUnit unit) {
ingo@2816:         this.unit = unit;
ingo@2816:     }
ingo@2816: 
ingo@2816: 
ingo@2813:     public void addValue(ImportSedimentDensityValue value) {
ingo@2813:         values.add(value);
ingo@2813:     }
ingo@2813: 
ingo@2813: 
ingo@2817:     public void storeDependencies(River river)
ingo@2817:     throws SQLException, ConstraintViolationException
ingo@2817:     {
ingo@2813:         log.info("store dependencies");
ingo@2817: 
ingo@2817:         if (depth != null) {
ingo@2817:             depth.storeDependencies();
ingo@2817:         }
ingo@2817: 
ingo@2817:         log.info("store sediment density values.");
ingo@2817: 
ingo@2817:         SedimentDensity peer = getPeer(river);
ingo@2817: 
ingo@2817:         for (ImportSedimentDensityValue value: values) {
ingo@2817:             value.storeDependencies(peer);
ingo@2817:         }
ingo@2813:     }
ingo@2813: 
ingo@2813: 
ingo@2813:     public SedimentDensity getPeer(River river) {
ingo@2813:         log.info("get peer");
ingo@3940:         
ingo@3940:         if (depth == null) {
ingo@3940:             log.warn("cannot store sediment density '" + description + "': no depth");
ingo@3940:             return null;
ingo@3940:         }
ingo@3940:         
ingo@3940:         if (unit == null) {
ingo@3940:             log.warn("cannot store sediment density '" + description + "': no unit");
ingo@3940:             return null;
ingo@3940:         }
ingo@2813: 
ingo@2817:         if (peer == null) {
ingo@2817:             Session session = ImporterSession.getInstance().getDatabaseSession();
ingo@2817: 
ingo@2817:             Query query = session.createQuery(
ingo@2817:                 "from SedimentDensity where " +
ingo@2817:                 "   river=:river and " +
ingo@2817:                 "   depth=:depth and " +
ingo@2817:                 "   unit=:unit");
ingo@2817: 
ingo@2817:             query.setParameter("river", river);
ingo@2817:             query.setParameter("depth", depth.getPeer());
ingo@2817:             query.setParameter("unit", unit.getPeer());
ingo@2817: 
ingo@2817:             List<SedimentDensity> density = query.list();
ingo@2817: 
ingo@2817:             if (density.isEmpty()) {
ingo@2817:                 log.debug("Create new SedimentDensity DB instance.");
ingo@2817: 
ingo@2817:                 peer = new SedimentDensity(
ingo@2817:                     river,
ingo@2817:                     depth.getPeer(),
ingo@2817:                     unit.getPeer(),
ingo@2817:                     description);
ingo@2817: 
ingo@2817:                 session.save(peer);
ingo@2817:             }
ingo@2817:             else {
ingo@2817:                 peer = density.get(0);
ingo@2817:             }
ingo@2817:         }
ingo@2817: 
ingo@2817:         return peer;
ingo@2813:     }
ingo@2813: }
ingo@2813: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :