Fix NPE with initial user state
[aquarium] / src / main / scala / gr / grnet / aquarium / computation / state / parts / OwnedResourcesSnapshot.scala
1 /*
2  * Copyright 2011-2012 GRNET S.A. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or
5  * without modification, are permitted provided that the following
6  * conditions are met:
7  *
8  *   1. Redistributions of source code must retain the above
9  *      copyright notice, this list of conditions and the following
10  *      disclaimer.
11  *
12  *   2. Redistributions in binary form must reproduce the above
13  *      copyright notice, this list of conditions and the following
14  *      disclaimer in the documentation and/or other materials
15  *      provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28  * POSSIBILITY OF SUCH DAMAGE.
29  *
30  * The views and conclusions contained in the software and
31  * documentation are those of the authors and should not be
32  * interpreted as representing official policies, either expressed
33  * or implied, of GRNET S.A.
34  */
35
36 package gr.grnet.aquarium.computation
37 package state
38 package parts
39
40 /**
41  *
42  * @author Christos KK Loverdos <loverdos@gmail.com>
43  */
44
45 case class OwnedResourcesSnapshot(resourceInstanceSnapshots: List[ResourceInstanceAmount]) {
46
47   def toResourcesMap: OwnedResourcesMap = {
48     val tuples = for(rc <- resourceInstanceSnapshots) yield ((rc.resource, rc.instanceID), (rc.instanceAmount))
49
50     new OwnedResourcesMap(Map(tuples.toSeq: _*))
51   }
52
53   def resourceInstanceSnapshotsExcept(resource: String, instanceId: String) = {
54     // Unfortunately, we have to use a List for parts, since JSON serialization is not as flexible
55     // (at least out of the box). Thus, the update is O(L), where L is the length of the parts List.
56     resourceInstanceSnapshots.filterNot(_.isSameResourceInstance(resource, instanceId))
57   }
58
59   def findResourceInstanceSnapshot(resource: String, instanceId: String): Option[ResourceInstanceAmount] = {
60     resourceInstanceSnapshots.find(x => resource == x.resource && instanceId == x.instanceID)
61   }
62
63   def getResourceInstanceAmount(resource: String, instanceId: String, defaultValue: Double): Double = {
64     findResourceInstanceSnapshot(resource, instanceId).map(_.instanceAmount).getOrElse(defaultValue)
65   }
66
67   def computeResourcesSnapshotUpdate(resource: String,   // resource name
68                                      instanceId: String, // resource instance id
69                                      newAmount: Double,
70                                      snapshotTime: Long): (OwnedResourcesSnapshot,
71                                                           Option[ResourceInstanceAmount],
72     ResourceInstanceAmount) = {
73
74     val newResourceInstance = ResourceInstanceAmount(resource, instanceId, newAmount)
75     val oldResourceInstanceOpt = this.findResourceInstanceSnapshot(resource, instanceId)
76
77     val newResourceInstances = oldResourceInstanceOpt match {
78       case Some(oldResourceInstance) ⇒
79         // Resource instance found, so delete the old one and add the new one
80         newResourceInstance :: resourceInstanceSnapshotsExcept(resource, instanceId)
81
82       case None ⇒
83         // Resource not found, so this is the first time and we just add the new snapshot
84         newResourceInstance :: resourceInstanceSnapshots
85     }
86
87     val newOwnedResources = OwnedResourcesSnapshot(newResourceInstances)
88
89     (newOwnedResources, oldResourceInstanceOpt, newResourceInstance)
90  }
91 }
92
93 object OwnedResourcesSnapshot {
94   final val Empty = OwnedResourcesSnapshot(Nil)
95 }