997526936ece9d196fa916bbd964fa0cf18a60ba
[taxeditor.git] / eu.etaxonomy.taxeditor.cdmlib / src / main / java / eu / etaxonomy / taxeditor / remoting / cache / CdmModelCacher.java
1 package eu.etaxonomy.taxeditor.remoting.cache;
2
3 import java.io.FileInputStream;
4 import java.io.FileOutputStream;
5 import java.io.IOException;
6 import java.io.ObjectInputStream;
7 import java.io.ObjectOutputStream;
8 import java.lang.reflect.Field;
9 import java.net.URISyntaxException;
10 import java.net.URL;
11 import java.util.HashMap;
12 import java.util.Iterator;
13 import java.util.Map;
14
15 import net.sf.ehcache.Cache;
16 import net.sf.ehcache.Element;
17
18 import org.eclipse.core.runtime.FileLocator;
19 import org.eclipse.core.runtime.Platform;
20 import org.hibernate.SessionFactory;
21 import org.hibernate.boot.MetadataSources;
22 import org.hibernate.boot.model.naming.ImplicitNamingStrategyJpaCompliantImpl;
23 import org.hibernate.boot.registry.StandardServiceRegistry;
24 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
25 import org.hibernate.cfg.Configuration;
26 import org.hibernate.mapping.PersistentClass;
27 import org.hibernate.mapping.Property;
28 import org.hibernate.metadata.ClassMetadata;
29 import org.hibernate.property.access.spi.Getter;
30 import org.osgi.framework.Bundle;
31
32 import org.hibernate.boot.Metadata;
33
34
35 public class CdmModelCacher {
36
37
38
39
40 public static String HB_CONFIG_FILE_PATH= "/eu/etaxonomy/cdm/mappings/hibernate.cfg.xml";
41
42 public static final String CDM_MAP_SER_FILE_PATH = "resources/cdm.map.ser";
43
44
45
46
47 public void cacheGetterFields(Cache cache) throws IOException, ClassNotFoundException, URISyntaxException {
48 Map<String, CdmModelFieldPropertyFromClass> modelClassMap = loadModelClassMap();
49
50 cache.removeAll();
51
52 for(Map.Entry<String, CdmModelFieldPropertyFromClass> entry : modelClassMap.entrySet()) {
53 cache.put(new Element(entry.getKey(), entry.getValue()));
54 }
55 }
56
57 public Map<String, CdmModelFieldPropertyFromClass> loadModelClassMap() throws URISyntaxException, IOException, ClassNotFoundException {
58
59 Bundle bundle = Platform.getBundle("eu.etaxonomy.taxeditor.cdmlib");
60
61 URL modelMapFileBundleURL = bundle.getEntry(CDM_MAP_SER_FILE_PATH);
62 URL modelMapFileURL = FileLocator.resolve(modelMapFileBundleURL);
63 String modelMapFilePath = modelMapFileURL.getFile();
64
65 FileInputStream fin = new FileInputStream(modelMapFilePath);
66 ObjectInputStream ois = new ObjectInputStream(fin);
67 Map<String, CdmModelFieldPropertyFromClass> modelClassMap = (Map<String, CdmModelFieldPropertyFromClass>) ois.readObject();
68 ois.close();
69 return modelClassMap;
70 }
71
72
73 public Map<String, CdmModelFieldPropertyFromClass> generateModelClassMap() {
74
75 // A SessionFactory is set up once for an application!
76 final StandardServiceRegistry registry = new StandardServiceRegistryBuilder()
77 .configure(HB_CONFIG_FILE_PATH) // configures settings from hibernate.cfg.xml
78 .build();
79 SessionFactory sessionFactory = null;
80 Map<String, CdmModelFieldPropertyFromClass> modelClassMap = new HashMap<String, CdmModelFieldPropertyFromClass>();
81 try {
82 sessionFactory = new MetadataSources( registry ).buildMetadata().buildSessionFactory();
83 Configuration configuration = buildConfiguration(HB_CONFIG_FILE_PATH);
84 Map<String, ClassMetadata> classMetaDataMap = sessionFactory.getAllClassMetadata();
85 Metadata metadata = new MetadataSources( registry ).getMetadataBuilder().applyImplicitNamingStrategy( ImplicitNamingStrategyJpaCompliantImpl.INSTANCE ).build();
86
87
88 for(ClassMetadata classMetaData :classMetaDataMap.values()) {
89 Class mappedClass = classMetaData.getMappedClass();
90
91 String mappedClassName = mappedClass.getName();
92
93 PersistentClass persistentClass =metadata.getEntityBinding(mappedClassName);
94 CdmModelFieldPropertyFromClass cmgmfc = new CdmModelFieldPropertyFromClass(mappedClassName);
95 System.out.println("Adding class : " + mappedClassName + " to cache");
96 addGetters(persistentClass, cmgmfc);
97 modelClassMap.put(mappedClassName, cmgmfc);
98 }
99 }
100 catch (Exception e) {
101 // The registry would be destroyed by the SessionFactory, but we had trouble building the SessionFactory
102 // so destroy it manually.
103 StandardServiceRegistryBuilder.destroy( registry );
104 }
105
106
107 return modelClassMap;
108 }
109
110
111 public static Configuration buildConfiguration(String hibernateConfigFilePath) {
112 Configuration configuration = new Configuration().configure(hibernateConfigFilePath);
113 configuration.buildMappings();
114 return configuration;
115 }
116
117 private void addGetters(PersistentClass persistentClass, CdmModelFieldPropertyFromClass cmgmfc) {
118 if (persistentClass != null) {
119 Iterator propertyIt = persistentClass.getPropertyIterator();
120
121 while(propertyIt.hasNext())
122 {
123 Property property = (Property)propertyIt.next();
124 Getter getter = property.getGetter(persistentClass.getMappedClass());
125 if(getter != null && getter.getMember() != null) {
126 Field field = (Field)getter.getMember();
127
128 //logger.info(" - contains field '" + field.getName() + "' of type '" + field.getType().getName() + "'");
129 cmgmfc.addGetMethods(field.getName());
130 }
131 }
132 addGetters(persistentClass.getSuperclass(), cmgmfc);
133 }
134 }
135
136
137
138 public static void main(String argv[]) {
139
140 // To re-create the serialised cdm map run,
141 // mvn exec:java -Dexec.mainClass="eu.etaxonomy.taxeditor.remoting.cache.CdmModelCacher"
142 // in the eu.etaxonomy.taxeditor.cdmlib project root dir
143 String CDM_MAP_SER_DIR = "resources/";
144
145 CdmModelCacher cdmModelCacher = new CdmModelCacher();
146 Map<String, CdmModelFieldPropertyFromClass> modelClassMap = cdmModelCacher.generateModelClassMap();
147 try{
148
149 FileOutputStream fout = new FileOutputStream(CDM_MAP_SER_DIR + "cdm.map.ser");
150 ObjectOutputStream oos = new ObjectOutputStream(fout);
151 oos.writeObject(modelClassMap);
152 oos.close();
153 System.out.println("CDM Map serialized");
154
155 }catch(Exception ex){
156 ex.printStackTrace();
157 }
158
159 }
160
161
162 }