X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Frpe%2Frmr.go;h=87c0e2bfe873ffd6ded20e6d7824c20d9b5aa31a;hb=c9a607db8f2dd0b745d30e55c95ef12af9a7bb54;hp=446d7ec4b5aa6ae9af515fcf8f9349e9ba38fe6b;hpb=ab07a4edbce65cc015e025109f51ea32022b2040;p=ric-plt%2Frtmgr.git

diff --git a/pkg/rpe/rmr.go b/pkg/rpe/rmr.go
index 446d7ec..87c0e2b 100644
--- a/pkg/rpe/rmr.go
+++ b/pkg/rpe/rmr.go
@@ -32,9 +32,10 @@ package rpe
 
 import (
 	"gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+	"routing-manager/pkg/models"
 	"routing-manager/pkg/rtmgr"
 	"strconv"
-	//"strings"
+	"strings"
 )
 
 type Rmr struct {
@@ -78,34 +79,103 @@ func (r *Rmr) generateRMRPolicies(eps rtmgr.Endpoints, rcs *rtmgr.RicComponents,
 				group += ";" + member
 			}
 		}
-		rawrte += group
 
-                if (rte.RouteType == "%meid") {
-                        rawrte += group + rte.RouteType
-                }
+		if rte.RouteType == "%meid" {
+			rawrte += "%" + "meid"
+		} else {
+			rawrte += group
+		}
 
 		rawrt = append(rawrt, rawrte+"\n")
 	}
-	for _,val := range rtmgr.DynamicRouteList {
-		rawrt = append(rawrt,val)
+	for _, val := range rtmgr.DynamicRouteList {
+		rawrt = append(rawrt, val)
 	}
 
 	rawrt = append(rawrt, key+"newrt|end\n")
-        count := 0
+	count := 0
+
+	rawrt = append(rawrt, key+"meid_map|start\n")
+	
+	keys := make(map[string]MeidEntry)
+	MEID := ""
+	E2TIP := ""
+	RECTYP := ""
+	for _, value := range rcs.MeidMap {
+		if strings.Contains(value, "mme_ar") {
+			tmpstr := strings.Split(value, "|")
+			//MEID entry for mme_ar must always contain 3 strings speartred by | i.e "mme_ar|<string1>|<string2>"
+			MEID = strings.TrimSuffix(tmpstr[2], "\n")
+			E2TIP = strings.TrimSuffix(tmpstr[1], "\n")
+			RECTYP = "mme_ar"
+		} else if strings.Contains(value, "mme_del") {
+			tmpstr := strings.Split(value, "|")
+			MEID = strings.TrimSuffix(tmpstr[1], "\n")
+			E2TIP = ""
+			RECTYP = "mme_del"
+		}
+		keys[MEID] = MeidEntry{RECTYP, E2TIP}
+	}
+
+	for k, v := range keys {
+		if v.recordtype == "mme_ar" {
+			rawrt = append(rawrt, key+v.recordtype+"|"+v.e2tip+"|"+k+"\n")
+		    count++
+		}
+	}
 
-	meidrt := key +"meid_map|start\n"
-        for _, value := range rcs.MeidMap {
-            meidrt += key + value + "\n"
-            count++
-        }
-        meidrt += key+"meid_map|end|" + strconv.Itoa(count) +"\n"
+	rawrt = removeEmptyStrings(rawrt)
+	rawrt = append(rawrt, key+"meid_map|end|"+strconv.Itoa(count)+"\n")
 
-	rawrt = append(rawrt, meidrt)
 	xapp.Logger.Debug("rmr.GeneratePolicies returns: %v", rawrt)
 	xapp.Logger.Debug("rmr.GeneratePolicies returns: %v", rcs)
 	return &rawrt
 }
 
+/*
+Produces the raw route message consumable by RMR
+*/
+func (r *Rmr) generatePartialRMRPolicies(eps rtmgr.Endpoints, xappSubData *models.XappSubscriptionData, key string, updatetype rtmgr.RMRUpdateType) *[]string {
+	rawrt := []string{key + "updatert|start\n"}
+	rt := r.generatePartialRouteTable(eps, xappSubData, updatetype)
+	for _, rte := range *rt {
+		rawrte := key + "mse|" + rte.MessageType
+		for _, tx := range rte.TxList {
+			rawrte += "," + tx.Ip + ":" + strconv.Itoa(int(tx.Port))
+		}
+		rawrte += "|" + strconv.Itoa(int(rte.SubID)) + "|"
+		group := ""
+		for _, rxg := range rte.RxGroups {
+			member := ""
+			for _, rx := range rxg {
+				if member == "" {
+					member += rx.Ip + ":" + strconv.Itoa(int(rx.Port))
+				} else {
+					member += "," + rx.Ip + ":" + strconv.Itoa(int(rx.Port))
+				}
+			}
+			if group == "" {
+				group += member
+			} else {
+				group += ";" + member
+			}
+		}
+
+		if rte.RouteType == "%meid" {
+			rawrte += "%" + "meid"
+		} else {
+			rawrte += group
+		}
+
+		rawrt = append(rawrt, rawrte+"\n")
+	}
+
+	rawrt = append(rawrt, key+"updatert|end\n")
+	//count := 0
+
+	xapp.Logger.Debug("rmr.GeneratePolicies returns: %v", rawrt)
+	return &rawrt
+}
 func (r *RmrPush) GeneratePolicies(eps rtmgr.Endpoints, rcs *rtmgr.RicComponents) *[]string {
 	xapp.Logger.Debug("Invoked rmr.GeneratePolicies, args: %v: ", eps)
 	return r.generateRMRPolicies(eps, rcs, "")
@@ -114,3 +184,18 @@ func (r *RmrPush) GeneratePolicies(eps rtmgr.Endpoints, rcs *rtmgr.RicComponents
 func (r *RmrPush) GenerateRouteTable(eps rtmgr.Endpoints) *rtmgr.RouteTable {
 	return r.generateRouteTable(eps)
 }
+
+func (r *RmrPush) GeneratePartialPolicies(eps rtmgr.Endpoints, xappSubData *models.XappSubscriptionData, updatetype rtmgr.RMRUpdateType) *[]string {
+	xapp.Logger.Debug("Invoked rmr.GeneratePartialPolicies, args: %v: ", eps)
+	return r.generatePartialRMRPolicies(eps, xappSubData, "", updatetype)
+}
+
+func removeEmptyStrings(s []string) []string {
+	var r []string
+	for _, str := range s {
+		if str != "" {
+			r = append(r, str)
+		}
+	}
+	return r
+}