Remove updated time setter from Sone, store update time in database.

[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemorySoneDatabase.java

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneDatabase.java
index 67f3425..8435959 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemorySoneDatabase.java
@@ -1,6 +1,8 @@
 package net.pterodactylus.sone.database.memory;
 
+import java.util.HashMap;
 import java.util.HashSet;
+import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.locks.ReadWriteLock;
 import java.util.concurrent.locks.ReentrantReadWriteLock;
@@ -17,6 +19,7 @@ public class MemorySoneDatabase {
        private final ReadWriteLock lock = new ReentrantReadWriteLock();
        private final ConfigurationLoader configurationLoader;
        private final Set<String> knownSones = new HashSet<String>();
+       private final Map<String, Long> soneUpdateTimes = new HashMap<String, Long>();
 
        public MemorySoneDatabase(ConfigurationLoader configurationLoader) {
                this.configurationLoader = configurationLoader;
@@ -68,4 +71,13 @@ public class MemorySoneDatabase {
                }
        }
 
+       void updateSoneTime(String soneId, long soneTime) {
+               lock.writeLock().lock();
+               try {
+                       soneUpdateTimes.put(soneId, soneTime);
+               } finally {
+                       lock.writeLock().unlock();
+               }
+       }
+
 }