[RICPLT-2523] Add SelectE2TInstance mechanism | Change Setup flow | Refactor code...
[ric-plt/e2mgr.git] / E2Manager / services / rnib_data_service.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
18 package services
19
20 import (
21         "e2mgr/configuration"
22         "e2mgr/logger"
23         "e2mgr/rNibWriter"
24         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
25         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
26         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
27         "net"
28         "time"
29 )
30
31 type RNibDataService interface {
32         SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error
33         UpdateNodebInfo(nodebInfo *entities.NodebInfo) error
34         SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) error
35         GetNodeb(ranName string) (*entities.NodebInfo, error)
36         GetListNodebIds() ([]*entities.NbIdentity, error)
37         PingRnib() bool
38         GetE2TInstance(address string) (*entities.E2TInstance, error)
39         GetE2TInstances(addresses []string) ([]*entities.E2TInstance, error)
40         GetE2TAddresses() ([]string, error)
41         SaveE2TInstance(e2tInstance *entities.E2TInstance) error
42         SaveE2TAddresses(addresses []string) error
43 }
44
45 type rNibDataService struct {
46         logger        *logger.Logger
47         rnibReader    reader.RNibReader
48         rnibWriter    rNibWriter.RNibWriter
49         maxAttempts   int
50         retryInterval time.Duration
51 }
52
53 func NewRnibDataService(logger *logger.Logger, config *configuration.Configuration, rnibReader reader.RNibReader, rnibWriter rNibWriter.RNibWriter) *rNibDataService {
54         return &rNibDataService{
55                 logger:        logger,
56                 rnibReader:    rnibReader,
57                 rnibWriter:    rnibWriter,
58                 maxAttempts:   config.MaxRnibConnectionAttempts,
59                 retryInterval: time.Duration(config.RnibRetryIntervalMs) * time.Millisecond,
60         }
61 }
62
63 func (w *rNibDataService) UpdateNodebInfo(nodebInfo *entities.NodebInfo) error {
64         w.logger.Infof("#RnibDataService.UpdateNodebInfo - nodebInfo: %s", nodebInfo)
65
66         err := w.retry("UpdateNodebInfo", func() (err error) {
67                 err = w.rnibWriter.UpdateNodebInfo(nodebInfo)
68                 return
69         })
70
71         return err
72 }
73
74 func (w *rNibDataService) SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error {
75         w.logger.Infof("#RnibDataService.SaveNodeb - nbIdentity: %s, nodebInfo: %s", nbIdentity, nb)
76
77         err := w.retry("SaveNodeb", func() (err error) {
78                 err = w.rnibWriter.SaveNodeb(nbIdentity, nb)
79                 return
80         })
81
82         return err
83 }
84
85 func (w *rNibDataService) SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) error {
86         w.logger.Infof("#RnibDataService.SaveRanLoadInformation - inventoryName: %s, ranLoadInformation: %s", inventoryName, ranLoadInformation)
87
88         err := w.retry("SaveRanLoadInformation", func() (err error) {
89                 err = w.rnibWriter.SaveRanLoadInformation(inventoryName, ranLoadInformation)
90                 return
91         })
92
93         return err
94 }
95
96 func (w *rNibDataService) GetNodeb(ranName string) (*entities.NodebInfo, error) {
97         w.logger.Infof("#RnibDataService.GetNodeb - RAN name: %s", ranName)
98
99         var nodeb *entities.NodebInfo = nil
100
101         err := w.retry("GetNodeb", func() (err error) {
102                 nodeb, err = w.rnibReader.GetNodeb(ranName)
103                 return
104         })
105
106         return nodeb, err
107 }
108
109 func (w *rNibDataService) GetListNodebIds() ([]*entities.NbIdentity, error) {
110         w.logger.Infof("#RnibDataService.GetListNodebIds")
111
112         var nodeIds []*entities.NbIdentity = nil
113
114         err := w.retry("GetListNodebIds", func() (err error) {
115                 nodeIds, err = w.rnibReader.GetListNodebIds()
116                 return
117         })
118
119         return nodeIds, err
120 }
121
122 func (w *rNibDataService) GetE2TInstance(address string) (*entities.E2TInstance, error) {
123         w.logger.Infof("#RnibDataService.GetE2TInstance")
124
125         var e2tInstance *entities.E2TInstance = nil
126
127         err := w.retry("GetE2TInstance", func() (err error) {
128                 e2tInstance, err = w.rnibReader.GetE2TInstance(address)
129                 return
130         })
131
132         return e2tInstance, err
133 }
134
135 func (w *rNibDataService) GetE2TInstances(addresses []string) ([]*entities.E2TInstance, error) {
136         w.logger.Infof("#RnibDataService.GetE2TInstances")
137
138         var e2tInstances []*entities.E2TInstance = nil
139
140         err := w.retry("GetE2TInstance", func() (err error) {
141                 e2tInstances, err = w.rnibReader.GetE2TInstances(addresses)
142                 return
143         })
144
145         return e2tInstances, err
146 }
147
148 func (w *rNibDataService) GetE2TAddresses() ([]string, error) {
149         w.logger.Infof("#RnibDataService.GetE2TAddresses")
150
151         var e2tAddresses []string = nil
152
153         err := w.retry("GetE2TAddresses", func() (err error) {
154                 e2tAddresses, err = w.rnibReader.GetE2TAddresses()
155                 return
156         })
157
158         return e2tAddresses, err
159 }
160
161 func (w *rNibDataService) SaveE2TInstance(e2tInstance *entities.E2TInstance) error {
162         w.logger.Infof("#RnibDataService.SaveE2TInstance")
163
164         err := w.retry("SaveE2TInstance", func() (err error) {
165                 err = w.rnibWriter.SaveE2TInstance(e2tInstance)
166                 return
167         })
168
169         return err
170 }
171
172 func (w *rNibDataService) SaveE2TAddresses(addresses []string) error {
173         w.logger.Infof("#RnibDataService.SaveE2TAddresses")
174
175         err := w.retry("SaveE2TAddresses", func() (err error) {
176                 err = w.rnibWriter.SaveE2TAddresses(addresses)
177                 return
178         })
179
180         return err
181 }
182
183 func (w *rNibDataService) PingRnib() bool {
184         err := w.retry("GetListNodebIds", func() (err error) {
185                 _, err = w.rnibReader.GetListNodebIds()
186                 return
187         })
188
189         return !isRnibConnectionError(err)
190 }
191
192 func (w *rNibDataService) retry(rnibFunc string, f func() error) (err error) {
193         attempts := w.maxAttempts
194
195         for i := 1; ; i++ {
196                 err = f()
197                 if err == nil {
198                         return
199                 }
200                 if !isRnibConnectionError(err) {
201                         return err
202                 }
203                 if i >= attempts {
204                         w.logger.Errorf("#RnibDataService.retry - after %d attempts of %s, last error: %s", attempts, rnibFunc, err)
205                         return err
206                 }
207                 time.Sleep(w.retryInterval)
208
209                 w.logger.Infof("#RnibDataService.retry - retrying %d %s after error: %s", i, rnibFunc, err)
210         }
211 }
212
213 func isRnibConnectionError(err error) bool {
214         internalErr, ok := err.(*common.InternalError)
215         if !ok {
216                 return false
217         }
218         _, ok = internalErr.Err.(*net.OpError)
219
220         return ok
221 }