X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=server%2Fruntime%2Forg.argeo.server.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fserver%2Fdao%2FAbstractMemoryDaoSupport.java;h=d18514a278fe155842877af874d32c131d4f0d2b;hb=770c76924f74e070100815c68c3b920e795a41c9;hp=39655d4fe04812fc9a2027cab7c4fd1723b27294;hpb=d8c2b3440de26af2f60e90e4e1b1f4fdbc7b0735;p=lgpl%2Fargeo-commons.git

diff --git a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractMemoryDaoSupport.java b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractMemoryDaoSupport.java
index 39655d4fe..d18514a27 100644
--- a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractMemoryDaoSupport.java
+++ b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractMemoryDaoSupport.java
@@ -12,7 +12,7 @@ import java.util.TreeMap;
 import org.apache.commons.io.IOUtils;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
-import org.argeo.server.ArgeoServerException;
+import org.argeo.ArgeoException;
 import org.springframework.beans.BeanWrapper;
 import org.springframework.beans.BeanWrapperImpl;
 import org.springframework.beans.BeansException;
@@ -37,7 +37,7 @@ public abstract class AbstractMemoryDaoSupport implements LightDaoSupport,
 
 	private List<String> scannedPackages = new ArrayList<String>();
 
-	private List<Resource> workbooks = new ArrayList<Resource>();
+	private List<Resource> resources = new ArrayList<Resource>();
 
 	private Map<Class<?>, PropertyEditor> customEditors = new HashMap<Class<?>, PropertyEditor>();;
 
@@ -50,19 +50,21 @@ public abstract class AbstractMemoryDaoSupport implements LightDaoSupport,
 	}
 
 	public void init() {
-		// used to resolve inner references
-		// Map<String, List<Object>> tempRefs = new HashMap<String,
-		// List<Object>>();
+		for (PropertyEditor propertyEditor : customEditors.values())
+			if (propertyEditor instanceof LightDaoAware) {
+				((LightDaoAware) propertyEditor).setLightDaoSupport(this);
+			}
 
+		// Load data
 		List<Reference> references = new ArrayList<Reference>();
 
-		for (Resource res : workbooks) {
+		for (Resource res : resources) {
 			InputStream in = null;
 			try {
 				in = res.getInputStream();
 				load(in, references);
 			} catch (Exception e) {
-				throw new ArgeoServerException("Cannot load stream", e);
+				throw new ArgeoException("Cannot load stream", e);
 			} finally {
 				IOUtils.closeQuietly(in);
 			}
@@ -112,6 +114,8 @@ public abstract class AbstractMemoryDaoSupport implements LightDaoSupport,
 
 	@SuppressWarnings("unchecked")
 	public <T> T getByKey(Class<T> clss, Object key) {
+		if (key == null)
+			throw new ArgeoException("Key is null for " + clss);
 		return (T) model.get(findClass(clss)).get(key);
 	}
 
@@ -196,12 +200,12 @@ public abstract class AbstractMemoryDaoSupport implements LightDaoSupport,
 		return scannedPackages;
 	}
 
-	public void setWorkbooks(List<Resource> workbooks) {
-		this.workbooks = workbooks;
+	public void setResources(List<Resource> workbooks) {
+		this.resources = workbooks;
 	}
 
-	public List<Resource> getWorkbooks() {
-		return workbooks;
+	public List<Resource> getResources() {
+		return resources;
 	}
 
 	public void setClassLoader(ClassLoader classLoader) {