Add one convenient method to user state store

But this will have to be consolidated at some point.

diff --git a/src/main/scala/gr/grnet/aquarium/store/UserStateStore.scala b/src/main/scala/gr/grnet/aquarium/store/UserStateStore.scala
index 4ec0528..425f4b6 100644 (file)
--- a/src/main/scala/gr/grnet/aquarium/store/UserStateStore.scala
+++ b/src/main/scala/gr/grnet/aquarium/store/UserStateStore.scala
@@ -53,6 +53,15 @@ trait UserStateStore {
    */
   def storeUserState(userState: UserState): Maybe[RecordID]
 
+
+  def storeUserState2(userState: UserState): Maybe[UserState] = {
+    for {
+      recordID <- storeUserState(userState)
+    } yield {
+      userState.copy(_id = recordID.id)
+    }
+  }
+
   /**
    * Find a state by user ID
    */