First commit

setup/e2mgr/tools/xappmock/sender/jsonSender.go

@@ -0,0 +1,179 @@
+//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//  platform project (RICP).
+
+// Copyright 2019 Nokia
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+
+//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//  platform project (RICP).
+
+package sender
+
+import (
+	"fmt"
+	"github.com/pkg/errors"
+	"os"
+	"reflect"
+	"strconv"
+	"strings"
+	"sync/atomic"
+	"time"
+	"unicode"
+	"xappmock/logger"
+	"xappmock/models"
+	"xappmock/rmr"
+)
+
+var counter uint64
+
+type JsonSender struct {
+	logger *logger.Logger
+}
+
+func NewJsonSender(logger *logger.Logger) *JsonSender {
+	return &JsonSender{
+		logger: logger,
+	}
+}
+
+func (s *JsonSender) SendJsonRmrMessage(command models.JsonCommand /*the copy is modified locally*/, xAction *[]byte, r *rmr.Service) error {
+	var payload []byte
+	_, err := fmt.Sscanf(command.PackedPayload, "%x", &payload)
+	if err != nil {
+		return errors.New(fmt.Sprintf("convert inputPayloadAsStr to payloadAsByte. Error: %v\n", err))
+	}
+	command.PackedPayload = string(payload)
+	command.TransactionId = expandTransactionId(command.TransactionId)
+	if len(command.TransactionId) == 0 {
+		command.TransactionId = string(*xAction)
+	}
+	command.PayloadHeader = s.expandPayloadHeader(command.PayloadHeader, &command)
+	s.logger.Infof("#JsonSender.SendJsonRmrMessage - command payload header: %s", command.PayloadHeader)
+	rmrMsgId, err := rmr.MessageIdToUint(command.RmrMessageType)
+	if err != nil {
+		return errors.New(fmt.Sprintf("invalid rmr message id: %s", command.RmrMessageType))
+	}
+
+	msg := append([]byte(command.PayloadHeader), payload...)
+	messageInfo := models.NewMessageInfo(int(rmrMsgId), command.RanName, msg, []byte(command.TransactionId))
+	s.logger.Infof("#JsonSender.SendJsonRmrMessage - going to send message: %s", messageInfo)
+
+	_, err = r.SendMessage(int(rmrMsgId), command.RanName, msg, []byte(command.TransactionId))
+	return err
+}
+
+/*
+ * transactionId (xAction): The value may have a fixed value or $ or <prefix>$.
+ * $ is replaced by a value generated at runtime (possibly unique per message sent).
+ * If the tag does not exist, then the mock shall use the value taken from the incoming message.
+ */
+func expandTransactionId(id string) string {
+	if len(id) == 1 && id[0] == '$' {
+		return fmt.Sprintf("%d", incAndGetCounter())
+	}
+	if len(id) > 1 && id[len(id)-1] == '$' {
+		return fmt.Sprintf("%s%d", id[:len(id)-1], incAndGetCounter())
+	}
+	return id
+}
+
+/*
+ * payloadHeader: A prefix to combine with the payload that will be the message’s payload. The value may include variables of the format $<name> or #<name> where:
+ *   $<name> expands to the value of <name> if it exists or the empty string if not.
+ *   #<name> expands to the length of the value of <name> if it exists or omitted if not.
+ * The intention is to allow the Mock to construct the payload header required by the setup messages (ranIp|ranPort|ranName|payload len|<payload>).
+ * Example: “payloadHeader”: “$ranIp|$ranPort|$ranName|#packedPayload|”
+ */
+
+func (s *JsonSender) expandPayloadHeader(header string, command *models.JsonCommand) string {
+	var name strings.Builder
+	var expandedHeader strings.Builder
+
+	r := strings.NewReader(header)
+	ch, err := r.ReadByte()
+	for {
+		if err != nil {
+			break
+		}
+
+		switch ch {
+		case '$':
+			for {
+				ch, err = r.ReadByte() //on error ch == 0
+				if unicode.IsDigit(rune(ch)) || unicode.IsLetter(rune(ch)) {
+					if name.Len() == 0 {
+						name.WriteByte(byte(unicode.ToUpper(rune(ch))))
+					} else {
+						name.WriteByte(ch)
+					}
+				} else {
+					if fieldValue := reflect.Indirect(reflect.ValueOf(command)).FieldByName(name.String()); fieldValue.IsValid() {
+						switch fieldValue.Kind() {
+						case reflect.String:
+							expandedHeader.WriteString(fieldValue.String())
+						case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
+							expandedHeader.WriteString(strconv.FormatInt(fieldValue.Int(), 10))
+						case reflect.Bool:
+							expandedHeader.WriteString(strconv.FormatBool(fieldValue.Bool()))
+						case reflect.Float64, reflect.Float32:
+							expandedHeader.WriteString(fmt.Sprintf("%g", fieldValue.Float()))
+						default:
+							s.logger.Errorf("#JsonSender.expandPayloadHeader - invalid type for $%s, value must be a string, an int, a bool or a float", name.String())
+							os.Exit(1)
+						}
+					}
+					name.Reset()
+					break
+				}
+			}
+		case '#':
+			for {
+				ch, err = r.ReadByte() //on error ch == 0
+				if unicode.IsDigit(rune(ch)) || unicode.IsLetter(rune(ch)) {
+					if name.Len() == 0 {
+						name.WriteByte(byte(unicode.ToUpper(rune(ch))))
+					} else {
+						name.WriteByte(ch)
+					}
+				} else {
+					if fieldValue := reflect.Indirect(reflect.ValueOf(command)).FieldByName(name.String()); fieldValue.IsValid() {
+						if fieldValue.Kind() == reflect.String {
+							expandedHeader.WriteString(strconv.FormatInt(int64(len(fieldValue.String())), 10))
+						} else {
+							s.logger.Errorf("#JsonSender.expandPayloadHeader - invalid type for #%s, value must be a string", name.String())
+							os.Exit(1)
+						}
+					}
+					name.Reset()
+					break
+				}
+			}
+		default:
+			if unicode.IsPrint(rune(ch)) {
+				expandedHeader.WriteByte(ch)
+			}
+			ch, err = r.ReadByte()
+		}
+	}
+	return expandedHeader.String()
+}
+
+func incAndGetCounter() uint64 {
+	return atomic.AddUint64(&counter, 1)
+}
+
+func init() {
+	counter = uint64(time.Now().Unix() - 1572000000)
+}