* The ID of the object
* @param object
* The object to store
+ * @return The real object, or the shell if there is no real object yet
*/
@SuppressWarnings("unchecked")
- public void put(String id, T object) {
+ public T put(String id, T object) {
if (!(object instanceof Shell<?>)) {
objectCache.put(id, object);
shellCache.remove(id);
- } else {
- shellCache.put(id, (Shell<T>) object);
+ return object;
}
+ if (objectCache.containsKey(id)) {
+ return objectCache.get(id);
+ }
+ shellCache.put(id, (Shell<T>) object);
+ return object;
}
/**