/*
- * Copyright 2011 GRNET S.A. All rights reserved.
+ * Copyright 2011-2012 GRNET S.A. All rights reserved.
*
* Redistribution and use in source and binary forms, with or
* without modification, are permitted provided that the following
package gr.grnet.aquarium.user
-import com.ckkloverdos.maybe.{Failed, NoVal, Just, Maybe}
-import gr.grnet.aquarium.logic.accounting.dsl.{DSLResourcesMap, DSLPolicy}
-import gr.grnet.aquarium.store.{PolicyStore, UserStateStore, ResourceEventStore}
+import scala.collection.mutable
import gr.grnet.aquarium.util.{ContextualLogger, Loggable}
-import gr.grnet.aquarium.logic.accounting.Accounting
-import gr.grnet.aquarium.logic.accounting.algorithm.SimpleCostPolicyAlgorithmCompiler
-import gr.grnet.aquarium.logic.events.{NewWalletEntry, ResourceEvent}
import gr.grnet.aquarium.util.date.{TimeHelpers, MutableDateCalc}
+import gr.grnet.aquarium.logic.accounting.dsl.{DSLAgreement, DSLResourcesMap}
+import gr.grnet.aquarium.store.{StoreProvider, PolicyStore}
+import gr.grnet.aquarium.logic.accounting.Accounting
+import gr.grnet.aquarium.logic.accounting.algorithm.CostPolicyAlgorithmCompiler
+import gr.grnet.aquarium.event.NewWalletEntry
+import gr.grnet.aquarium.event.resource.ResourceEventModel
+import gr.grnet.aquarium.event.im.{IMEventModel, StdIMEvent}
+import gr.grnet.aquarium.{AquariumInternalError, AquariumException}
/**
*
* @author Christos KK Loverdos <loverdos@gmail.com>
*/
class UserStateComputations extends Loggable {
- def createFirstUserState(userId: String, agreementName: String = "default") = {
- val now = 0L
+ def createInitialUserState(imEvent: IMEventModel, credits: Double, agreementName: String) = {
+ if(!imEvent.isCreateUser) {
+ throw new AquariumInternalError(
+ "Got '%s' instead of '%s'".format(imEvent.eventType, IMEventModel.EventTypeNames.create))
+ }
+
+ val userID = imEvent.userID
+ val userCreationMillis = imEvent.occurredMillis
+ val now = TimeHelpers.nowMillis()
+
UserState(
- userId,
- now,
+ true,
+ userID,
+ userCreationMillis,
0L,
false,
null,
ImplicitlyIssuedResourceEventsSnapshot(List(), now),
- Nil, Nil,
+ Nil,
+ Nil,
LatestResourceEventsSnapshot(List(), now),
- 0L, 0L,
- ActiveStateSnapshot(false, now),
- CreditSnapshot(0, now),
- AgreementSnapshot(Agreement(agreementName, now) :: Nil, now),
- RolesSnapshot(List(), now),
- OwnedResourcesSnapshot(List(), now)
+ 0L,
+ 0L,
+ IMStateSnapshot(imEvent, now),
+ CreditSnapshot(credits, now),
+ AgreementSnapshot(List(Agreement(agreementName, userCreationMillis)), now),
+ OwnedResourcesSnapshot(Nil, now),
+ Nil,
+ InitialUserStateSetup
)
}
- def createFirstUserState(userId: String, agreementName: String, resourcesMap: DSLResourcesMap) = {
- val now = 0L
- UserState(
- userId,
- now,
- 0L,
- false,
- null,
- ImplicitlyIssuedResourceEventsSnapshot(List(), now),
- Nil, Nil,
- LatestResourceEventsSnapshot(List(), now),
- 0L, 0L,
- ActiveStateSnapshot(false, now),
- CreditSnapshot(0, now),
- AgreementSnapshot(Agreement(agreementName, now) :: Nil, now),
- RolesSnapshot(List(), now),
- OwnedResourcesSnapshot(List(), now)
- )
- }
+ def createInitialUserState(userID: String,
+ userCreationMillis: Long,
+ isActive: Boolean,
+ credits: Double,
+ roleNames: List[String] = List(),
+ agreementName: String = DSLAgreement.DefaultAgreementName) = {
+ val now = userCreationMillis
+
+ UserState(
+ true,
+ userID,
+ userCreationMillis,
+ 0L,
+ false,
+ null,
+ ImplicitlyIssuedResourceEventsSnapshot(List(), now),
+ Nil,
+ Nil,
+ LatestResourceEventsSnapshot(List(), now),
+ 0L,
+ 0L,
+ IMStateSnapshot(
+ StdIMEvent(
+ "",
+ now, now, userID,
+ "",
+ isActive, roleNames.headOption.getOrElse("default"),
+ "1.0",
+ IMEventModel.EventTypeNames.create, Map()),
+ now
+ ),
+ CreditSnapshot(credits, now),
+ AgreementSnapshot(List(Agreement(agreementName, userCreationMillis)), now),
+ OwnedResourcesSnapshot(Nil, now),
+ Nil,
+ InitialUserStateSetup
+ )
+ }
+
+ def createInitialUserStateFrom(us: UserState): UserState = {
+ createInitialUserState(
+ us.imStateSnapshot.imEvent,
+ us.creditsSnapshot.creditAmount,
+ us.agreementsSnapshot.agreementsByTimeslot.valuesIterator.toList.last)
+ }
def findUserStateAtEndOfBillingMonth(userId: String,
billingMonthInfo: BillingMonthInfo,
- userStateStore: UserStateStore,
- resourceEventStore: ResourceEventStore,
- policyStore: PolicyStore,
- userCreationMillis: Long,
+ storeProvider: StoreProvider,
currentUserState: UserState,
- zeroUserState: UserState,
- defaultPolicy: DSLPolicy,
defaultResourcesMap: DSLResourcesMap,
accounting: Accounting,
- contextualLogger: Maybe[ContextualLogger] = NoVal): Maybe[UserState] = {
+ algorithmCompiler: CostPolicyAlgorithmCompiler,
+ calculationReason: UserStateChangeReason,
+ clogOpt: Option[ContextualLogger] = None): UserState = {
val clog = ContextualLogger.fromOther(
- contextualLogger,
+ clogOpt,
logger,
"findUserStateAtEndOfBillingMonth(%s)", billingMonthInfo)
clog.begin()
- def doCompute: Maybe[UserState] = {
- clog.debug("Computing full month billing")
+ def doCompute: UserState = {
doFullMonthlyBilling(
userId,
billingMonthInfo,
- userStateStore,
- resourceEventStore,
- policyStore,
- userCreationMillis,
+ storeProvider,
currentUserState,
- zeroUserState,
- defaultPolicy,
defaultResourcesMap,
accounting,
- Just(clog))
+ algorithmCompiler,
+ calculationReason,
+ Some(clog))
}
+ val userStateStore = storeProvider.userStateStore
+ val resourceEventStore = storeProvider.resourceEventStore
+
+ val userCreationMillis = currentUserState.userCreationMillis
val userCreationDateCalc = new MutableDateCalc(userCreationMillis)
val billingMonthStartMillis = billingMonthInfo.startMillis
val billingMonthStopMillis = billingMonthInfo.stopMillis
if(billingMonthStopMillis < userCreationMillis) {
// If the user did not exist for this billing month, piece of cake
- clog.debug("User did not exist before %s. Returning %s", userCreationDateCalc, zeroUserState)
- clog.endWith(Just(zeroUserState))
+ clog.debug("User did not exist before %s", userCreationDateCalc)
+
+ // NOTE: Reason here will be: InitialUserStateSetup$
+ val initialUserState0 = createInitialUserStateFrom(currentUserState)
+ val initialUserState1 = userStateStore.insertUserState(initialUserState0)
+
+ clog.debug("Returning INITIAL state [_id=%s] %s".format(initialUserState1._id, initialUserState1))
+ clog.end()
+
+ initialUserState1
} else {
// Ask DB cache for the latest known user state for this billing period
- val latestUserStateM = userStateStore.findLatestUserStateForEndOfBillingMonth(
+ val latestUserStateOpt = userStateStore.findLatestUserStateForEndOfBillingMonth(
userId,
billingMonthInfo.year,
billingMonthInfo.month)
- latestUserStateM match {
- case NoVal ⇒
+ latestUserStateOpt match {
+ case None ⇒
// Not found, must compute
clog.debug("No user state found from cache, will have to (re)compute")
- clog.endWith(doCompute)
-
- case failed @ Failed(_, _) ⇒
- clog.warn("Failure while quering cache for user state: %s", failed)
- clog.endWith(failed)
+ val result = doCompute
+ clog.end()
+ result
- case Just(latestUserState) ⇒
+ case Some(latestUserState) ⇒
// Found a "latest" user state but need to see if it is indeed the true and one latest.
// For this reason, we must count the events again.
val latestStateOOSEventsCounter = latestUserState.billingPeriodOutOfSyncResourceEventsCounter
- val actualOOSEventsCounterM = resourceEventStore.countOutOfSyncEventsForBillingPeriod(
+ val actualOOSEventsCounter = resourceEventStore.countOutOfSyncEventsForBillingPeriod(
userId,
billingMonthStartMillis,
billingMonthStopMillis)
- actualOOSEventsCounterM match {
- case NoVal ⇒
- val errMsg = "No counter computed for out of sync events. Should at least be zero."
+ val counterDiff = actualOOSEventsCounter - latestStateOOSEventsCounter
+ counterDiff match {
+ // ZERO, we are OK!
+ case 0 ⇒
+ // NOTE: Keep the caller's calculation reason
+ latestUserState.copyForChangeReason(calculationReason)
+
+ // We had more, so must recompute
+ case n if n > 0 ⇒
+ clog.debug(
+ "Found %s out of sync events (%s more), will have to (re)compute user state", actualOOSEventsCounter, n)
+ val result = doCompute
+ clog.end()
+ result
+
+ // We had less????
+ case n if n < 0 ⇒
+ val errMsg = "Found %s out of sync events (%s less). DB must be inconsistent".format(actualOOSEventsCounter, n)
clog.warn(errMsg)
- clog.endWith(Failed(new Exception(errMsg)))
-
- case failed @ Failed(_, _) ⇒
- clog.warn("Failure while querying for out of sync events: %s", failed)
- clog.endWith(failed)
-
- case Just(actualOOSEventsCounter) ⇒
- val counterDiff = actualOOSEventsCounter - latestStateOOSEventsCounter
- counterDiff match {
- // ZERO, we are OK!
- case 0 ⇒
- latestUserStateM
-
- // We had more, so must recompute
- case n if n > 0 ⇒
- clog.debug(
- "Found %s out of sync events (%s more), will have to (re)compute user state", actualOOSEventsCounter, n)
- clog.endWith(doCompute)
-
- // We had less????
- case n if n < 0 ⇒
- val errMsg = "Found %s out of sync events (%s less). DB must be inconsistent".format(actualOOSEventsCounter, n)
- clog.warn(errMsg)
- clog.endWith(Failed(new Exception(errMsg)))
- }
+ throw new AquariumException(errMsg)
}
}
}
}
+ //+ Utility methods
+ def rcDebugInfo(rcEvent: ResourceEventModel) = {
+ rcEvent.toDebugString(false)
+ }
+ //- Utility methods
+
+ def processResourceEvent(startingUserState: UserState,
+ userStateWorker: UserStateWorker,
+ currentResourceEvent: ResourceEventModel,
+ policyStore: PolicyStore,
+ stateChangeReason: UserStateChangeReason,
+ billingMonthInfo: BillingMonthInfo,
+ walletEntriesBuffer: mutable.Buffer[NewWalletEntry],
+ algorithmCompiler: CostPolicyAlgorithmCompiler,
+ clogOpt: Option[ContextualLogger] = None): UserState = {
+
+ val clog = ContextualLogger.fromOther(clogOpt, logger, "walletEntriesForResourceEvent(%s)", currentResourceEvent.id)
+
+ var _workingUserState = startingUserState
+
+ val theResource = currentResourceEvent.safeResource
+ val theInstanceId = currentResourceEvent.safeInstanceId
+ val theValue = currentResourceEvent.value
+
+ val accounting = userStateWorker.accounting
+ val resourcesMap = userStateWorker.resourcesMap
+
+ val currentResourceEventDebugInfo = rcDebugInfo(currentResourceEvent)
+ clog.begin(currentResourceEventDebugInfo)
+
+ userStateWorker.debugTheMaps(clog)(rcDebugInfo)
+
+ // Ignore the event if it is not billable (but still record it in the "previous" stuff).
+ // But to make this decision, first we need the resource definition (and its cost policy).
+ val dslResourceOpt = resourcesMap.findResource(theResource)
+ dslResourceOpt match {
+ // We have a resource (and thus a cost policy)
+ case Some(dslResource) ⇒
+ val costPolicy = dslResource.costPolicy
+ clog.debug("Cost policy %s for %s", costPolicy, dslResource)
+ val isBillable = costPolicy.isBillableEventBasedOnValue(theValue)
+ if(!isBillable) {
+ // The resource event is not billable
+ clog.debug("Ignoring not billable event %s", currentResourceEventDebugInfo)
+ } else {
+ // The resource event is billable
+ // Find the previous event.
+ // This is (potentially) needed to calculate new credit amount and new resource instance amount
+ val previousResourceEventOpt = userStateWorker.findAndRemovePreviousResourceEvent(theResource, theInstanceId)
+ clog.debug("PreviousM %s", previousResourceEventOpt.map(rcDebugInfo(_)))
+
+ val havePreviousResourceEvent = previousResourceEventOpt.isDefined
+ val needPreviousResourceEvent = costPolicy.needsPreviousEventForCreditAndAmountCalculation
+ if(needPreviousResourceEvent && !havePreviousResourceEvent) {
+ // This must be the first resource event of its kind, ever.
+ // TODO: We should normally check the DB to verify the claim (?)
+ clog.debug("Ignoring first event of its kind %s", currentResourceEventDebugInfo)
+ userStateWorker.updateIgnored(currentResourceEvent)
+ } else {
+ val defaultInitialAmount = costPolicy.getResourceInstanceInitialAmount
+ val oldAmount = _workingUserState.getResourceInstanceAmount(theResource, theInstanceId, defaultInitialAmount)
+ val oldCredits = _workingUserState.creditsSnapshot.creditAmount
+
+ // A. Compute new resource instance accumulating amount
+ val newAmount = costPolicy.computeNewAccumulatingAmount(oldAmount, theValue)
+
+ clog.debug("theValue = %s, oldAmount = %s, newAmount = %s, oldCredits = %s", theValue, oldAmount, newAmount, oldCredits)
+
+ // B. Compute new wallet entries
+ clog.debug("agreementsSnapshot = %s", _workingUserState.agreementsSnapshot)
+ val alltimeAgreements = _workingUserState.agreementsSnapshot.agreementsByTimeslot
+
+ // clog.debug("Computing full chargeslots")
+ val (referenceTimeslot, fullChargeslots) = accounting.computeFullChargeslots(
+ previousResourceEventOpt,
+ currentResourceEvent,
+ oldCredits,
+ oldAmount,
+ newAmount,
+ dslResource,
+ resourcesMap,
+ alltimeAgreements,
+ algorithmCompiler,
+ policyStore,
+ Some(clog)
+ )
+
+ // We have the chargeslots, let's associate them with the current event
+ if(fullChargeslots.length == 0) {
+ // At least one chargeslot is required.
+ throw new AquariumInternalError("No chargeslots computed for resource event %s".format(currentResourceEvent.id))
+ }
+ clog.debugSeq("fullChargeslots", fullChargeslots, 0)
+
+ // C. Compute new credit amount (based on the charge slots)
+ val newCreditsDiff = fullChargeslots.map(_.computedCredits.get).sum
+ val newCredits = oldCredits - newCreditsDiff
+
+ if(stateChangeReason.shouldStoreCalculatedWalletEntries) {
+ val newWalletEntry = NewWalletEntry(
+ userStateWorker.userId,
+ newCreditsDiff,
+ oldCredits,
+ newCredits,
+ TimeHelpers.nowMillis(),
+ referenceTimeslot,
+ billingMonthInfo.year,
+ billingMonthInfo.month,
+ if(havePreviousResourceEvent)
+ List(currentResourceEvent, previousResourceEventOpt.get)
+ else
+ List(currentResourceEvent),
+ fullChargeslots,
+ dslResource,
+ currentResourceEvent.isSynthetic
+ )
+ clog.debug("New %s", newWalletEntry)
+
+ walletEntriesBuffer += newWalletEntry
+ } else {
+ clog.debug("newCreditsDiff = %s, newCredits = %s", newCreditsDiff, newCredits)
+ }
+
+ _workingUserState = _workingUserState.copy(
+ creditsSnapshot = CreditSnapshot(newCredits, TimeHelpers.nowMillis()),
+ stateChangeCounter = _workingUserState.stateChangeCounter + 1,
+ totalEventsProcessedCounter = _workingUserState.totalEventsProcessedCounter + 1
+ )
+ }
+ }
+
+ // After processing, all events billable or not update the previous state
+ userStateWorker.updatePrevious(currentResourceEvent)
+
+ _workingUserState = _workingUserState.copy(
+ latestResourceEventsSnapshot = userStateWorker.previousResourceEvents.toImmutableSnapshot(TimeHelpers.nowMillis())
+ )
+
+ // We do not have a resource (and thus, no cost policy)
+ case None ⇒
+ // Now, this is a matter of politics: what do we do if no policy was found?
+ clog.warn("Unknown resource for %s", currentResourceEventDebugInfo)
+ } // dslResourceOpt match
+
+ clog.end(currentResourceEventDebugInfo)
+
+ _workingUserState
+ }
+
+ def processResourceEvents(resourceEvents: Traversable[ResourceEventModel],
+ startingUserState: UserState,
+ userStateWorker: UserStateWorker,
+ policyStore: PolicyStore,
+ stateChangeReason: UserStateChangeReason,
+ billingMonthInfo: BillingMonthInfo,
+ walletEntriesBuffer: mutable.Buffer[NewWalletEntry],
+ algorithmCompiler: CostPolicyAlgorithmCompiler,
+ clogOpt: Option[ContextualLogger] = None): UserState = {
+
+ var _workingUserState = startingUserState
+
+ for(currentResourceEvent ← resourceEvents) {
+
+ _workingUserState = processResourceEvent(
+ _workingUserState,
+ userStateWorker,
+ currentResourceEvent,
+ policyStore,
+ stateChangeReason,
+ billingMonthInfo,
+ walletEntriesBuffer,
+ algorithmCompiler,
+ clogOpt
+ )
+ }
+
+ _workingUserState
+ }
+
+
def doFullMonthlyBilling(userId: String,
billingMonthInfo: BillingMonthInfo,
- userStateStore: UserStateStore,
- resourceEventStore: ResourceEventStore,
- policyStore: PolicyStore,
- userCreationMillis: Long,
+ storeProvider: StoreProvider,
currentUserState: UserState,
- zeroUserState: UserState,
- defaultPolicy: DSLPolicy,
defaultResourcesMap: DSLResourcesMap,
accounting: Accounting,
- contextualLogger: Maybe[ContextualLogger] = NoVal): Maybe[UserState] = Maybe {
+ algorithmCompiler: CostPolicyAlgorithmCompiler,
+ calculationReason: UserStateChangeReason = NoSpecificChangeReason,
+ clogOpt: Option[ContextualLogger] = None): UserState = {
+
val clog = ContextualLogger.fromOther(
- contextualLogger,
+ clogOpt,
logger,
"doFullMonthlyBilling(%s)", billingMonthInfo)
clog.begin()
- val previousBillingMonthUserStateM = findUserStateAtEndOfBillingMonth(
+ val clogSome = Some(clog)
+
+ val previousBillingMonthUserState = findUserStateAtEndOfBillingMonth(
userId,
billingMonthInfo.previousMonth,
- userStateStore,
- resourceEventStore,
- policyStore,
- userCreationMillis,
+ storeProvider,
currentUserState,
- zeroUserState,
- defaultPolicy,
defaultResourcesMap,
accounting,
- Just(clog)
+ algorithmCompiler,
+ calculationReason.forPreviousBillingMonth,
+ clogSome
)
-
- previousBillingMonthUserStateM match {
- case NoVal ⇒
- null // not really... (must throw an exception here probably...)
- case failed @ Failed(e, _) ⇒
- throw e
- case Just(startingUserState) ⇒
- // This is the real deal
-
- // This is a collection of all the latest resource events.
- // We want these in order to correlate incoming resource events with their previous (in `occurredMillis` time)
- // ones.
- // Will be updated on processing the next resource event.
- val previousResourceEvents = startingUserState.latestResourceEventsSnapshot.toMutableWorker
- clog.debug("previousResourceEvents = %s", previousResourceEvents)
-
- val billingMonthStartMillis = billingMonthInfo.startMillis
- val billingMonthEndMillis = billingMonthInfo.stopMillis
-
- // Keep the working (current) user state. This will get updated as we proceed with billing for the month
- // specified in the parameters.
- var _workingUserState = startingUserState
-
- // Prepare the implicit OFF resource events
- // (we keep the terminology for historical reasons)
- val theImplicitOFFs = _workingUserState.implicitlyTerminatedSnapshot.toMutableWorker
- clog.debug("theImplicitOFFs = %s", theImplicitOFFs)
-
- /**
- * Finds the previous resource event by checking two possible sources: a) The implicit OFF resource events and
- * b) the explicit previous resource events. If the event is found, it is removed from the respective source.
- *
- * If the event is not found, then this must be for a new resource instance.
- * (and probably then some `zero` resource event must be implied as the previous one)
- *
- * @param resource
- * @param instanceId
- * @return
- */
- def findAndRemovePreviousResourceEvent(resource: String, instanceId: String): Maybe[ResourceEvent] = {
- // implicit OFFs are checked first
- theImplicitOFFs.findAndRemoveResourceEvent(resource, instanceId) match {
- case just @ Just(_) ⇒
- just
- case NoVal ⇒
- // explicit previous are checked second
- previousResourceEvents.findAndRemoveResourceEvent(resource, instanceId) match {
- case just @ Just(_) ⇒
- just
- case noValOrFailed ⇒
- noValOrFailed
- }
- case failed ⇒
- failed
- }
- }
- def rcDebugInfo(rcEvent: ResourceEvent) = {
- rcEvent.toDebugString(defaultResourcesMap, false)
- }
+ val startingUserState = previousBillingMonthUserState
- // Find the actual resource events from DB
- val allResourceEventsForMonth = resourceEventStore.findAllRelevantResourceEventsForBillingPeriod(
- userId,
- billingMonthStartMillis,
- billingMonthEndMillis)
- var _eventCounter = 0
-
- clog.debug("resourceEventStore = %s".format(resourceEventStore))
- clog.debug("Found %s resource events, starting processing...", allResourceEventsForMonth.size)
-
- for {
- currentResourceEvent <- allResourceEventsForMonth
- } {
- _eventCounter = _eventCounter + 1
- val theResource = currentResourceEvent.safeResource
- val theInstanceId = currentResourceEvent.safeInstanceId
- val theValue = currentResourceEvent.value
-
- clog.indent()
- clog.debug("Processing %s", currentResourceEvent)
- clog.debug("Friendlier %s", rcDebugInfo(currentResourceEvent))
- clog.indent()
-
- if(previousResourceEvents.size > 0) {
- clog.debug("%s previousResourceEvents", previousResourceEvents.size)
- clog.indent()
- previousResourceEvents.foreach(ev ⇒ clog.debug("%s", rcDebugInfo(ev)))
- clog.unindent()
- }
- if(theImplicitOFFs.size > 0) {
- clog.debug("%s theImplicitOFFs", theImplicitOFFs.size)
- clog.indent()
- theImplicitOFFs.foreach(ev ⇒ clog.debug("%s", rcDebugInfo(ev)))
- clog.unindent()
- }
+ val userStateStore = storeProvider.userStateStore
+ val resourceEventStore = storeProvider.resourceEventStore
+ val policyStore = storeProvider.policyStore
- // Ignore the event if it is not billable (but still record it in the "previous" stuff).
- // But to make this decision, first we need the resource definition (and its cost policy).
- val resourceDefM = defaultResourcesMap.findResourceM(theResource)
- resourceDefM match {
- // We have a resource (and thus a cost policy)
- case Just(resourceDef) ⇒
- val costPolicy = resourceDef.costPolicy
- clog.debug("Cost policy: %s", costPolicy)
- val isBillable = costPolicy.isBillableEventBasedOnValue(theValue)
- isBillable match {
- // The resource event is not billable
- case false ⇒
- clog.debug("Ignoring not billable event %s", rcDebugInfo(currentResourceEvent))
-
- // The resource event is billable
- case true ⇒
- // Find the previous event.
- // This is (potentially) needed to calculate new credit amount and new resource instance amount
- val previousResourceEventM = findAndRemovePreviousResourceEvent(theResource, theInstanceId)
- clog.debug("PreviousM %s", previousResourceEventM.map(rcDebugInfo(_)))
- val defaultInitialAmount = costPolicy.getResourceInstanceInitialAmount
- val oldAmount = _workingUserState.getResourceInstanceAmount(theResource, theInstanceId, defaultInitialAmount)
- val oldCredits = _workingUserState.creditsSnapshot.creditAmount
-
- // A. Compute new resource instance accumulating amount
- val newAmount = costPolicy.computeNewAccumulatingAmount(oldAmount, theValue)
-
- // B. Compute new wallet entries
- val alltimeAgreements = _workingUserState.agreementsSnapshot.agreementsByTimeslot
-
- val fullChargeslotsM = accounting.computeFullChargeslots(
- previousResourceEventM,
- currentResourceEvent,
- oldCredits,
- oldAmount,
- newAmount,
- resourceDef,
- defaultResourcesMap,
- alltimeAgreements,
- SimpleCostPolicyAlgorithmCompiler
- )
-
- // We have the charges lots, let's associate them with the current event
- fullChargeslotsM match {
- case Just(fullChargeslots) ⇒
- // C. Compute new credit amount (based on the charge slots)
- val newCreditsDiff = fullChargeslots.map(_.computedCredits.get).sum
- val newCredits = oldCredits + newCreditsDiff
- val newWalletEntry = NewWalletEntry(
- userId,
- newCreditsDiff,
- TimeHelpers.nowMillis,
- billingMonthInfo.year,
- billingMonthInfo.month,
- currentResourceEvent,
- previousResourceEventM.toOption,
- fullChargeslots,
- resourceDef
- )
-
- case NoVal ⇒
- // At least one chargeslot is required.
- throw new Exception("No chargeslots computed")
-
- case failed @ Failed(e, m) ⇒
- throw new Exception(m, e)
- }
- }
-
- // After processing, all event, billable or not update the previous state
- previousResourceEvents.updateResourceEvent(currentResourceEvent)
-
- // We do not have a resource (and no cost policy)
- case NoVal ⇒
- // Now, this is a matter of politics: what do we do if no policy was found?
- clog.error("No cost policy for %s", rcDebugInfo(currentResourceEvent))
-
- // Could not retrieve resource (unlikely to happen)
- case failed @ Failed(e, m) ⇒
- clog.error("Error obtaining cost policy for %s", rcDebugInfo(currentResourceEvent))
- clog.error(e, m)
- }
+ val billingMonthStartMillis = billingMonthInfo.startMillis
+ val billingMonthEndMillis = billingMonthInfo.stopMillis
+
+ // Keep the working (current) user state. This will get updated as we proceed with billing for the month
+ // specified in the parameters.
+ // NOTE: The calculation reason is not the one we get from the previous user state but the one our caller specifies
+ var _workingUserState = startingUserState.copyForChangeReason(calculationReason)
+
+ val userStateWorker = UserStateWorker.fromUserState(_workingUserState, accounting, defaultResourcesMap)
+
+ userStateWorker.debugTheMaps(clog)(rcDebugInfo)
+
+ // First, find and process the actual resource events from DB
+ val allResourceEventsForMonth = resourceEventStore.findAllRelevantResourceEventsForBillingPeriod(
+ userId,
+ billingMonthStartMillis,
+ billingMonthEndMillis)
+
+ val newWalletEntries = scala.collection.mutable.ListBuffer[NewWalletEntry]()
+
+ _workingUserState = processResourceEvents(
+ allResourceEventsForMonth,
+ _workingUserState,
+ userStateWorker,
+ policyStore,
+ calculationReason,
+ billingMonthInfo,
+ newWalletEntries,
+ algorithmCompiler,
+ clogSome
+ )
+
+ // Second, for the remaining events which must contribute an implicit OFF, we collect those OFFs
+ // ... in order to generate an implicit ON later
+ val (specialEvents, theirImplicitEnds) = userStateWorker.
+ findAndRemoveGeneratorsOfImplicitEndEvents(billingMonthEndMillis)
+ if(specialEvents.lengthCompare(1) >= 0 || theirImplicitEnds.lengthCompare(1) >= 0) {
+ clog.debug("")
+ clog.debug("Process implicitly issued events")
+ clog.debugSeq("specialEvents", specialEvents, 0)
+ clog.debugSeq("theirImplicitEnds", theirImplicitEnds, 0)
+ }
+
+ // Now, the previous and implicitly started must be our base for the following computation, so we create an
+ // appropriate worker
+ val specialUserStateWorker = UserStateWorker(
+ userStateWorker.userId,
+ LatestResourceEventsWorker.fromList(specialEvents),
+ ImplicitlyIssuedResourceEventsWorker.Empty,
+ IgnoredFirstResourceEventsWorker.Empty,
+ userStateWorker.accounting,
+ userStateWorker.resourcesMap
+ )
+
+ _workingUserState = processResourceEvents(
+ theirImplicitEnds,
+ _workingUserState,
+ specialUserStateWorker,
+ policyStore,
+ calculationReason,
+ billingMonthInfo,
+ newWalletEntries,
+ algorithmCompiler,
+ clogSome
+ )
+
+ val lastUpdateTime = TimeHelpers.nowMillis()
+
+ _workingUserState = _workingUserState.copy(
+ implicitlyIssuedSnapshot = userStateWorker.implicitlyIssuedStartEvents.toImmutableSnapshot(lastUpdateTime),
+ latestResourceEventsSnapshot = userStateWorker.previousResourceEvents.toImmutableSnapshot(lastUpdateTime),
+ stateChangeCounter = _workingUserState.stateChangeCounter + 1,
+ parentUserStateId = startingUserState.idOpt,
+ newWalletEntries = newWalletEntries.toList
+ )
+
+ clog.debug("calculationReason = %s", calculationReason)
+
+ if(calculationReason.shouldStoreUserState) {
+ val storedUserState = userStateStore.insertUserState(_workingUserState)
+ clog.debug("Saved [_id=%s] %s", storedUserState._id, storedUserState)
+ _workingUserState = storedUserState
+ }
+
+ clog.debug("RETURN %s", _workingUserState)
+ clog.end()
+ _workingUserState
+ }
+}
- clog.unindent()
- clog.unindent()
+/**
+ * A helper object holding intermediate state/results during resource event processing.
+ *
+ * @param previousResourceEvents
+ * This is a collection of all the latest resource events.
+ * We want these in order to correlate incoming resource events with their previous (in `occurredMillis` time)
+ * ones. Will be updated on processing the next resource event.
+ *
+ * @param implicitlyIssuedStartEvents
+ * The implicitly issued resource events at the beginning of the billing period.
+ *
+ * @param ignoredFirstResourceEvents
+ * The resource events that were first (and unused) of their kind.
+ *
+ * @author Christos KK Loverdos <loverdos@gmail.com>
+ */
+case class UserStateWorker(userId: String,
+ previousResourceEvents: LatestResourceEventsWorker,
+ implicitlyIssuedStartEvents: ImplicitlyIssuedResourceEventsWorker,
+ ignoredFirstResourceEvents: IgnoredFirstResourceEventsWorker,
+ accounting: Accounting,
+ resourcesMap: DSLResourcesMap) {
+
+ /**
+ * Finds the previous resource event by checking two possible sources: a) The implicitly terminated resource
+ * events and b) the explicit previous resource events. If the event is found, it is removed from the
+ * respective source.
+ *
+ * If the event is not found, then this must be for a new resource instance.
+ * (and probably then some `zero` resource event must be implied as the previous one)
+ *
+ * @param resource
+ * @param instanceId
+ * @return
+ */
+ def findAndRemovePreviousResourceEvent(resource: String, instanceId: String): Option[ResourceEventModel] = {
+ // implicitly issued events are checked first
+ implicitlyIssuedStartEvents.findAndRemoveResourceEvent(resource, instanceId) match {
+ case some @ Some(_) ⇒
+ some
+ case None ⇒
+ // explicit previous resource events are checked second
+ previousResourceEvents.findAndRemoveResourceEvent(resource, instanceId) match {
+ case some @ Some(_) ⇒
+ some
+ case _ ⇒
+ None
}
-
+ }
+ }
+
+ def updateIgnored(resourceEvent: ResourceEventModel): Unit = {
+ ignoredFirstResourceEvents.updateResourceEvent(resourceEvent)
+ }
+
+ def updatePrevious(resourceEvent: ResourceEventModel): Unit = {
+ previousResourceEvents.updateResourceEvent(resourceEvent)
+ }
- clog.endWith(_workingUserState)
+ def debugTheMaps(clog: ContextualLogger)(rcDebugInfo: ResourceEventModel ⇒ String): Unit = {
+ if(previousResourceEvents.size > 0) {
+ val map = previousResourceEvents.latestEventsMap.map { case (k, v) => (k, rcDebugInfo(v)) }
+ clog.debugMap("previousResourceEvents", map, 0)
}
+ if(implicitlyIssuedStartEvents.size > 0) {
+ val map = implicitlyIssuedStartEvents.implicitlyIssuedEventsMap.map { case (k, v) => (k, rcDebugInfo(v)) }
+ clog.debugMap("implicitlyTerminatedResourceEvents", map, 0)
+ }
+ if(ignoredFirstResourceEvents.size > 0) {
+ val map = ignoredFirstResourceEvents.ignoredFirstEventsMap.map { case (k, v) => (k, rcDebugInfo(v)) }
+ clog.debugMap("ignoredFirstResourceEvents", map, 0)
+ }
+ }
+
+// private[this]
+// def allPreviousAndAllImplicitlyStarted: List[ResourceEvent] = {
+// val buffer: FullMutableResourceTypeMap = scala.collection.mutable.Map[FullResourceType, ResourceEvent]()
+//
+// buffer ++= implicitlyIssuedStartEvents.implicitlyIssuedEventsMap
+// buffer ++= previousResourceEvents.latestEventsMap
+//
+// buffer.valuesIterator.toList
+// }
+
+ /**
+ * Find those events from `implicitlyIssuedStartEvents` and `previousResourceEvents` that will generate implicit
+ * end events along with those implicitly issued events. Before returning, remove the events that generated the
+ * implicit ends from the internal state of this instance.
+ *
+ * @see [[gr.grnet.aquarium.logic.accounting.dsl.DSLCostPolicy]]
+ */
+ def findAndRemoveGeneratorsOfImplicitEndEvents(newOccuredMillis: Long
+ ): (List[ResourceEventModel], List[ResourceEventModel]) = {
+ val buffer = mutable.ListBuffer[(ResourceEventModel, ResourceEventModel)]()
+ val checkSet = mutable.Set[ResourceEventModel]()
+
+ def doItFor(map: ResourceEventModel.FullMutableResourceTypeMap): Unit = {
+ val resourceEvents = map.valuesIterator
+ for {
+ resourceEvent ← resourceEvents
+ dslResource ← resourcesMap.findResource(resourceEvent.safeResource)
+ costPolicy = dslResource.costPolicy
+ } {
+ if(costPolicy.supportsImplicitEvents) {
+ if(costPolicy.mustConstructImplicitEndEventFor(resourceEvent)) {
+ val implicitEnd = costPolicy.constructImplicitEndEventFor(resourceEvent, newOccuredMillis)
+
+ if(!checkSet.contains(resourceEvent)) {
+ checkSet.add(resourceEvent)
+ buffer append ((resourceEvent, implicitEnd))
+ }
+
+ // remove it anyway
+ map.remove((resourceEvent.safeResource, resourceEvent.safeInstanceId))
+ }
+ }
+ }
+ }
+
+ doItFor(previousResourceEvents.latestEventsMap) // we give priority for previous
+ doItFor(implicitlyIssuedStartEvents.implicitlyIssuedEventsMap) // ... over implicitly issued...
+
+ (buffer.view.map(_._1).toList, buffer.view.map(_._2).toList)
+ }
+}
+
+object UserStateWorker {
+ def fromUserState(userState: UserState, accounting: Accounting, resourcesMap: DSLResourcesMap): UserStateWorker = {
+ UserStateWorker(
+ userState.userID,
+ userState.latestResourceEventsSnapshot.toMutableWorker,
+ userState.implicitlyIssuedSnapshot.toMutableWorker,
+ IgnoredFirstResourceEventsWorker.Empty,
+ accounting,
+ resourcesMap
+ )
}
}