From f24baa15b417bf23739bb749762fbca280c4cde2 Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Tue, 24 Nov 2020 15:58:15 +0100
Subject: [PATCH] RateS - RateProfileCost with CorrectCost method

---
 engine/rateprofile.go | 39 +++++++++++++++++++++++++++++++++++----
 1 file changed, 35 insertions(+), 4 deletions(-)

diff --git a/engine/rateprofile.go b/engine/rateprofile.go
index cb6a4a038..f1f7045ba 100644
--- a/engine/rateprofile.go
+++ b/engine/rateprofile.go
@@ -36,8 +36,8 @@ type RateProfile struct {
 	FilterIDs          []string
 	ActivationInterval *utils.ActivationInterval
 	Weight             float64
-	RoundingMethod     string
 	RoundingDecimals   int
+	RoundingMethod     string
 	MinCost            float64
 	MaxCost            float64
 	MaxCostStrategy    string
@@ -180,10 +180,41 @@ type RateSIncrement struct {
 	cost *decimal.Big // unexported total increment cost
 }
 
+// RateProfileCost is the cost returned by RateS at cost queries
 type RateProfileCost struct {
-	ID             string // RateProfileID
-	Cost           float64
-	RateSIntervals []*RateSInterval
+	ID               string // RateProfileID
+	Cost             float64
+	RoundingDecimals int
+	RoundingMethod   string
+	MinCost          float64
+	MaxCost          float64
+	MaxCostStrategy  string
+	RateSIntervals   []*RateSInterval
+	Altered          []string
+}
+
+// CorrectCost should be called in final phase of cost calculation
+// in order to apply further correction like Min/MaxCost or rounding
+func (rPc *RateProfileCost) CorrectCost(rndDec *int, rndMtd string) {
+	if rndDec != nil {
+		rPc.RoundingDecimals = *rndDec
+		if rndMtd != utils.EmptyString {
+			rPc.RoundingMethod = rndMtd
+		}
+
+	}
+	if rPc.Cost < rPc.MinCost {
+		rPc.Cost = rPc.MinCost
+		rPc.Altered = append(rPc.Altered, utils.MinCost)
+	}
+	if rPc.Cost > rPc.MaxCost {
+		rPc.Cost = rPc.MaxCost
+		rPc.Altered = append(rPc.Altered, utils.MaxCost)
+	}
+	if rPc.RoundingDecimals != 0 {
+		rPc.Cost = utils.Round(rPc.Cost, rPc.RoundingDecimals, rPc.RoundingMethod)
+		rPc.Altered = append(rPc.Altered, utils.RoundingDecimals)
+	}
 }
 
 // Sort will sort the IntervalRates from each Rate based on IntervalStart