Release 0.4.2
[ric-plt/submgr.git] / pkg / teststub / rmrenv.go
1 /*
2 ==================================================================================
3   Copyright (c) 2019 AT&T Intellectual Property.
4   Copyright (c) 2019 Nokia
5
6    Licensed under the Apache License, Version 2.0 (the "License");
7    you may not use this file except in compliance with the License.
8    You may obtain a copy of the License at
9
10        http://www.apache.org/licenses/LICENSE-2.0
11
12    Unless required by applicable law or agreed to in writing, software
13    distributed under the License is distributed on an "AS IS" BASIS,
14    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15    See the License for the specific language governing permissions and
16    limitations under the License.
17 ==================================================================================
18 */
19
20 package teststub
21
22 import (
23         "gerrit.o-ran-sc.org/r/ric-plt/submgr/pkg/xapptweaks"
24         "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
25         "os"
26         "strconv"
27 )
28
29 //-----------------------------------------------------------------------------
30 //
31 //-----------------------------------------------------------------------------
32
33 type RmrRouteTable struct {
34         tmpfile string
35         routes  []string
36         meids   []string
37 }
38
39 func (rrt *RmrRouteTable) AddRoute(mtype int, src string, subid int, trg string) {
40
41         line := "mse|"
42         line += strconv.FormatInt(int64(mtype), 10)
43         if len(src) > 0 {
44                 line += "," + src
45         }
46         line += "|"
47         line += strconv.FormatInt(int64(subid), 10)
48         line += "|"
49         line += trg
50         rrt.routes = append(rrt.routes, line)
51 }
52
53 func (rrt *RmrRouteTable) AddMeid(trg string, meids []string) {
54
55         line := "mme_ar"
56         line += "|"
57         line += trg
58         line += "|"
59         for _, str := range meids {
60                 line += " " + str
61         }
62         rrt.meids = append(rrt.meids, line)
63 }
64
65 func (rrt *RmrRouteTable) DelMeid(meids []string) {
66
67         line := "mme_del"
68         line += "|"
69         for _, str := range meids {
70                 line += " " + str
71         }
72         rrt.meids = append(rrt.meids, line)
73 }
74
75 func (rrt *RmrRouteTable) FileName() string {
76         return rrt.tmpfile
77 }
78
79 func (rrt *RmrRouteTable) Table() string {
80         allrt := "newrt|start\n"
81         for _, val := range rrt.routes {
82                 allrt += val + "\n"
83         }
84         allrt += "newrt|end\n"
85         allrt += "meid_map | start\n"
86         for _, val := range rrt.meids {
87                 allrt += val + "\n"
88         }
89         allrt += "meid_map | end | " + strconv.FormatInt(int64(len(rrt.meids)), 10) + "\n"
90         return allrt
91 }
92
93 func (rrt *RmrRouteTable) Enable() {
94         if len(rrt.tmpfile) > 0 {
95                 os.Remove(rrt.tmpfile)
96         }
97         rrt.tmpfile, _ = CreateTmpFile(rrt.Table())
98         os.Setenv("RMR_SEED_RT", rrt.tmpfile)
99         xapp.Logger.Info("Using rt file %s", os.Getenv("RMR_SEED_RT"))
100 }
101
102 func (rrt *RmrRouteTable) Disable() {
103         if len(rrt.tmpfile) > 0 {
104                 os.Remove(rrt.tmpfile)
105                 os.Unsetenv("RMR_SEED_RT")
106                 rrt.tmpfile = ""
107                 xapp.Logger.Info("Not using rt file ")
108         }
109 }
110
111 //-----------------------------------------------------------------------------
112 //
113 //-----------------------------------------------------------------------------
114
115 type RmrSrcId struct {
116         xapptweaks.RmrEndpoint
117 }
118
119 func (rsi *RmrSrcId) Enable() {
120         if rsi.Port > 0 {
121                 os.Setenv("RMR_SRC_ID", rsi.String())
122                 xapp.Logger.Info("Using src id  %s", os.Getenv("RMR_SRC_ID"))
123         }
124 }
125
126 func (rsi *RmrSrcId) Disable() {
127         os.Unsetenv("RMR_SRC_ID")
128         xapp.Logger.Info("Not using Using src id")
129 }
130
131 //-----------------------------------------------------------------------------
132 //
133 //-----------------------------------------------------------------------------
134 type RmrRtgSvc struct {
135         xapptweaks.RmrEndpoint
136 }
137
138 func (rrs *RmrRtgSvc) Enable() {
139         if rrs.Port > 0 {
140                 os.Setenv("RMR_RTG_SVC", rrs.String())
141                 xapp.Logger.Info("Using rtg svc  %s", os.Getenv("RMR_SRC_ID"))
142         }
143 }
144
145 func (rrs *RmrRtgSvc) Disable() {
146         os.Unsetenv("RMR_RTG_SVC")
147         xapp.Logger.Info("Not using Using rtg svc")
148 }