[RICPLT-2165] Add rnibDataService to support retries
[ric-plt/e2mgr.git] / E2Manager / handlers / httpmsghandlers / delete_all_request_handler.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 httpmsghandlers
19
20 import "C"
21 import (
22         "e2mgr/configuration"
23         "e2mgr/e2managererrors"
24         "e2mgr/logger"
25         "e2mgr/models"
26         "e2mgr/rmrCgo"
27         "e2mgr/services"
28         "e2mgr/stateMachine"
29         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
30         "time"
31 )
32
33 type DeleteAllRequestHandler struct {
34         rnibDataService services.RNibDataService
35         rmrService *services.RmrService
36         config *configuration.Configuration
37         logger         *logger.Logger
38 }
39
40 func NewDeleteAllRequestHandler(logger *logger.Logger, rmrService *services.RmrService, config *configuration.Configuration, rnibDataService services.RNibDataService) *DeleteAllRequestHandler {
41         return &DeleteAllRequestHandler{
42                 logger:         logger,
43                 rnibDataService: rnibDataService,
44                 rmrService:     rmrService,
45                 config:         config,
46         }
47 }
48
49 func (handler *DeleteAllRequestHandler) Handle(request models.Request) error {
50
51         err, continueFlow := handler.updateNodebStates(false)
52         if err != nil {
53                 return err
54         }
55
56         if continueFlow == false {
57                 return nil
58         }
59
60         //TODO change to rmr_request
61         response := models.NotificationResponse{MgsType: rmrCgo.RIC_SCTP_CLEAR_ALL}
62         if err := handler.rmrService.SendRmrMessage(&response); err != nil {
63                 handler.logger.Errorf("#DeleteAllRequestHandler.Handle - failed to send sctp clear all message to RMR: %s", err)
64                 return e2managererrors.NewRmrError()
65         }
66
67         time.Sleep(time.Duration(handler.config.BigRedButtonTimeoutSec) * time.Second)
68         handler.logger.Infof("#DeleteAllRequestHandler.Handle - timer expired")
69
70         err, _ = handler.updateNodebStates(true)
71         if err != nil {
72                 return err
73         }
74
75         return nil
76 }
77
78 func (handler *DeleteAllRequestHandler) updateNodebStates(timeoutExpired bool) (error, bool) {
79         nbIdentityList, err := handler.rnibDataService.GetListNodebIds()
80
81         if err != nil {
82                 handler.logger.Errorf("#DeleteAllRequestHandler.updateNodebStates - failed to get nodes list from RNIB. Error: %s", err.Error())
83                 return e2managererrors.NewRnibDbError(), false
84         }
85
86         if len(nbIdentityList) == 0 {
87                 return nil, false
88         }
89
90         numOfRanToShutDown := 0
91         for _, nbIdentity := range nbIdentityList {
92
93                 node, err := handler.rnibDataService.GetNodeb((*nbIdentity).GetInventoryName())
94
95                 if err != nil {
96                         handler.logger.Errorf("#DeleteAllRequestHandler.updateNodebStates - failed to get nodeB entity for ran name: %v from RNIB. Error: %s",
97                                 (*nbIdentity).GetInventoryName(), err.Error())
98                         continue
99                 }
100
101                 if timeoutExpired {
102
103                         if handler.saveNodebShutDownState(nbIdentity, node) {
104                                 numOfRanToShutDown++
105                         }
106                         continue
107                 }
108                 if handler.saveNodebNextState(nbIdentity, node) {
109                         numOfRanToShutDown++
110                 }
111         }
112
113         if numOfRanToShutDown > 0 {
114                 handler.logger.Infof("#DeleteAllRequestHandler.updateNodebStates - update nodebs states in RNIB completed")
115         } else {
116                 handler.logger.Infof("#DeleteAllRequestHandler.updateNodebStates - nodebs states are not updated ")
117                 return nil, false
118         }
119
120         return nil, true
121 }
122
123 func (handler *DeleteAllRequestHandler) saveNodebNextState(nbIdentity *entities.NbIdentity, node *entities.NodebInfo) bool {
124
125         if node.ConnectionStatus == entities.ConnectionStatus_SHUTTING_DOWN {
126                 return true
127         }
128
129         nextStatus, res := stateMachine.NodeNextStateDeleteAll(node.ConnectionStatus)
130         if res == false {
131                 return false
132         }
133
134         node.ConnectionStatus = nextStatus
135
136         err := handler.rnibDataService.SaveNodeb(nbIdentity, node)
137
138         if err != nil {
139                 handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebNextState - failed to save nodeB entity for inventory name: %v to RNIB. Error: %s",
140                         (*nbIdentity).GetInventoryName(), err.Error())
141                 return false
142         }
143
144         if handler.logger.DebugEnabled() {
145                 handler.logger.Debugf("#DeleteAllRequestHandler.saveNodebNextState - connection status of inventory name: %v changed to %v",
146                         (*nbIdentity).GetInventoryName(), nextStatus.String())
147         }
148         return true
149 }
150
151 func (handler *DeleteAllRequestHandler) saveNodebShutDownState(nbIdentity *entities.NbIdentity, node *entities.NodebInfo) bool {
152
153         if node.ConnectionStatus == entities.ConnectionStatus_SHUT_DOWN {
154                 return false
155         }
156
157         if node.ConnectionStatus != entities.ConnectionStatus_SHUTTING_DOWN {
158                 handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - ignore, status is not Shutting Down, inventory name: %v ", (*nbIdentity).GetInventoryName())
159                 return false
160         }
161
162         node.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
163
164         err := handler.rnibDataService.SaveNodeb(nbIdentity, node)
165
166         if err != nil {
167                 handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - failed to save nodeB entity for inventory name: %v to RNIB. Error: %s",
168                         (*nbIdentity).GetInventoryName(), err.Error())
169                 return false
170         }
171
172         handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - Shut Down , inventory name: %v ", (*nbIdentity).GetInventoryName())
173         return true
174 }