2 * Copyright 2011 GRNET S.A. All rights reserved.
4 * Redistribution and use in source and binary forms, with or
5 * without modification, are permitted provided that the following
8 * 1. Redistributions of source code must retain the above
9 * copyright notice, this list of conditions and the following
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.
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.
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.
36 package gr.grnet.aquarium.logic.accounting
38 import dsl.{Timeslot, DSLPolicy, DSL}
39 import gr.grnet.aquarium.Configurator._
40 import java.io.{InputStream, FileInputStream, File}
42 import com.ckkloverdos.maybe.{Maybe, Just}
43 import gr.grnet.aquarium.util.date.TimeHelpers
44 import gr.grnet.aquarium.logic.events.PolicyEntry
45 import gr.grnet.aquarium.util.{CryptoUtils, Loggable}
48 * Searches for and loads the applicable accounting policy
50 * @author Georgios Gousios <gousiosg@gmail.com>
52 object Policy extends DSL with Loggable {
54 private var policies = {reloadPolicies}
56 lazy val policy = loadPolicyFromFile(policyFile)
58 def policy(at: Date): Maybe[DSLPolicy] = Maybe {
60 a => a._1.from.before(at) &&
65 throw new AccountingException("No valid policy for date: %s".format(at))
69 def policies(from: Date, to: Date): List[DSLPolicy] = {
71 a => a._1.from.before(from) &&
76 def policies(t: Timeslot): List[DSLPolicy] = policies(t.from, t.to)
78 def loadPolicyFromFile(pol: File): DSLPolicy = {
80 val stream = pol.exists() match {
82 logger.info("Using policy file %s".format(pol.getAbsolutePath))
83 new FileInputStream(pol)
85 logger.warn(("Cannot find user configured policy file %s, " +
86 "looking for default policy").format(pol.getAbsolutePath))
87 getClass.getClassLoader.getResourceAsStream("policy.yaml") match {
88 case x: InputStream =>
89 logger.warn("Using default policy, this is problably not what you want")
92 logger.error("No valid policy file found, Aquarium will fail")
99 private def policyFile: File =
100 MasterConfigurator.props.get(Keys.aquarium_policy) match {
101 case Just(x) => new File(x)
102 case _ => new File("/etc/aquarium/policy.yaml")
105 private def reloadPolicies(): Map[Timeslot, DSLPolicy] = synchronized {
106 //1. Load policies from db
107 val policies = MasterConfigurator.policyEventStore.loadPolicies(0)
109 //2. Check whether policy file has been updated
110 val latestPolicyChange = if (policies.isEmpty) 0 else policies.last.validFrom
111 val policyf = policyFile
114 if (policyf.exists) {
115 if (policyf.lastModified > latestPolicyChange) {
116 logger.info("Policy changed since last check, reloading")
119 logger.info("Policy not changed since last check")
122 logger.warn("User specified policy file %s does not exist, " +
123 "using stored policy information".format(policyf.getAbsolutePath))
126 val toAdd = updated match {
128 val ts = TimeHelpers.nowMillis
129 val toUpdate = policies.last.copy(validTo = ts)
130 val parsedNew = loadPolicyFromFile(policyf)
131 val yaml = parsedNew.toYAML
132 val newPolicy = PolicyEntry(CryptoUtils.sha1(yaml), ts, ts, yaml, ts + 1, -1)
134 MasterConfigurator.policyEventStore.updatePolicy(toUpdate)
135 MasterConfigurator.policyEventStore.storePolicy(newPolicy)
137 List(toUpdate, newPolicy)
141 policies.init.++(toAdd).foldLeft(Map[Timeslot, DSLPolicy]()){
143 acc ++ Map(Timeslot(new Date(p.validFrom), new Date(p.validTo)) -> parse(p.policyYAML))