ingo@2850: package de.intevation.flys.importer;
ingo@2850: 
ingo@2850: import java.sql.SQLException;
ingo@2850: import java.util.ArrayList;
ingo@2850: import java.util.List;
ingo@2850: 
ingo@2850: import org.apache.log4j.Logger;
ingo@2850: 
ingo@2850: import org.hibernate.Session;
ingo@2850: import org.hibernate.Query;
ingo@2850: import org.hibernate.exception.ConstraintViolationException;
ingo@2850: 
ingo@2850: import de.intevation.flys.model.River;
ingo@2850: import de.intevation.flys.model.Unit;
ingo@2850: import de.intevation.flys.model.WaterlevelDifference;
ingo@2850: 
ingo@2850: 
ingo@2850: public class ImportWaterlevelDifference {
ingo@2850: 
ingo@3943:     private static final Logger log = Logger
ingo@3943:         .getLogger(ImportWaterlevelDifference.class);
ingo@2850: 
ingo@2850:     private ImportUnit unit;
ingo@2850: 
ingo@2850:     private String description;
ingo@2850: 
ingo@2850:     private List<ImportWaterlevelDifferenceColumn> columns;
ingo@2850: 
ingo@2850:     private WaterlevelDifference peer;
ingo@2850: 
ingo@2850:     public ImportWaterlevelDifference(String description) {
ingo@2850:         this.columns = new ArrayList<ImportWaterlevelDifferenceColumn>();
ingo@2850: 
ingo@2850:         this.description = description;
ingo@2850:     }
ingo@2850: 
ingo@2850:     public void setUnit(ImportUnit unit) {
ingo@2850:         this.unit = unit;
ingo@2850:     }
ingo@2850: 
ingo@2850:     public void addValue(ImportWaterlevelDifferenceColumn column) {
ingo@2850:         this.columns.add(column);
ingo@2850:     }
ingo@2850: 
ingo@3943:     public void storeDependencies(River river) throws SQLException,
ingo@3943:         ConstraintViolationException {
ingo@2850:         log.info("store dependencies");
ingo@2850: 
ingo@2850:         WaterlevelDifference peer = getPeer(river);
ingo@2850: 
ingo@3943:         if (peer != null) {
ingo@3943:             int i = 0;
ingo@2850: 
ingo@3943:             for (ImportWaterlevelDifferenceColumn column : columns) {
ingo@3943:                 column.storeDependencies(peer);
ingo@3943:                 i++;
ingo@3943:             }
ingo@3943: 
ingo@3943:             log.info("stored " + i + " waterlevel difference columns");
ingo@2850:         }
ingo@2850:     }
ingo@2850: 
ingo@2850:     public WaterlevelDifference getPeer(River river) {
ingo@2850:         Unit u = unit != null ? unit.getPeer() : null;
ingo@2850:         if (u == null) {
sascha@3662:             log.warn("IWD: skip invalid waterlevel difference - no unit set!");
ingo@2850:             return null;
ingo@2850:         }
ingo@2850: 
ingo@2850:         if (peer == null) {
ingo@3943:             Session session = ImporterSession.getInstance()
ingo@3943:                 .getDatabaseSession();
ingo@3943:             Query query = session
ingo@3943:                 .createQuery("from WaterlevelDifference where "
ingo@3943:                     + "   river=:river and " + "   unit=:unit and "
ingo@3943:                     + "   description=:description");
ingo@2850: 
ingo@2850:             query.setParameter("river", river);
ingo@2850:             query.setParameter("unit", u);
ingo@2850:             query.setParameter("description", description);
ingo@2850: 
ingo@2850:             List<WaterlevelDifference> diffs = query.list();
ingo@2850:             if (diffs.isEmpty()) {
ingo@2850:                 peer = new WaterlevelDifference(river, u, description);
ingo@2850:                 session.save(peer);
ingo@2850:             }
ingo@2850:             else {
ingo@2850:                 peer = diffs.get(0);
ingo@2850:             }
ingo@2850:         }
ingo@2850: 
ingo@2850:         return peer;
ingo@2850:     }
ingo@2850: }
ingo@2850: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :