sascha@196: package de.intevation.flys.importer;
sascha@196: 
sascha@196: import de.intevation.flys.model.Wst;
ingo@478: import de.intevation.flys.model.WstColumnValue;
sascha@201: import de.intevation.flys.model.River;
sascha@201: 
ingo@478: import de.intevation.flys.backend.SessionFactoryProvider;
ingo@478: 
ingo@478: import org.apache.log4j.Logger;
ingo@478: 
sascha@201: import org.hibernate.Session;
ingo@478: import org.hibernate.SessionFactory;
sascha@201: import org.hibernate.Query;
sascha@201: 
ingo@478: import java.util.ArrayList;
ingo@478: import java.util.HashMap;
sascha@201: import java.util.List;
sascha@201: import java.util.Map;
sascha@196: 
sascha@196: public class ImportWst
sascha@196: {
ingo@478:     private static Logger logger = Logger.getLogger(ImportWst.class);
ingo@478: 
sascha@201:     protected String description;
sascha@201: 
sascha@467:     protected Integer kind;
sascha@467: 
sascha@474:     //protected Map<String, ImportWstColumn> columns;
sascha@474:     protected List<ImportWstColumn> columns;
sascha@201: 
ingo@478:     protected Map<WstColumnValueKey, WstColumnValue> wstCache;
ingo@478: 
sascha@196:     protected Wst peer;
sascha@196: 
sascha@196:     public ImportWst() {
sascha@467:         kind = 0;
sascha@474:         columns = new ArrayList<ImportWstColumn>();
ingo@478: 
ingo@478:         // This cache is used to avoid db queries for each WstColumnValue later
ingo@478:         wstCache = getWstColumnValueCache();
sascha@196:     }
sascha@196: 
sascha@201:     public ImportWst(String description) {
sascha@201:         this();
sascha@201:         this.description = description;
sascha@201:     }
sascha@201: 
sascha@201:     public String getDescription() {
sascha@201:         return description;
sascha@201:     }
sascha@201: 
sascha@482:     public Integer getKind() {
sascha@482:         return kind;
sascha@482:     }
sascha@482: 
sascha@482:     public void setKind(Integer kind) {
sascha@482:         this.kind = kind;
sascha@482:     }
sascha@482: 
sascha@482: 
sascha@201:     public void setDescription(String description) {
sascha@201:         this.description = description;
sascha@201:     }
sascha@201: 
sascha@474:     public void setNumberColumns(int numColumns) {
sascha@474:         for (int i = 0; i < numColumns; ++i) {
ingo@478:             columns.add(new ImportWstColumn(this, null, null, i, wstCache));
sascha@201:         }
sascha@474:     }
sascha@474: 
sascha@474:     public int getNumberColumns() {
sascha@474:         return columns.size();
sascha@474:     }
sascha@474: 
sascha@474:     public ImportWstColumn getColumn(int index) {
sascha@474:         return columns.get(index);
sascha@201:     }
sascha@201: 
sascha@201:     public void storeDependencies(River river) {
sascha@201:         Wst wst = getPeer(river);
sascha@201: 
sascha@474:         for (ImportWstColumn column: columns) {
sascha@201:             column.storeDependencies(river);
sascha@201:         }
sascha@201:     }
sascha@201: 
ingo@478: 
ingo@478:     public Map<WstColumnValueKey, WstColumnValue> getWstColumnValueCache() {
ingo@478:         SessionFactory sessionFactory =
ingo@478:             SessionFactoryProvider.createSessionFactory();
ingo@478: 
ingo@478:         Session session = sessionFactory.openSession();
ingo@478: 
ingo@478:         Importer.sessionHolder.set(session);
ingo@478: 
ingo@478:         try {
ingo@478:             Query query = session.createQuery("from WstColumnValue");
ingo@478: 
ingo@478:             List<WstColumnValue> values = query.list();
ingo@478:             Map<WstColumnValueKey, WstColumnValue> cache =
ingo@478:                 new HashMap<WstColumnValueKey, WstColumnValue>();
ingo@478: 
ingo@478:             for (WstColumnValue value: values) {
ingo@478:                 cache.put(
ingo@478:                     new WstColumnValueKey(value),
ingo@478:                     value);
ingo@478:             }
ingo@478: 
ingo@478:             logger.info("++++++++++++++++++++++++++++++++++++++++++++++++++++");
ingo@478:             logger.info(cache.size() + " WstColumnValue objects in WST cache.");
ingo@478:             logger.info("++++++++++++++++++++++++++++++++++++++++++++++++++++");
ingo@478: 
ingo@478:             return cache;
ingo@478:         }
ingo@478:         finally {
ingo@478:             session.close();
ingo@478:             Importer.sessionHolder.remove();
ingo@478:         }
ingo@478:     }
ingo@478: 
sascha@201:     public Wst getPeer(River river) {
sascha@196:         if (peer == null) {
sascha@201:             Session session = Importer.sessionHolder.get();
sascha@201:             Query query = session.createQuery(
sascha@201:                 "from Wst where " +
sascha@467:                 "river=:river and description=:description and kind=:kind");
sascha@201:             query.setParameter("river",       river);
sascha@201:             query.setParameter("description", description);
sascha@467:             query.setParameter("kind",        kind);
sascha@201:             List<Wst> wsts = query.list();
sascha@201:             if (wsts.isEmpty()) {
sascha@467:                 peer = new Wst(river, description, kind);
sascha@201:                 session.save(peer);
sascha@201:             }
sascha@201:             else {
sascha@201:                 peer = wsts.get(0);
sascha@201:             }
sascha@201: 
sascha@196:         }
sascha@196:         return peer;
sascha@196:     }
sascha@196: }
sascha@196: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :