
woko.inmemory.InMemoryObjectStore Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2001-2012 Remi Vankeisbelck
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package woko.inmemory;
import woko.persistence.ObjectStore;
import woko.persistence.ResultIterator;
import woko.util.Util;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* In-memory implementation of ObjectStore
. Clearly not for production use !
*/
public class InMemoryObjectStore implements ObjectStore {
private final Map> objects = new HashMap>();
@Override
public void close() {
}
@Override
public Class getObjectClass(Object o) {
return o.getClass();
}
public void addObject(String key, Object obj) {
Util.assertArg("obj", obj);
Util.assertArg("key", key);
String className = getClassMapping(getObjectClass(obj));
Map objectsForClass = objects.get(className);
if (objectsForClass == null) {
objectsForClass = new HashMap();
objects.put(className, objectsForClass);
}
objectsForClass.put(key, obj);
}
public Object load(String className, String key) {
if (className == null) {
return null;
} else {
if (key == null) {
// create object from class
Class mappedClass = getMappedClass(className);
if (mappedClass != null) {
try {
return mappedClass.newInstance();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
return null;
} else {
Map objectsForClass = objects.get(className);
if (objectsForClass == null) {
return null;
}
return objectsForClass.get(key);
}
}
}
public Object save(Object obj) {
Util.assertArg("obj", obj);
addObject(getKey(obj), obj);
return obj;
}
public String getKey(Object obj) {
Util.assertArg("obj", obj);
for (String className : objects.keySet()) {
Map objectsForClass = objects.get(className);
for (String key : objectsForClass.keySet()) {
Object value = objectsForClass.get(key);
if (value != null && value.equals(obj)) {
return key;
}
}
}
return null;
}
public String getClassMapping(Class clazz) {
Util.assertArg("clazz", clazz);
return clazz.getName();
}
public Class getMappedClass(String className) {
if (className == null) {
return null;
}
try {
return Class.forName(className);
} catch (ClassNotFoundException e) {
return null;
}
}
public Object delete(Object obj) {
Util.assertArg("obj", obj);
String key = getKey(obj);
if (key != null) {
String className = getClassMapping(getObjectClass(obj));
if (className != null) {
Map objectsForClass = objects.get(className);
if (objectsForClass != null) {
return objectsForClass.remove(key);
}
}
}
return null;
}
public ResultIterator list(String className, Integer start, Integer limit) {
throw new UnsupportedOperationException();
}
public List> getMappedClasses() {
List> res = new ArrayList>();
for (String k : objects.keySet()) {
try {
res.add(Class.forName(k));
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
return res;
}
public ResultIterator search(Object query, Integer start, Integer limit) {
throw new UnsupportedOperationException("search is not implemented for the in-memory store");
}
}