Update available policies when on configuration file update

[aquarium] / src / main / scala / gr / grnet / aquarium / store / mongodb / MongoDBStore.scala

diff --git a/src/main/scala/gr/grnet/aquarium/store/mongodb/MongoDBStore.scala b/src/main/scala/gr/grnet/aquarium/store/mongodb/MongoDBStore.scala
index 48387ac..16721ed 100644 (file)
--- a/src/main/scala/gr/grnet/aquarium/store/mongodb/MongoDBStore.scala
+++ b/src/main/scala/gr/grnet/aquarium/store/mongodb/MongoDBStore.scala
@@ -48,10 +48,10 @@ import gr.grnet.aquarium.logic.events.UserEvent.{JsonNames => UserEventJsonNames
 import gr.grnet.aquarium.logic.events.WalletEntry.{JsonNames => WalletJsonNames}
 import java.util.Date
 import com.ckkloverdos.maybe.Maybe
-import gr.grnet.aquarium.logic.events.{UserEvent, WalletEntry, ResourceEvent, AquariumEvent}
-import com.mongodb._
 import gr.grnet.aquarium.logic.accounting.Policy
-import gr.grnet.aquarium.logic.accounting.dsl.DSLComplexResource
+import gr.grnet.aquarium.logic.accounting.dsl.{Timeslot, DSLPolicy, DSLComplexResource}
+import gr.grnet.aquarium.logic.events._
+import com.mongodb._
 
 /**
  * Mongodb implementation of the various aquarium stores.
@@ -68,12 +68,14 @@ class MongoDBStore(
   with UserStateStore
   with WalletEntryStore
   with UserEventStore
+  with PolicyStore
   with Loggable {
 
   private[store] lazy val resourceEvents = getCollection(MongoDBStore.RESOURCE_EVENTS_COLLECTION)
   private[store] lazy val userStates     = getCollection(MongoDBStore.USER_STATES_COLLECTION)
   private[store] lazy val userEvents     = getCollection(MongoDBStore.USER_EVENTS_COLLECTION)
   private[store] lazy val walletEntries  = getCollection(MongoDBStore.WALLET_ENTRIES_COLLECTION)
+  private[store] lazy val policies      = getCollection(MongoDBStore.POLICIES_COLLECTION)
 
   private[this] def getCollection(name: String): DBCollection = {
     val db = mongo.getDB(database)
@@ -302,6 +304,24 @@ class MongoDBStore(
     MongoDBStore.runQuery(query, userEvents)(MongoDBStore.dbObjectToUserEvent)(Some(_sortByTimestampAsc))
   }
   //-UserEventStore
+
+  //+PolicyStore
+  def loadPolicies(after: Long): List[PolicyEntry] = {
+    val query = new BasicDBObject(PolicyEntry.JsonNames.validFrom,
+      new BasicDBObject("$gt", after))
+    MongoDBStore.runQuery(query, policies)(MongoDBStore.dbObjectToPolicyEvent)(Some(_sortByTimestampAsc))
+  }
+
+  def storePolicy(policy: PolicyEntry): Maybe[RecordID] = MongoDBStore.storeAquariumEvent(policy, policies)
+
+
+  def updatePolicy(policy: PolicyEntry) = {
+    //Find the entry
+    val query = new BasicDBObject(PolicyEntry.JsonNames.id, policy.id)
+    val policyObject = MongoDBStore.jsonSupportToDBObject(policy)
+    policies.update(query, policyObject, true, false)
+  }
+  //-PolicyStore
 }
 
 object MongoDBStore {
@@ -330,7 +350,6 @@ object MongoDBStore {
    */
   final val USER_EVENTS_COLLECTION = "userevents"
 
-
   /**
    * Collection holding [[gr.grnet.aquarium.logic.events.WalletEntry]].
    *
@@ -338,6 +357,11 @@ object MongoDBStore {
    */
   final val WALLET_ENTRIES_COLLECTION = "wallets"
 
+  /**
+   * Collection holding [[gr.grnet.aquarium.logic.accounting.dsl.DSLPolicy]].
+   */
+  final val POLICIES_COLLECTION = "policies"
+
   /* TODO: Some of the following methods rely on JSON (de-)serialization).
   * A method based on proper object serialization would be much faster.
   */
@@ -357,6 +381,10 @@ object MongoDBStore {
     UserEvent.fromJson(JSON.serialize(dbObj))
   }
 
+  def dbObjectToPolicyEvent(dbObj: DBObject): PolicyEntry = {
+    PolicyEntry.fromJson(JSON.serialize(dbObj))
+  }
+
   def findById[A >: Null <: AquariumEvent](id: String, collection: DBCollection, deserializer: (DBObject) => A) : Maybe[A] = Maybe {
     val query = new BasicDBObject(ResourceJsonNames.id, id)
     val cursor = collection find query