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

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