X-Git-Url: https://code.grnet.gr/git/aquarium/blobdiff_plain/d818f53ce4173f6bd49431ab5c2af4735fbd6f07..c66d56a7b64c619e49ae2bc2b0ddcf9905fbbbd9:/src/main/scala/gr/grnet/aquarium/actor/message/GetUserBalanceResponse.scala?ds=sidebyside

diff --git a/src/main/scala/gr/grnet/aquarium/actor/message/GetUserBalanceResponse.scala b/src/main/scala/gr/grnet/aquarium/actor/message/GetUserBalanceResponse.scala
index 790984e..e1158d8 100644
--- a/src/main/scala/gr/grnet/aquarium/actor/message/GetUserBalanceResponse.scala
+++ b/src/main/scala/gr/grnet/aquarium/actor/message/GetUserBalanceResponse.scala
@@ -35,12 +35,24 @@
 
 package gr.grnet.aquarium.actor.message
 
+import gr.grnet.aquarium.AquariumInternalError
+
 /**
  *
  * @author Christos KK Loverdos <loverdos@gmail.com>
  */
 
 case class GetUserBalanceResponse(
-    userID: String,
-    balance: Either[String, Double])
-extends RouterResponseMessage(balance)
+    balance: Either[String, GetUserBalanceResponseData],
+    override val suggestedHTTPStatus: Int = 200)
+extends UserActorResponseMessage(balance, suggestedHTTPStatus) {
+  def userID = balance match {
+    case Left(error) ⇒
+      throw new AquariumInternalError("Could not obtain userID. %s".format(error))
+
+    case Right(data) ⇒
+      data.userID
+  }
+}
+
+case class GetUserBalanceResponseData(userID: String, balance: String)