WIP: New state machine for message processing

[aquarium] / src / main / scala / gr / grnet / aquarium / charging / state / UserStateBootstrap.scala

diff --git a/src/main/scala/gr/grnet/aquarium/charging/state/UserStateBootstrap.scala b/src/main/scala/gr/grnet/aquarium/charging/state/UserStateBootstrap.scala
index f0888b8..d107f5e 100644 (file)
--- a/src/main/scala/gr/grnet/aquarium/charging/state/UserStateBootstrap.scala
+++ b/src/main/scala/gr/grnet/aquarium/charging/state/UserStateBootstrap.scala
@@ -36,6 +36,7 @@
 package gr.grnet.aquarium.charging.state
 
 import gr.grnet.aquarium.policy.UserAgreementModel
+import gr.grnet.aquarium.event.CreditsModel
 
 /**
  * This is used to bootstrap the [[gr.grnet.aquarium.charging.state.UserStateModel]].
@@ -43,9 +44,13 @@ import gr.grnet.aquarium.policy.UserAgreementModel
  * @author Christos KK Loverdos <loverdos@gmail.com>
  */
 
-case class UserStateBootstrap(
-    userID: String,
-    userCreationMillis: Long,
+final case class UserStateBootstrap(
     initialAgreement: UserAgreementModel,
-    initialCredits: Double
-)
+    initialCredits: CreditsModel.Type
+) {
+  def userID = initialAgreement.msg.getUserID
+
+  def userCreationMillis = initialAgreement.msg.getValidFromMillis
+
+  override def toString = "userID = %s".format(userID)
+}