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.apache.logging.log4j.LogManager;
  17 import org.apache.logging.log4j.Logger;
  18 import org.hibernate.Hibernate;
  19 import org.hibernate.HibernateException;
  20 import org.hibernate.Session;
  21 import org.hibernate.event.spi.MergeEvent;
  22 import org.hibernate.event.spi.MergeEventListener;
  23
  24 import eu.etaxonomy.cdm.model.common.CdmBase;
  25 import eu.etaxonomy.cdm.model.description.PolytomousKey;
  26 import eu.etaxonomy.cdm.model.description.PolytomousKeyNode;
  27 import eu.etaxonomy.cdm.model.taxon.TaxonNode;
  28 import eu.etaxonomy.cdm.model.term.TermNode;
  29 import eu.etaxonomy.cdm.model.term.TermTree;
  30
  31 /**
  32  * @author cmathew
  33  * @since 23 Sep 2015
  34  */
  35 public class PostMergeEntityListener implements MergeEventListener {
  36
  37     private static final long serialVersionUID = 1565797119368313987L;
  38     @SuppressWarnings("unused")
  39     private static final Logger logger = LogManager.getLogger(PostMergeEntityListener.class);
  40
  41     private static Map<Session, Set<CdmBase>> newEntitiesMap = new ConcurrentHashMap<>();
  42
  43     public static void addSession(Session session) {
  44         newEntitiesMap.put(session, new HashSet<>());
  45     }
  46
  47     public static void removeSession(Session session) {
  48         newEntitiesMap.remove(session);
  49     }
  50
  51     public static Set<CdmBase> getNewEntities(Session session) {
  52         return newEntitiesMap.get(session);
  53     }
  54
  55     @Override
  56     public void onMerge(MergeEvent event) throws HibernateException {
  57 //        Object entity = event.getEntity();
  58     }
  59
  60     @Override
  61     public void onMerge(MergeEvent event, Map copiedAlready) throws HibernateException {
  62         // any new entities are added to a map which is retrieved at the end of the
  63         // CdmEntityDaoBase.merge(T transientObject, boolean returnTransientEntity) call
  64         if(event.getOriginal() != null && CdmBase.class.isAssignableFrom(event.getOriginal().getClass()) &&
  65                 event.getResult() != null && CdmBase.class.isAssignableFrom(event.getResult().getClass())) {
  66             CdmBase original = (CdmBase) event.getOriginal();
  67             CdmBase result = (CdmBase) event.getResult();
  68             removeNullFromCollections(result);
  69             if(original != null && Hibernate.isInitialized(original) && original.getId() == 0 &&
  70                     result != null && Hibernate.isInitialized(result) && result.getId() > 0) {
  71                 original.setId(result.getId());
  72                 Set<CdmBase> newEntities = newEntitiesMap.get(event.getSession());
  73                 if(newEntities != null) {
  74                     newEntities.add(result);
  75                 }
  76             }
  77         }
  78     }
  79
  80     private static void removeNullFromCollections(Object entity) {
  81         if (entity != null){
  82             Class<?> entityClazz = entity.getClass();
  83
  84             if (TaxonNode.class.isAssignableFrom(entityClazz)){
  85                 TaxonNode node = (TaxonNode)entity;
  86 //                node.removeNullValueFromChildren();
  87             } else if (PolytomousKeyNode.class.isAssignableFrom(entityClazz)){
  88                 PolytomousKeyNode node = (PolytomousKeyNode) entity;
  89                 if (node.getChildren() != null && Hibernate.isInitialized(node.getChildren()) ){
  90                     node.removeNullValueFromChildren();
  91                     for (PolytomousKeyNode childNode: node.getChildren()){
  92                         removeNullFromCollections(childNode);
  93                     }
  94                 }
  95             }else if (PolytomousKey.class.isAssignableFrom(entityClazz)){
  96                 PolytomousKey key = (PolytomousKey) entity;
  97                 PolytomousKeyNode node = key.getRoot();
  98                 if (node != null && node.getChildren() != null && Hibernate.isInitialized(node.getChildren()) ){
  99                     node.removeNullValueFromChildren();
 100                     for (PolytomousKeyNode childNode: node.getChildren()){
 101                         removeNullFromCollections(childNode);
 102                     }
 103                 }
 104             }else if(TermTree.class.isAssignableFrom(entityClazz)){
 105                 //do nothing (remove if #8127/#3722 is fully solved
 106             } else if (TermNode.class.isAssignableFrom(entityClazz)){
 107                 //do nothing (remove if #8127/#3722 is fully solved
 108             }
 109         }
 110     }
 111 }