fix: cache removal

[gigi.git] / src / org / cacert / gigi / dbObjects / ObjectCache.java

diff --git a/src/org/cacert/gigi/dbObjects/ObjectCache.java b/src/org/cacert/gigi/dbObjects/ObjectCache.java
index c17dcacff4095ce666bb825a7d1e8394b5b5125b..125477defc5fd26e33bff9d51153d9617c06a15b 100644 (file)
--- a/src/org/cacert/gigi/dbObjects/ObjectCache.java
+++ b/src/org/cacert/gigi/dbObjects/ObjectCache.java
@@ -14,8 +14,9 @@ public class ObjectCache<T extends IdCachable> {
         caches.add(this);
     }
 
-    public void put(T c) {
+    public T put(T c) {
         hashmap.put(c.getId(), new WeakReference<T>(c));
+        return c;
     }
 
     public T get(int id) {
@@ -33,6 +34,6 @@ public class ObjectCache<T extends IdCachable> {
     }
 
     public void remove(T toRm) {
-        hashmap.remove(toRm);
+        hashmap.remove(toRm.getId());
     }
 }