code.grnet.gr Git - aquarium/blob - src/main/scala/gr/grnet/aquarium/charging/bill/BillEntry.scala

   1 package gr.grnet.aquarium.charging.bill
   2
   3 import gr.grnet.aquarium.charging.state.WorkingUserState
   4 import gr.grnet.aquarium.util.json.JsonSupport
   5 import com.ckkloverdos.resource.FileStreamResource
   6 import java.io.File
   7 import com.ckkloverdos.props.Props
   8 import gr.grnet.aquarium.converter.{CompactJsonTextFormat, PrettyJsonTextFormat, StdConverters}
   9 import gr.grnet.aquarium.{Aquarium, ResourceLocator, AquariumBuilder}
  10 import gr.grnet.aquarium.store.memory.MemStoreProvider
  11 import gr.grnet.aquarium.converter.StdConverters._
  12 import scala._
  13 import gr.grnet.aquarium.logic.accounting.dsl.Timeslot
  14 import java.util.concurrent.atomic.AtomicLong
  15 import java.util.{Date, Calendar, GregorianCalendar}
  16 import gr.grnet.aquarium.charging.wallet.WalletEntry
  17 import scala.collection.parallel.mutable
  18 import scala.collection.mutable.ListBuffer
  19 import gr.grnet.aquarium.Aquarium.EnvKeys
  20 import gr.grnet.aquarium.charging.Chargeslot
  21 import scala.collection.immutable.TreeMap
  22 import scala.Some
  23 import gr.grnet.aquarium.charging.Chargeslot
  24
  25
  26 /*
  27 * Copyright 2011-2012 GRNET S.A. All rights reserved.
  28 *
  29 * Redistribution and use in source and binary forms, with or
  30 * without modification, are permitted provided that the following
  31 * conditions are met:
  32 *
  33 *   1. Redistributions of source code must retain the above
  34 *      copyright notice, this list of conditions and the following
  35 *      disclaimer.
  36 *
  37 *   2. Redistributions in binary form must reproduce the above
  38 *      copyright notice, this list of conditions and the following
  39 *      disclaimer in the documentation and/or other materials
  40 *      provided with the distribution.
  41 *
  42 * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  43 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  44 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  45 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  46 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  47 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  48 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  49 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  50 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  51 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  52 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  53 * POSSIBILITY OF SUCH DAMAGE.
  54 *
  55 * The views and conclusions contained in the software and
  56 * documentation are those of the authors and should not be
  57 * interpreted as representing official policies, either expressed
  58 * or implied, of GRNET S.A.
  59 */
  60
  61
  62 /*
  63 * @author Prodromos Gerakios <pgerakios@grnet.gr>
  64 */
  65
  66 case class ChargeEntry(val id:String,
  67                        val unitPrice:String,
  68                        val startTime:String,
  69                        val endTime:String,
  70                        val ellapsedTime:String,
  71                        val credits:String)
  72   extends JsonSupport {}
  73
  74
  75 class EventEntry(val eventType : String,
  76                  val details   : List[ChargeEntry])
  77  extends JsonSupport {}
  78
  79
  80 case class ResourceEntry(val resourceName : String,
  81                          val resourceType : String,
  82                          val unitName : String,
  83                          val totalCredits : String,
  84                          val details : List[EventEntry])
  85 extends JsonSupport {}
  86
  87
  88 abstract class AbstractBillEntry
  89  extends JsonSupport {}
  90
  91 class BillEntry(val id:String,
  92                 val userID : String,
  93                 val status : String,
  94                 val remainingCredits:String,
  95                 val deductedCredits:String,
  96                 val startTime:String,
  97                 val endTime:String,
  98                 val bill:List[ResourceEntry]
  99               )
 100  extends AbstractBillEntry {}
 101
 102
 103 object AbstractBillEntry {
 104
 105   private[this] val counter = new AtomicLong(0L)
 106   private[this] def nextUIDObject() = counter.getAndIncrement
 107
 108   /*private[this] def walletTimeslot(i:WalletEntry) : Timeslot = {
 109     val cal = new GregorianCalendar
 110     cal.set(i.billingYear,i.billingMonth,1)
 111     val dstart = cal.getTime
 112     val lastDate = cal.getActualMaximum(Calendar.DATE)
 113     cal.set(Calendar.DATE, lastDate)
 114     val dend = cal.getTime
 115    Timeslot(dstart,dend)
 116   } */
 117
 118   private[this] def toChargeEntry(c:Chargeslot) : ChargeEntry = {
 119     val unitPrice = c.unitPrice.toString
 120     val startTime = c.startMillis.toString
 121     val endTime   = c.stopMillis.toString
 122     val difTime   = (c.stopMillis - c.startMillis).toString
 123     val credits   = c.creditsToSubtract.toString
 124     new ChargeEntry(counter.getAndIncrement.toString,unitPrice,
 125                     startTime,endTime,difTime,credits)
 126   }
 127
 128   private[this] def toEventEntry(eventType:String,c:Chargeslot) : EventEntry =
 129     new EventEntry(eventType,List(toChargeEntry(c)))
 130
 131
 132   private[this] def toResourceEntry(w:WalletEntry) : ResourceEntry = {
 133     assert(w.sumOfCreditsToSubtract==0.0 || w.chargslotCount > 0)
 134     val rcName = w.currentResourceEvent.clientID match {
 135             case "pithos" =>
 136               w.currentResourceEvent.details("path")
 137             case _ =>
 138               w.currentResourceEvent.instanceID
 139         }
 140     val rcType =  w.resourceType.name
 141     val rcUnitName = w.resourceType.unit
 142     val eventEntry = new ListBuffer[EventEntry]
 143     val credits = w.sumOfCreditsToSubtract
 144     val eventType = //TODO: This is hardcoded; find a better solution
 145         w.currentResourceEvent.clientID match {
 146           case "pithos" =>
 147             val action = w.currentResourceEvent.details("action")
 148             val path = w.currentResourceEvent.details("path")
 149             "%s@%s".format(action,path)
 150           case "cyclades" =>
 151             w.currentResourceEvent.value.toInt match {
 152               case 0 => // OFF
 153                   "offOn"
 154               case 1 =>  // ON
 155                  "onOff"
 156               case 2 =>
 157                  "destroy"
 158               case _ =>
 159                  "BUG"
 160             }
 161           case "astakos" =>
 162             "once"
 163         }
 164
 165     for { c <- w.chargeslots }{
 166       if(c.creditsToSubtract != 0.0) {
 167         //Console.err.println("c.creditsToSubtract : " + c.creditsToSubtract)
 168         eventEntry += toEventEntry(eventType.toString,c)
 169         //credits += c.creditsToSubtract
 170       }
 171     }
 172     //Console.err.println("TOTAL resource event credits: " + credits)
 173     new ResourceEntry(rcName,rcType,rcUnitName,credits.toString,eventEntry.toList)
 174   }
 175
 176   private[this] def resourceEntriesAt(t:Timeslot,w:WorkingUserState) : (List[ResourceEntry],Double) = {
 177     val ret = new ListBuffer[ResourceEntry]
 178     var sum = 0.0
 179     //Console.err.println("Wallet entries: " + w.walletEntries)
 180     val walletEntries = w.walletEntries
 181     for { i <- walletEntries} {
 182       if(t.contains(i.referenceTimeslot) && i.sumOfCreditsToSubtract != 0.0){
 183         //Console.err.println("i.sumOfCreditsToSubtract : " + i.sumOfCreditsToSubtract)
 184         if(i.sumOfCreditsToSubtract > 0.0D) sum += i.sumOfCreditsToSubtract
 185         ret += toResourceEntry(i)
 186       } else {
 187         //Console.err.println("WALLET ENTERY : " + i + "\n" +
 188         //             t + "  does not contain " +  i.referenceTimeslot + "  !!!!")
 189       }
 190     }
 191     (ret.toList,sum)
 192   }
 193
 194   private[this] def aggregateResourceEntries(re:List[ResourceEntry]) : List[ResourceEntry] = {
 195     def addResourceEntries(a:ResourceEntry,b:ResourceEntry) : ResourceEntry = {
 196       assert(a.resourceName == b.resourceName)
 197       val totalCredits = (a.totalCredits.toDouble+b.totalCredits.toDouble).toString
 198       a.copy(a.resourceName,a.resourceType,a.unitName,totalCredits,a.details ::: b.details)
 199     }
 200     re.foldLeft(TreeMap[String,ResourceEntry]()){ (map,r1) =>
 201       map.get(r1.resourceName) match {
 202         case None => map + ((r1.resourceName,r1))
 203         case Some(r0) => (map - r0.resourceName) +
 204                          ((r0.resourceName, addResourceEntries(r0,r1)))
 205       }
 206     }.values.toList
 207   }
 208
 209   def fromWorkingUserState(t:Timeslot,userID:String,w:Option[WorkingUserState]) : AbstractBillEntry = {
 210     val ret = w match {
 211       case None =>
 212           new BillEntry(counter.getAndIncrement.toString,
 213                         userID,"processing",
 214                         "0.0",
 215                         "0.0",
 216                         t.from.getTime.toString,t.to.getTime.toString,
 217                         Nil)
 218       case Some(w) =>
 219         val (rcEntries,rcEntriesCredits) = resourceEntriesAt(t,w)
 220         val resMap = aggregateResourceEntries(rcEntries)
 221         new BillEntry(counter.getAndIncrement.toString,
 222                       userID,"ok",
 223                       w.totalCredits.toString,
 224                       rcEntriesCredits.toString,
 225                       t.from.getTime.toString,t.to.getTime.toString,
 226                       resMap)
 227     }
 228     //Console.err.println("JSON: " +  ret.toJsonString)
 229     ret
 230   }
 231
 232   val jsonSample = "{\n  \"id\":\"2\",\n  \"userID\":\"loverdos@grnet.gr\",\n  \"status\":\"ok\",\n  \"remainingCredits\":\"3130.0000027777783\",\n  \"deductedCredits\":\"5739.9999944444435\",\n  \"startTime\":\"1341090000000\",\n  \"endTime\":\"1343768399999\",\n  \"bill\":[{\n    \"resourceName\":\"diskspace\",\n    \"resourceType\":\"diskspace\",\n    \"unitName\":\"MB/Hr\",\n    \"totalCredits\":\"2869.9999972222217\",\n    \"eventType\":\"object update@/Papers/GOTO_HARMFUL.PDF\",\n\t    \"details\":[\n\t     {\"totalCredits\":\"2869.9999972222217\",\n\t      \"details\":[{\n\t      \"id\":\"0\",\n\t      \"unitPrice\":\"0.01\",\n\t      \"startTime\":\"1342735200000\",\n\t      \"endTime\":\"1343768399999\",\n\t      \"ellapsedTime\":\"1033199999\",\n\t      \"credits\":\"2869.9999972222217\"\n\t    \t}]\n\t    }\n\t  ]\n  },{\n    \"resourceName\":\"diskspace\",\n    \"resourceType\":\"diskspace\",\n    \"unitName\":\"MB/Hr\",\n    \"totalCredits\":\"2869.9999972222217\",\n    \"eventType\":\"object update@/Papers/GOTO_HARMFUL.PDF\",\n    \"details\":[\t     {\"totalCredits\":\"2869.9999972222217\",\n\t      \"details\":[{\n\t      \"id\":\"0\",\n\t      \"unitPrice\":\"0.01\",\n\t      \"startTime\":\"1342735200000\",\n\t      \"endTime\":\"1343768399999\",\n\t      \"ellapsedTime\":\"1033199999\",\n\t      \"credits\":\"2869.9999972222217\"\n\t    \t}]\n\t    }\n\t]\n  }]\n}"
 233
 234   def main0(args: Array[String]) = {
 235      val b : BillEntry = StdConverters.AllConverters.convertEx[BillEntry](CompactJsonTextFormat(jsonSample))
 236      val l0 = b.bill
 237      val l1 = aggregateResourceEntries(l0)
 238
 239      Console.err.println("Initial resources: ")
 240      for{ i <- l0 } Console.err.println("RESOURCE: " + i.toJsonString)
 241     Console.err.println("Aggregate resources: ")
 242     for{ a <- l1 } {
 243       Console.err.println("RESOURCE:  %s\n  %s\nEND RESOURCE".format(a.resourceName,a.toJsonString))
 244     }
 245
 246     val aggr = new BillEntry(b.id,b.userID,b.status,b.remainingCredits,b.deductedCredits,b.startTime,b.endTime,l1)
 247     Console.err.println("Aggregate:\n" + aggr.toJsonString)
 248   }
 249
 250   //
 251   def main(args: Array[String]) = {
 252     //Console.err.println("JSON: " +  (new BillEntry).toJsonString)
 253     val propsfile = new FileStreamResource(new File("aquarium.properties"))
 254     var _props: Props = Props(propsfile)(StdConverters.AllConverters).getOr(Props()(StdConverters.AllConverters))
 255     val aquarium = new AquariumBuilder(_props, ResourceLocator.DefaultPolicyModel).
 256       update(Aquarium.EnvKeys.storeProvider, new MemStoreProvider).
 257       update(Aquarium.EnvKeys.eventsStoreFolder,Some(new File(".."))).
 258       build()
 259     aquarium.start()
 260     ()
 261   }
 262 }