X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Fusecases%2Fodusliceassurance%2Fgoversion%2Finternal%2Fstructures%2Fsliceassurance.go;h=d7c25569ae9baef83abe36bda8c4c54f456be4ee;hb=df5eeb6e3fe42f87ac399f624edef20c87d1e475;hp=b73deb214ff9008d7f840d7db9f4be8a0bc0ad30;hpb=534a1fe8136ed8d72615707008bdd6061d1df2fe;p=nonrtric.git

diff --git a/test/usecases/odusliceassurance/goversion/internal/structures/sliceassurance.go b/test/usecases/odusliceassurance/goversion/internal/structures/sliceassurance.go
index b73deb21..d7c25569 100644
--- a/test/usecases/odusliceassurance/goversion/internal/structures/sliceassurance.go
+++ b/test/usecases/odusliceassurance/goversion/internal/structures/sliceassurance.go
@@ -20,38 +20,125 @@
 
 package structures
 
+import (
+	"fmt"
+	"regexp"
+	"strconv"
+
+	log "github.com/sirupsen/logrus"
+	"oransc.org/usecase/oduclosedloop/messages"
+)
+
+type MapKey struct {
+	Duid string
+	sd   int
+	sst  int
+}
+
 type SliceAssuranceMeas struct {
-	Metrics  []*SliceMetric
+	Metrics  map[MapKey]*SliceMetric
 	Policies map[string]*PolicyRatio
 }
 
 func NewSliceAssuranceMeas() *SliceAssuranceMeas {
 	s := SliceAssuranceMeas{}
-	s.Metrics = make([]*SliceMetric, 0)
+	s.Metrics = make(map[MapKey]*SliceMetric)
 	s.Policies = make(map[string]*PolicyRatio)
 	return &s
 }
 
-func (sa *SliceAssuranceMeas) AddNewPolicy(pr *PolicyRatio) {
-	sa.Policies[pr.PolicyRatioId] = pr
+func (sa *SliceAssuranceMeas) AddNewPolicy(duid string, rrmPolicyRatio messages.RRMPolicyRatio) {
+	for _, policyMember := range rrmPolicyRatio.RRMPolicyMembers {
+		metric := sa.GetSliceMetric(duid, policyMember.SliceDifferentiator, policyMember.SliceServiceType)
+		if metric != nil {
+			pr := NewPolicyRatio(rrmPolicyRatio.Id, rrmPolicyRatio.RRMPolicyMaxRatio, rrmPolicyRatio.RRMPolicyMinRatio, rrmPolicyRatio.RRMPolicyDedicatedRatio)
+			_, check := sa.Policies[pr.PolicyRatioId]
+			if !check {
+				log.Infof(" new policy has been added %+v", *pr)
+			}
+			sa.Policies[pr.PolicyRatioId] = pr
+			metric.RRMPolicyRatioId = rrmPolicyRatio.Id
+
+		}
+	}
 }
 
 func (sa *SliceAssuranceMeas) GetSliceMetric(duid string, sd int, sst int) *SliceMetric {
-	for _, metric := range sa.Metrics {
-		if metric.DUId == duid && metric.SliceDiff == sd && metric.SliceServiceType == sst {
-			return metric
-		}
+	key := MapKey{duid, sd, sst}
+	value, check := sa.Metrics[key]
+
+	if check {
+		return value
 	}
+
 	return nil
 }
 
-func (sa *SliceAssuranceMeas) AddOrUpdateMetric(sm *SliceMetric) {
-	metric := sa.GetSliceMetric(sm.DUId, sm.SliceDiff, sm.SliceServiceType)
-	if metric != nil {
-		for key, value := range sm.PM {
-			metric.PM[key] = value
+func (sa *SliceAssuranceMeas) AddOrUpdateMetric(meas messages.Measurement) (string, error) {
+
+	var duid string
+	var sd, sst int
+
+	regex := *regexp.MustCompile(`\/(.*)network-function\/distributed-unit-functions\[id=\'(.*)\'\]\/cell\[id=\'(.*)\'\]\/supported-measurements\[performance-measurement-type=\'(.*)\'\]\/supported-snssai-subcounter-instances\[slice-differentiator=\'(\d+)\'\]\[slice-service-type=\'(\d+)\'\]`)
+	res := regex.FindAllStringSubmatch(meas.MeasurementTypeInstanceReference, -1)
+
+	if res != nil && len(res[0]) == 7 {
+		duid = res[0][2]
+		sd = toInt(res[0][5])
+		sst = toInt(res[0][6])
+
+		key := MapKey{duid, sd, sst}
+		value, check := sa.Metrics[key]
+
+		if check {
+			sa.updateMetric(key, value, res[0][4], meas.Value)
+		} else {
+			// Only add new one if value exceeds threshold
+			sa.addMetric(res, meas.Value)
 		}
 	} else {
-		sa.Metrics = append(sa.Metrics, sm)
+		return duid, fmt.Errorf(" wrong format for MeasurementTypeInstanceReference")
+	}
+	return duid, nil
+}
+
+func (sa *SliceAssuranceMeas) addMetric(res [][]string, metricValue int) {
+	if metricValue > 700 {
+		metric := NewSliceMetric(res[0][2], res[0][3], toInt(res[0][5]), toInt(res[0][6]))
+		metric.PM[res[0][3]] = metricValue
+		key := MapKey{res[0][2], toInt(res[0][5]), toInt(res[0][6])}
+		sa.Metrics[key] = metric
+		log.Infof(" new metric has been added %+v", *metric)
+	}
+}
+
+func (sa *SliceAssuranceMeas) updateMetric(key MapKey, value *SliceMetric, metricName string, metricValue int) {
+	if metricValue < 700 {
+		delete(sa.Metrics, key)
+		log.Infof(" metric with key %+v has been deleted", key)
+	} else {
+		value.PM[metricName] = metricValue
+		log.Infof(" metric value has been updated, new value: %v", metricValue)
+	}
+}
+
+func toInt(num string) int {
+	res, err := strconv.Atoi(num)
+	if err != nil {
+		return -1
+	}
+	return res
+}
+
+func (sa *SliceAssuranceMeas) PrintStructures() {
+	fmt.Printf("SliceAssurance Metrics: \n")
+	for key, metric := range sa.Metrics {
+		fmt.Printf("Key: %+v\n", key)
+		fmt.Printf("Metric: %+v\n", metric)
+	}
+	fmt.Printf("SliceAssurance Policies: \n")
+	for key, metric := range sa.Policies {
+		fmt.Printf("Key: %+v\n", key)
+		fmt.Printf("Metric: %+v\n", metric)
 	}
 }