[RICPLT-2523] Acceptance fixes - RnibDataService / E2TInstancesManager...
[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
124         var e2tInstance *entities.E2TInstance = nil
125
126         err := w.retry("GetE2TInstance", func() (err error) {
127                 e2tInstance, err = w.rnibReader.GetE2TInstance(address)
128                 return
129         })
130
131         if err == nil {
132                 w.logger.Infof("#RnibDataService.GetE2TInstance - E2T instance address: %s, state: %s, associated RANs count: %d, keep Alive ts: %d", e2tInstance.Address, e2tInstance.State, len(e2tInstance.AssociatedRanList), e2tInstance.KeepAliveTimestamp)
133         }
134
135         return e2tInstance, err
136 }
137
138 func (w *rNibDataService) GetE2TInstances(addresses []string) ([]*entities.E2TInstance, error) {
139         w.logger.Infof("#RnibDataService.GetE2TInstances - addresses: %s", addresses)
140         var e2tInstances []*entities.E2TInstance = nil
141
142         err := w.retry("GetE2TInstance", func() (err error) {
143                 e2tInstances, err = w.rnibReader.GetE2TInstances(addresses)
144                 return
145         })
146
147         return e2tInstances, err
148 }
149
150 func (w *rNibDataService) GetE2TAddresses() ([]string, error) {
151
152         var e2tAddresses []string = nil
153
154         err := w.retry("GetE2TAddresses", func() (err error) {
155                 e2tAddresses, err = w.rnibReader.GetE2TAddresses()
156                 return
157         })
158
159         if err == nil {
160                 w.logger.Infof("#RnibDataService.GetE2TAddresses - addresses: %s", e2tAddresses)
161         }
162
163         return e2tAddresses, err
164 }
165
166 func (w *rNibDataService) SaveE2TInstance(e2tInstance *entities.E2TInstance) error {
167         w.logger.Infof("#RnibDataService.SaveE2TInstance - E2T instance address: %s, state: %s, associated RANs count: %d, keep Alive ts: %d", e2tInstance.Address, e2tInstance.State, len(e2tInstance.AssociatedRanList), e2tInstance.KeepAliveTimestamp)
168         err := w.retry("SaveE2TInstance", func() (err error) {
169                 err = w.rnibWriter.SaveE2TInstance(e2tInstance)
170                 return
171         })
172
173         return err
174 }
175
176 func (w *rNibDataService) SaveE2TAddresses(addresses []string) error {
177         w.logger.Infof("#RnibDataService.SaveE2TAddresses - addresses: %s", addresses)
178
179         err := w.retry("SaveE2TAddresses", func() (err error) {
180                 err = w.rnibWriter.SaveE2TAddresses(addresses)
181                 return
182         })
183
184         return err
185 }
186
187 func (w *rNibDataService) PingRnib() bool {
188         err := w.retry("GetListNodebIds", func() (err error) {
189                 _, err = w.rnibReader.GetListNodebIds()
190                 return
191         })
192
193         return !isRnibConnectionError(err)
194 }
195
196 func (w *rNibDataService) retry(rnibFunc string, f func() error) (err error) {
197         attempts := w.maxAttempts
198
199         for i := 1; ; i++ {
200                 err = f()
201                 if err == nil {
202                         return
203                 }
204                 if !isRnibConnectionError(err) {
205                         return err
206                 }
207                 if i >= attempts {
208                         w.logger.Errorf("#RnibDataService.retry - after %d attempts of %s, last error: %s", attempts, rnibFunc, err)
209                         return err
210                 }
211                 time.Sleep(w.retryInterval)
212
213                 w.logger.Infof("#RnibDataService.retry - retrying %d %s after error: %s", i, rnibFunc, err)
214         }
215 }
216
217 func isRnibConnectionError(err error) bool {
218         internalErr, ok := err.(*common.InternalError)
219         if !ok {
220                 return false
221         }
222         _, ok = internalErr.Err.(*net.OpError)
223
224         return ok
225 }