cdmlib-persistence/src/main/java/eu/etaxonomy/cdm/persistence/hibernate/PostMergeEntityListener.java

   1 // $Id$
   2 /**
   3  * Copyright (C) 2015 EDIT
   4  * European Distributed Institute of Taxonomy
   5  * http://www.e-taxonomy.eu
   6  *
   7  * The contents of this file are subject to the Mozilla Public License Version 1.1
   8  * See LICENSE.TXT at the top of this package for the full license terms.
   9  */
  10 package eu.etaxonomy.cdm.persistence.hibernate;
  11
  12 import java.util.HashSet;
  13 import java.util.Map;
  14 import java.util.Set;
  15 import java.util.concurrent.ConcurrentHashMap;
  16
  17 import org.hibernate.Hibernate;
  18 import org.hibernate.HibernateException;
  19 import org.hibernate.Session;
  20 import org.hibernate.event.spi.MergeEvent;
  21 import org.hibernate.event.spi.MergeEventListener;
  22
  23 import eu.etaxonomy.cdm.model.common.CdmBase;
  24 import eu.etaxonomy.cdm.model.description.FeatureNode;
  25 import eu.etaxonomy.cdm.model.description.FeatureTree;
  26 import eu.etaxonomy.cdm.model.description.PolytomousKeyNode;
  27 import eu.etaxonomy.cdm.model.taxon.TaxonNode;
  28
  29 /**
  30  * @author cmathew
  31  * @date 23 Sep 2015
  32  *
  33  */
  34 public class PostMergeEntityListener implements MergeEventListener {
  35     private static final long serialVersionUID = 1565797119368313987L;
  36
  37     private static Map<Session, Set<CdmBase>> newEntitiesMap = new ConcurrentHashMap<Session, Set<CdmBase>>();
  38
  39     public static void addSession(Session session) {
  40         newEntitiesMap.put(session, new HashSet<CdmBase>());
  41     }
  42
  43     public static void removeSession(Session session) {
  44         newEntitiesMap.remove(session);
  45     }
  46
  47     public static Set<CdmBase> getNewEntities(Session session) {
  48         return newEntitiesMap.get(session);
  49     }
  50
  51     @Override
  52     public void onMerge(MergeEvent event) throws HibernateException {
  53         Object entity = event.getEntity();
  54
  55     }
  56
  57     @Override
  58     public void onMerge(MergeEvent event, Map copiedAlready) throws HibernateException {
  59         // any new entities are added to a map which is retrieved at the end of the
  60         // CdmEntityDaoBase.merge(T transientObject, boolean returnTransientEntity) call
  61         if(event.getOriginal() != null && CdmBase.class.isAssignableFrom(event.getOriginal().getClass()) &&
  62                 event.getResult() != null && CdmBase.class.isAssignableFrom(event.getResult().getClass())) {
  63             CdmBase original = (CdmBase) event.getOriginal();
  64             CdmBase result = (CdmBase) event.getResult();
  65             removeNullFromCollections(result);
  66             if(original != null && Hibernate.isInitialized(original) && original.getId() == 0 &&
  67                     result != null && Hibernate.isInitialized(result) && result.getId() > 0) {
  68                 original.setId(result.getId());
  69                 Set<CdmBase> newEntities = newEntitiesMap.get(event.getSession());
  70                 if(newEntities != null) {
  71                     newEntities.add(result);
  72                 }
  73             }
  74         }
  75
  76
  77     }
  78
  79     /**
  80      * @param entity
  81      */
  82     private static void removeNullFromCollections(Object entity) {
  83         if (entity != null){
  84             Class<?> entityClazz = entity.getClass();
  85
  86             if (TaxonNode.class.isAssignableFrom(entityClazz)){
  87                 TaxonNode node = (TaxonNode)entity;
  88                 node.removeNullValueFromChildren();
  89             } else if (PolytomousKeyNode.class.isAssignableFrom(entityClazz)){
  90                 PolytomousKeyNode node = (PolytomousKeyNode) entity;
  91                 if (node.getChildren() != null && Hibernate.isInitialized(node.getChildren()) ){
  92                     node.removeNullValueFromChildren();
  93
  94                 }
  95
  96             }   else if(FeatureTree.class.isAssignableFrom(entityClazz)){
  97
  98                 FeatureTree tree = (FeatureTree)entity;
  99                 for (FeatureNode node:tree.getRootChildren()){
 100                     node.removeNullValueFromChildren();
 101                     if (node.getChildNodes() != null){
 102                         for (FeatureNode childNode: node.getChildNodes()){
 103                             removeNullFromCollections(childNode);
 104                         }
 105                     }
 106
 107                 }
 108             } else if (FeatureNode.class.isAssignableFrom(entityClazz)){
 109                 FeatureNode node = (FeatureNode)entity;
 110                 node.removeNullValueFromChildren();
 111             }
 112
 113         }
 114
 115     }
 116
 117 }