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