Merge "Improve unit test coverage"
[ric-plt/e2mgr.git] / E2Manager / services / rmrsender / rmr_sender.go
1 //
2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 //      http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16
17 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 //  platform project (RICP).
19
20
21 package rmrsender
22
23 import (
24         "e2mgr/logger"
25         "e2mgr/models"
26         "e2mgr/rmrCgo"
27 )
28
29 type RmrSender struct {
30         logger    *logger.Logger
31         messenger rmrCgo.RmrMessenger
32 }
33
34 func NewRmrSender(logger *logger.Logger, messenger rmrCgo.RmrMessenger) *RmrSender {
35         return &RmrSender{
36                 logger:    logger,
37                 messenger: messenger,
38         }
39 }
40
41 func (r *RmrSender) WhSend(rmrMessage *models.RmrMessage) error {
42         msg := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction, rmrMessage.GetMsgSrc())
43
44         _, err := r.messenger.WhSendMsg(msg, true)
45
46         if err != nil {
47                 r.logger.Errorf("#RmrSender.WhSend - RAN name: %s , Message type: %d - Failed sending message. Error: %v", rmrMessage.RanName, rmrMessage.MsgType, err)
48                 return err
49         }
50
51         r.logger.Infof("#RmrSender.WhSend - RAN name: %s , Message type: %d - Successfully sent RMR message", rmrMessage.RanName, rmrMessage.MsgType)
52         return nil
53 }
54
55 func (r *RmrSender) Send(rmrMessage *models.RmrMessage) error {
56         msg := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction, rmrMessage.GetMsgSrc())
57
58         _, err := r.messenger.SendMsg(msg, true)
59
60         if err != nil {
61                 r.logger.Errorf("#RmrSender.Send - RAN name: %s , Message type: %d - Failed sending message. Error: %v", rmrMessage.RanName, rmrMessage.MsgType, err)
62                 return err
63         }
64
65         r.logger.Infof("#RmrSender.Send - RAN name: %s , Message type: %d - Successfully sent RMR message", rmrMessage.RanName, rmrMessage.MsgType)
66         return nil
67 }
68
69 func (r *RmrSender) SendWithoutLogs(rmrMessage *models.RmrMessage) error {
70         msg := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction, rmrMessage.GetMsgSrc())
71
72         _, err := r.messenger.SendMsg(msg, false)
73
74         if err != nil {
75                 r.logger.Errorf("#RmrSender.Send - RAN name: %s , Message type: %d - Failed sending message. Error: %v", rmrMessage.RanName, rmrMessage.MsgType, err)
76                 return err
77         }
78
79         return nil
80 }