X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Frco%2Frco.go;h=6ba36d0868420aea36aefb8a7c8652980ba11641;hb=93cc3e245f87798c8753209980817727e0648401;hp=5109530ad78c77b8ee73fc5a61f56076818fed62;hpb=4511475f03107eb53c88545649898e3cfbcc2765;p=ric-plt%2Fsubmgr.git

diff --git a/test/rco/rco.go b/test/rco/rco.go
index 5109530..6ba36d0 100644
--- a/test/rco/rco.go
+++ b/test/rco/rco.go
@@ -20,30 +20,121 @@
 package main
 
 import (
-	"gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+	"encoding/hex"
+	"errors"
 	submgr "gerrit.o-ran-sc.org/r/ric-plt/submgr/pkg/control"
+	"gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+	"github.com/spf13/viper"
 	"time"
 	"os"
 )
 
 type Rco struct {
+	submgr.E2ap
 }
 
-var	c chan submgr.RmrDatagram = make(chan submgr.RmrDatagram, 1)
+var c chan submgr.RmrDatagram = make(chan submgr.RmrDatagram, 1)
+var params *xapp.RMRParams
+
+var REQUESTRAWDATA string
+var DELETERAWDATA string
+var SEEDSN uint16
+var DELETESEEDSN uint16
 
-func (m Rco) Consume(mtype, sub_id int, len int, payload []byte) (err error) {
+func init() {
+	viper.AutomaticEnv()
+	viper.SetEnvPrefix("rco")
+	viper.AllowEmptyEnv(true)
+	REQUESTRAWDATA = viper.GetString("rawdata")
+	if REQUESTRAWDATA == "" {
+		REQUESTRAWDATA = "000003ea7e000500aaaaccccea6300020000ea81000e00045465737400ea6b0003000100"
+	}
+	DELETERAWDATA = viper.GetString("deleterawdata")
+	if DELETERAWDATA == "" {
+		DELETERAWDATA = "000002ea7e000500aaaabbbbea6300020000"
+	}
+	xapp.Logger.Info("Initial RAW DATA: %v", REQUESTRAWDATA)
+	xapp.Logger.Info("Initial DELETE RAW DATA: %v", DELETERAWDATA)
+	SEEDSN = uint16(viper.GetInt("seed_sn"))
+	if SEEDSN == 0 || SEEDSN > 65535 {
+		SEEDSN = 12345
+	}
+	DELETESEEDSN = SEEDSN
+	xapp.Logger.Info("Initial SEQUENCE NUMBER: %v", SEEDSN)
+}
+
+func (r *Rco) GeneratePayload(sub_id uint16) (payload []byte, err error) {
+	skeleton, err := hex.DecodeString(REQUESTRAWDATA)
+	if err != nil {
+		return make([]byte, 0), errors.New("Unable to decode data provided in RCO_RAWDATA environment variable")
+	}
+	payload, err = r.SetSubscriptionRequestSequenceNumber(skeleton, sub_id)
 	return
 }
 
-func (r *Rco) send(datagram submgr.RmrDatagram) {
-	xapp.Rmr.Send(datagram.MessageType, datagram.SubscriptionId, len(datagram.Payload), datagram.Payload)
+func (r *Rco) GenerateDeletePayload(sub_id uint16) (payload []byte, err error) {
+	skeleton, err := hex.DecodeString(DELETERAWDATA)
+	if err != nil {
+		return make([]byte, 0), errors.New("Unable to decode data provided in RCO_DELETE RAWDATA environment variable")
+	}
+	xapp.Logger.Info("SetSubscriptionDeleteRequestSequenceNumber1")
+	payload, err = r.SetSubscriptionDeleteRequestSequenceNumber(skeleton, sub_id)
+	xapp.Logger.Info("SetSubscriptionDeleteRequestSequenceNumber2")
+	return
+}
+
+func (r Rco) Consume(params *xapp.RMRParams) (err error) {
+	payload_seq_num, err := r.GetSubscriptionResponseSequenceNumber(params.Payload)
+	if err != nil {
+		xapp.Logger.Error("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())	
+	}
+	xapp.Logger.Info("Message Received: RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v", params.SubId, payload_seq_num)
+	return
+}
+
+func (r *Rco) SendRequests() (err error) {
+	message, err := r.GeneratePayload(SEEDSN)
+	if err != nil {
+		xapp.Logger.Debug(err.Error())
+		return
+	}
+	deletemessage, err := r.GenerateDeletePayload(DELETESEEDSN)
+	if err != nil {
+		xapp.Logger.Debug(err.Error())
+		return
+	}
+	for {
+		time.Sleep(2 * time.Second)
+		c <- submgr.RmrDatagram{12010, SEEDSN, message}
+		time.Sleep(2 * time.Second)
+		c <- submgr.RmrDatagram{12020, DELETESEEDSN, deletemessage}
+		DELETESEEDSN++
+	}
+	return
 }
 
 func (r *Rco) Run() {
 	for {
-		message := <- c
-		xapp.Logger.Info("RCO Message - Type=%v SubID=%v", message.MessageType, message.SubscriptionId)
-		r.send(message)
+		message := <-c
+		payload_seq_num, err := r.GetSubscriptionRequestSequenceNumber(message.Payload)
+		if err != nil {
+			xapp.Logger.Debug("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())	
+		}
+		params.SubId = int(message.SubscriptionId)
+		params.Mtype = message.MessageType
+		params.PayloadLen = len(message.Payload)
+		params.Payload = message.Payload
+		xapp.Logger.Info("Sending Message: TYPE: %v | RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v)", message.MessageType, message.SubscriptionId, payload_seq_num)
+		xapp.Rmr.Send(params, false)
+	}
+}
+
+func (r *Rco) sendInvalidTestMessages(){
+	for {
+		time.Sleep(7 * time.Second)
+		c <- submgr.RmrDatagram{10000, 0, make([]byte, 1)}
+		time.Sleep(7 * time.Second)
+		c <- submgr.RmrDatagram{12010, 0, make([]byte, 1)}
 	}
 }
 
@@ -51,21 +142,9 @@ func main() {
 	rco := Rco{}
 	go xapp.Rmr.Start(rco)
 	go rco.Run()
-	asn1 := submgr.Asn1{}
-	message, err := asn1.Encode(submgr.RmrPayload{8, 1111, "RCO: Subscription Request"})
+	go rco.sendInvalidTestMessages()
+	err := rco.SendRequests()
 	if err != nil {
-		xapp.Logger.Debug(err.Error())
 		os.Exit(1)
 	}
-	doSubscribe := true
-	for {
-		time.Sleep(2 * time.Second)
-		if doSubscribe {
-			c <- submgr.RmrDatagram{12010, 9999, message}
-			doSubscribe = false
-		} else {
-			c <- submgr.RmrDatagram{10000, 9999, make([]byte,0)}
-			doSubscribe = true
-		}
-	}
-}
\ No newline at end of file
+}