First commit

setup/e2mgr/E2Manager/rmrCgo/rmrCgoApi.go

@@ -0,0 +1,161 @@
+//
+// Copyright 2019 AT&T Intellectual Property
+// 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 rmrCgo
+
+// #cgo LDFLAGS: -L/usr/local/lib -lrmr_si
+// #include <rmr/rmr.h>
+// #include <stdlib.h>
+import "C"
+import (
+	"fmt"
+	"github.com/pkg/errors"
+	"strings"
+	"time"
+	"unsafe"
+
+	"e2mgr/logger"
+)
+
+func (*Context) Init(port string, maxMsgSize int, flags int, logger *logger.Logger) RmrMessenger {
+	pp := C.CString(port)
+	defer C.free(unsafe.Pointer(pp))
+	logger.Debugf("#rmrCgoApi.Init - Going to initiate RMR router")
+	ctx := NewContext(maxMsgSize, flags, C.rmr_init(pp, C.int(maxMsgSize), C.int(flags)), logger)
+	start := time.Now()
+	//TODO use time.Ticker()
+	for !ctx.IsReady() {
+		time.Sleep(time.Second)
+		if time.Since(start) >= time.Minute {
+			logger.Debugf("#rmrCgoApi.Init - Routing table is not ready")
+			start = time.Now()
+		}
+	}
+	logger.Infof("#rmrCgoApi.Init - RMR router has been initiated")
+
+	// Configure the rmr to make rounds of attempts to send a message before notifying the application that it should retry.
+	// Each round is about 1000 attempts with a short sleep between each round.
+	C.rmr_set_stimeout(ctx.RmrCtx, C.int(1000))
+	r := RmrMessenger(ctx)
+	return r
+}
+
+func (ctx *Context) SendMsg(msg *MBuf, printLogs bool) (*MBuf, error) {
+	ctx.checkContextInitialized()
+	ctx.Logger.Debugf("#rmrCgoApi.SendMsg - Going to send message. MBuf: %v", *msg)
+	allocatedCMBuf := ctx.getAllocatedCRmrMBuf(ctx.Logger, msg, ctx.MaxMsgSize)
+	state := allocatedCMBuf.state
+	if state != RMR_OK {
+		errorMessage := fmt.Sprintf("#rmrCgoApi.SendMsg - Failed to get allocated message. state: %v - %s", state, states[int(state)])
+		return nil, errors.New(errorMessage)
+	}
+
+	if printLogs {
+		//TODO: if debug enabled
+		transactionId := string(*msg.XAction)
+		tmpTid := strings.TrimSpace(transactionId)
+		ctx.Logger.Infof("[E2 Manager -> RMR] #rmrCgoApi.SendMsg - Going to send message %v for transaction id: %s", *msg, tmpTid)
+	}
+
+	currCMBuf := C.rmr_send_msg(ctx.RmrCtx, allocatedCMBuf)
+	defer C.rmr_free_msg(currCMBuf)
+
+	state = currCMBuf.state
+
+	if state != RMR_OK {
+		errorMessage := fmt.Sprintf("#rmrCgoApi.SendMsg - Failed to send message. state: %v - %s", state, states[int(state)])
+		return nil, errors.New(errorMessage)
+	}
+
+	return convertToMBuf(ctx.Logger, currCMBuf), nil
+}
+
+func (ctx *Context) WhSendMsg(msg *MBuf, printLogs bool) (*MBuf, error) {
+	ctx.checkContextInitialized()
+	ctx.Logger.Debugf("#rmrCgoApi.WhSendMsg - Going to wormhole send message. MBuf: %v", *msg)
+
+	whid := C.rmr_wh_open(ctx.RmrCtx, (*C.char)(msg.GetMsgSrc()))		// open direct connection, returns wormhole ID
+	ctx.Logger.Infof("#rmrCgoApi.WhSendMsg - The wormhole id %v has been received", whid)
+	defer C.rmr_wh_close(ctx.RmrCtx, whid)
+
+	allocatedCMBuf := ctx.getAllocatedCRmrMBuf(ctx.Logger, msg, ctx.MaxMsgSize)
+	state := allocatedCMBuf.state
+	if state != RMR_OK {
+		errorMessage := fmt.Sprintf("#rmrCgoApi.WhSendMsg - Failed to get allocated message. state: %v - %s", state, states[int(state)])
+		return nil, errors.New(errorMessage)
+	}
+
+	if printLogs {
+		transactionId := string(*msg.XAction)
+		tmpTid := strings.TrimSpace(transactionId)
+		ctx.Logger.Infof("[E2 Manager -> RMR] #rmrCgoApi.WhSendMsg - Going to send message %v for transaction id: %s", *msg, tmpTid)
+	}
+
+	currCMBuf := C.rmr_wh_send_msg(ctx.RmrCtx, whid, allocatedCMBuf)
+	defer C.rmr_free_msg(currCMBuf)
+
+	state = currCMBuf.state
+
+	if state != RMR_OK {
+		errorMessage := fmt.Sprintf("#rmrCgoApi.WhSendMsg - Failed to send message. state: %v - %s", state, states[int(state)])
+		return nil, errors.New(errorMessage)
+	}
+
+	return convertToMBuf(ctx.Logger, currCMBuf), nil
+}
+
+
+func (ctx *Context) RecvMsg() (*MBuf, error) {
+	ctx.checkContextInitialized()
+	ctx.Logger.Debugf("#rmrCgoApi.RecvMsg - Going to receive message")
+	allocatedCMBuf := C.rmr_alloc_msg(ctx.RmrCtx, C.int(ctx.MaxMsgSize))
+
+	currCMBuf := C.rmr_rcv_msg(ctx.RmrCtx, allocatedCMBuf)
+	defer C.rmr_free_msg(currCMBuf)
+
+	state := currCMBuf.state
+
+	if state != RMR_OK {
+		errorMessage := fmt.Sprintf("#rmrCgoApi.RecvMsg - Failed to receive message. state: %v - %s", state, states[int(state)])
+		ctx.Logger.Errorf(errorMessage)
+		return nil, errors.New(errorMessage)
+	}
+
+	mbuf := convertToMBuf(ctx.Logger, currCMBuf)
+
+	if mbuf.MType != E2_TERM_KEEP_ALIVE_RESP {
+
+		transactionId := string(*mbuf.XAction)
+		tmpTid := strings.TrimSpace(transactionId)
+		ctx.Logger.Infof("[RMR -> E2 Manager] #rmrCgoApi.RecvMsg - message %v has been received for transaction id: %s", *mbuf, tmpTid)
+	}
+	return mbuf, nil
+}
+
+func (ctx *Context) IsReady() bool {
+	ctx.Logger.Debugf("#rmrCgoApi.IsReady - Going to check if routing table is initialized")
+	return int(C.rmr_ready(ctx.RmrCtx)) != 0
+}
+
+func (ctx *Context) Close() {
+	ctx.Logger.Debugf("#rmrCgoApi.Close - Going to close RMR context")
+	C.rmr_close(ctx.RmrCtx)
+	time.Sleep(100 * time.Millisecond)
+}