Produce JSON policy configuration from Avro message
[aquarium] / src / main / scala / gr / grnet / aquarium / message / avro / MessageFactory.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.message.avro
37
38 import scala.collection.JavaConverters.seqAsJavaListConverter
39 import scala.collection.JavaConverters.mapAsJavaMapConverter
40 /**
41  *
42  * @author Christos KK Loverdos <loverdos@gmail.com>
43  */
44 object MessageFactory {
45   def newEffectiveUnitPrice(unitPrice: Double, whenOpt: Option[_CronSpecTuple] = None) = {
46     _EffectiveUnitPrice.newBuilder().
47       setUnitPrice(unitPrice).
48       setWhen(whenOpt.getOrElse(null)).
49     build()
50   }
51
52   def newEffectivePriceTable(priceOverrides: _EffectiveUnitPrice*) = {
53     _EffectivePriceTable.newBuilder().
54       setPriceOverrides(priceOverrides.asJava).
55     build()
56   }
57
58   def newSelectorValue(ept: _EffectivePriceTable): _SelectorValue = {
59     _SelectorValue.newBuilder().
60       setSelectorValue(ept).
61     build()
62   }
63
64   def newSelectorValue(map: Map[CharSequence, _SelectorValue]): _SelectorValue = {
65     _SelectorValue.newBuilder().
66       setSelectorValue(map.asJava).
67     build()
68   }
69
70   def newSelectorValue(pairs: (CharSequence, _SelectorValue)*): _SelectorValue = {
71     _SelectorValue.newBuilder().
72       setSelectorValue(Map(pairs:_*).asJava).
73     build()
74   }
75
76 //  def newFullPriceTable(perResource: Map[CharSequence, Map[CharSequence, _SelectorValue]]) = {
77 //    _FullPriceTable.newBuilder().
78 //      setPerResource(
79 //        (for((k, v) ← perResource) yield (k, v.asJava)).asJava
80 //      ).
81 //    build()
82 //  }
83
84   def newFullPriceTable(perResource: (CharSequence, Map[CharSequence, _SelectorValue])*) = {
85     _FullPriceTable.newBuilder().
86       setPerResource(
87         Map((for((k, v) ← perResource) yield (k, v.asJava)):_*).asJava
88       ).
89     build()
90   }
91
92   def newRoleMapping(map: Map[CharSequence, _FullPriceTable]): java.util.Map[CharSequence, _FullPriceTable] = {
93     map.asJava
94   }
95
96   def newRoleMapping(pairs: (CharSequence, _FullPriceTable)*): java.util.Map[CharSequence, _FullPriceTable] = {
97     Map(pairs:_*).asJava
98   }
99
100   def newResourceType(name: CharSequence, unit: CharSequence, chargingBehavior: CharSequence) = {
101     _ResourceType.newBuilder().
102       setName(name).
103       setUnit(unit).
104       setChargingBehaviorClass(chargingBehavior).
105     build()
106   }
107
108   def newResourceTypes(rts: _ResourceType*) = {
109     rts.asJava
110   }
111
112   def newChargingBehaviors(cbs: CharSequence*) = {
113     cbs.asJava
114   }
115 }