Reverting the last change
[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/xapp-frame/pkg/xapp"
24         "os"
25         "strconv"
26 )
27
28 //-----------------------------------------------------------------------------
29 //
30 //-----------------------------------------------------------------------------
31
32 type RmrRouteTable struct {
33         tmpfile string
34         routes  []string
35         meids   []string
36 }
37
38 func (rrt *RmrRouteTable) AddRoute(mtype int, src string, subid int, trg string) {
39
40         line := "mse|"
41         line += strconv.FormatInt(int64(mtype), 10)
42         if len(src) > 0 {
43                 line += "," + src
44         }
45         line += "|"
46         line += strconv.FormatInt(int64(subid), 10)
47         line += "|"
48         line += trg
49         rrt.routes = append(rrt.routes, line)
50 }
51
52 func (rrt *RmrRouteTable) AddMeid(trg string, meids []string) {
53
54         line := "mme_ar"
55         line += " | "
56         line += trg
57         line += " | "
58         for _, str := range meids {
59                 line += " " + str
60         }
61         rrt.meids = append(rrt.meids, line)
62 }
63
64 func (rrt *RmrRouteTable) DelMeid(meids []string) {
65
66         line := "mme_del"
67         line += " | "
68         for _, str := range meids {
69                 line += " " + str
70         }
71         rrt.meids = append(rrt.meids, line)
72 }
73
74 func (rrt *RmrRouteTable) FileName() string {
75         return rrt.tmpfile
76 }
77
78 func (rrt *RmrRouteTable) Table() string {
79         allrt := "newrt|start\n"
80         for _, val := range rrt.routes {
81                 allrt += val + "\n"
82         }
83         allrt += "newrt|end\n"
84         allrt += "\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                 rrt.tmpfile, _ = CreateTmpFile(rrt.Table())
96         }
97         os.Setenv("RMR_SEED_RT", rrt.tmpfile)
98         os.Setenv("RMR_RTG_SVC", "-1")
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                 os.Unsetenv("RMR_RTG_SVC")
107                 rrt.tmpfile = ""
108                 xapp.Logger.Info("Not using rt file ")
109         }
110 }
111
112 //-----------------------------------------------------------------------------
113 //
114 //-----------------------------------------------------------------------------
115
116 type RmrSrcId struct {
117         xapp.RmrEndpoint
118 }
119
120 func (rsi *RmrSrcId) Enable() {
121         if rsi.Port > 0 {
122                 os.Setenv("RMR_SRC_ID", rsi.String())
123                 xapp.Logger.Info("Using src id  %s", os.Getenv("RMR_SRC_ID"))
124         }
125 }
126
127 func (rsi *RmrSrcId) Disable() {
128         os.Unsetenv("RMR_SRC_ID")
129         xapp.Logger.Info("Not using Using src id")
130 }
131
132 //-----------------------------------------------------------------------------
133 //
134 //-----------------------------------------------------------------------------
135 type RmrRtgSvc struct {
136         xapp.RmrEndpoint
137 }
138
139 func (rrs *RmrRtgSvc) Enable() {
140         if rrs.Port > 0 {
141                 os.Setenv("RMR_RTG_SVC", rrs.String())
142                 xapp.Logger.Info("Using rtg svc  %s", os.Getenv("RMR_SRC_ID"))
143         }
144 }
145
146 func (rrs *RmrRtgSvc) Disable() {
147         os.Unsetenv("RMR_RTG_SVC")
148         xapp.Logger.Info("Not using Using rtg svc")
149 }