code.grnet.gr Git - aquarium/blob - src/main/scala/gr/grnet/aquarium/logic/accounting/dsl/DSLPolicy.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.logic.accounting.dsl
  37
  38 import gr.grnet.aquarium.util.CryptoUtils
  39 import gr.grnet.aquarium.util.date.TimeHelpers
  40 import gr.grnet.aquarium.event.model.PolicyEntry
  41
  42 /**
  43  * Root object for the Aquarium policy configuration tree.
  44  *
  45  * @author Georgios Gousios <gousiosg@gmail.com>
  46  */
  47 case class DSLPolicy (
  48   algorithms: List[DSLAlgorithm],
  49   pricelists: List[DSLPriceList],
  50   resources: List[DSLResource],
  51   creditplans: List[DSLCreditPlan],
  52   agreements: List[DSLAgreement]
  53 ) extends DSLItem {
  54
  55   /**Find a resource by name */
  56   def findResource(name: String): Option[DSLResource] = {
  57     resources.find(a => a.name.equals(name))
  58   }
  59
  60   /**Find a pricelist by name */
  61   def findPriceList(name: String): Option[DSLPriceList] = {
  62     pricelists.find(a => a.name.equals(name))
  63   }
  64
  65   /**Find an algorithm by name */
  66   def findAlgorithm(name: String): Option[DSLAlgorithm] = {
  67     algorithms.find(a => a.name.equals(name))
  68   }
  69
  70   /**Find a crediplan by name */
  71   def findCreditPlan(name: String): Option[DSLCreditPlan] = {
  72     creditplans.find(a => a.name.equals(name))
  73   }
  74
  75   /**Find an agreement by name */
  76   def findAgreement(name: String): Option[DSLAgreement] = {
  77     agreements.find(a => a.name.equals(name))
  78   }
  79
  80   override def toMap() = {
  81     val policy = Map(Vocabulary.resources   -> resources.map{r => r.toMap})   ++
  82     Map(Vocabulary.algorithms  -> algorithms.map{a => a.toMap})  ++
  83     Map(Vocabulary.pricelists  -> pricelists.map{p => p.toMap})  ++
  84     Map(Vocabulary.creditplans -> creditplans.map{c => c.toMap}) ++
  85     Map(Vocabulary.agreements  -> agreements.map{a => a.toMap})
  86     Map(Vocabulary.aquariumpolicy -> policy)
  87   }
  88
  89   def resourcesMap: DSLResourcesMap =
  90     new DSLResourcesMap(resources)
  91
  92   def toPolicyEntry: PolicyEntry = {
  93     val yaml = toYAML
  94     val ts = TimeHelpers.nowMillis()
  95     PolicyEntry(CryptoUtils.sha1(yaml), ts, ts, yaml, ts + 1, Long.MaxValue)
  96   }
  97
  98   def toPolicyEntry(occurredMillis: Long, validFromMillis: Long, validToMillis: Long = Long.MaxValue) = {
  99     val yaml = toYAML
 100     val id = CryptoUtils.sha1(yaml)
 101     PolicyEntry(id, occurredMillis,  occurredMillis, yaml, validFromMillis, validToMillis)
 102   }
 103 }
 104
 105 object DSLPolicy {
 106
 107   val emptyPolicy = DSLPolicy(List(), List(), List(), List(), List())
 108
 109   object JsonNames {
 110     val valid = "valid"
 111     val _id = "_id"
 112   }
 113 }