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 rcType =  w.resourceType.name
 135     val rcName = rcType match {
 136             case "diskspace" =>
 137               w.currentResourceEvent.details("path")
 138             case _ =>
 139               w.currentResourceEvent.instanceID
 140         }
 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         rcType match {
 146           case "diskspace" =>
 147             val action = w.currentResourceEvent.details("action")
 148             val path = w.currentResourceEvent.details("path")
 149             //"%s@%s".format(action,path)
 150             action
 151           case "vmtime" =>
 152             w.currentResourceEvent.value.toInt match {
 153               case 0 => // OFF
 154                   "offOn"
 155               case 1 =>  // ON
 156                  "onOff"
 157               case 2 =>
 158                  "destroy"
 159               case _ =>
 160                  "BUG"
 161             }
 162           case "addcredits" =>
 163             "once"
 164         }
 165
 166     for { c <- w.chargeslots }{
 167       if(c.creditsToSubtract != 0.0) {
 168         //Console.err.println("c.creditsToSubtract : " + c.creditsToSubtract)
 169         eventEntry += toEventEntry(eventType.toString,c)
 170         //credits += c.creditsToSubtract
 171       }
 172     }
 173     //Console.err.println("TOTAL resource event credits: " + credits)
 174     new ResourceEntry(rcName,rcType,rcUnitName,credits.toString,eventEntry.toList)
 175   }
 176
 177   private[this] def resourceEntriesAt(t:Timeslot,w:WorkingUserState) : (List[ResourceEntry],Double) = {
 178     val ret = new ListBuffer[ResourceEntry]
 179     var sum = 0.0
 180     //Console.err.println("Wallet entries: " + w.walletEntries)
 181     val walletEntries = w.walletEntries
 182     /*Console.err.println("Wallet entries ")
 183     for { i <- walletEntries }
 184       Console.err.println("WALLET ENTRY\n%s\nEND WALLET ENTRY".format(i.toJsonString))
 185     Console.err.println("End wallet entries")*/
 186     for { i <- walletEntries} {
 187       if(t.contains(i.referenceTimeslot) && i.sumOfCreditsToSubtract != 0.0){
 188         /*Console.err.println("i.sumOfCreditsToSubtract : " + i.sumOfCreditsToSubtract)*/
 189         if(i.sumOfCreditsToSubtract > 0.0D) sum += i.sumOfCreditsToSubtract
 190         ret += toResourceEntry(i)
 191       } else {
 192         /*Console.err.println("WALLET ENTERY : " + i.toJsonString + "\n" +
 193                      t + "  does not contain " +  i.referenceTimeslot + "  !!!!")*/
 194       }
 195     }
 196     (ret.toList,sum)
 197   }
 198
 199   private[this] def aggregateResourceEntries(re:List[ResourceEntry]) : List[ResourceEntry] = {
 200     def addResourceEntries(a:ResourceEntry,b:ResourceEntry) : ResourceEntry = {
 201       assert(a.resourceName == b.resourceName)
 202       val totalCredits = (a.totalCredits.toDouble+b.totalCredits.toDouble).toString
 203       a.copy(a.resourceName,a.resourceType,a.unitName,totalCredits,a.details ::: b.details)
 204     }
 205     re.foldLeft(TreeMap[String,ResourceEntry]()){ (map,r1) =>
 206       map.get(r1.resourceName) match {
 207         case None => map + ((r1.resourceName,r1))
 208         case Some(r0) => (map - r0.resourceName) +
 209                          ((r0.resourceName, addResourceEntries(r0,r1)))
 210       }
 211     }.values.toList
 212   }
 213
 214   def fromWorkingUserState(t:Timeslot,userID:String,w:Option[WorkingUserState]) : AbstractBillEntry = {
 215     val ret = w match {
 216       case None =>
 217           new BillEntry(counter.getAndIncrement.toString,
 218                         userID,"processing",
 219                         "0.0",
 220                         "0.0",
 221                         t.from.getTime.toString,t.to.getTime.toString,
 222                         Nil)
 223       case Some(w) =>
 224         val (rcEntries,rcEntriesCredits) = resourceEntriesAt(t,w)
 225         val resMap = aggregateResourceEntries(rcEntries)
 226         Console.err.println("Working user state: %s".format(w.toString))
 227         new BillEntry(counter.getAndIncrement.toString,
 228                       userID,"ok",
 229                       w.totalCredits.toString,
 230                       rcEntriesCredits.toString,
 231                       t.from.getTime.toString,t.to.getTime.toString,
 232                       resMap)
 233     }
 234     //Console.err.println("JSON: " +  ret.toJsonString)
 235     ret
 236   }
 237
 238   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}"
 239
 240   def main0(args: Array[String]) = {
 241      val b : BillEntry = StdConverters.AllConverters.convertEx[BillEntry](CompactJsonTextFormat(jsonSample))
 242      val l0 = b.bill
 243      val l1 = aggregateResourceEntries(l0)
 244
 245      Console.err.println("Initial resources: ")
 246      for{ i <- l0 } Console.err.println("RESOURCE: " + i.toJsonString)
 247     Console.err.println("Aggregate resources: ")
 248     for{ a <- l1 } {
 249       Console.err.println("RESOURCE:  %s\n  %s\nEND RESOURCE".format(a.resourceName,a.toJsonString))
 250     }
 251
 252     val aggr = new BillEntry(b.id,b.userID,b.status,b.remainingCredits,b.deductedCredits,b.startTime,b.endTime,l1)
 253     Console.err.println("Aggregate:\n" + aggr.toJsonString)
 254   }
 255
 256   //
 257   def main(args: Array[String]) = {
 258     //Console.err.println("JSON: " +  (new BillEntry).toJsonString)
 259     val propsfile = new FileStreamResource(new File("aquarium.properties"))
 260     var _props: Props = Props(propsfile)(StdConverters.AllConverters).getOr(Props()(StdConverters.AllConverters))
 261     val aquarium = new AquariumBuilder(_props, ResourceLocator.DefaultPolicyModel).
 262       update(Aquarium.EnvKeys.storeProvider, new MemStoreProvider).
 263       update(Aquarium.EnvKeys.eventsStoreFolder,Some(new File(".."))).
 264       build()
 265     aquarium.start()
 266     ()
 267   }
 268 }