[RICPLT-2165] Add rnibDataService to support retries
[ric-plt/e2mgr.git] / E2Manager / handlers / rmrmsghandlers / setup_response_notification_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 package rmrmsghandlers
18
19 import (
20         "e2mgr/logger"
21         "e2mgr/managers"
22         "e2mgr/models"
23         "e2mgr/services"
24         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
25 )
26
27 type SetupResponseNotificationHandler struct {
28         rnibDataService      services.RNibDataService
29         setupResponseManager managers.ISetupResponseManager
30         notificationType     string
31 }
32
33 func NewSetupResponseNotificationHandler(rnibDataService services.RNibDataService, setupResponseManager managers.ISetupResponseManager, notificationType string) SetupResponseNotificationHandler {
34         return SetupResponseNotificationHandler{
35                 rnibDataService:      rnibDataService,
36                 setupResponseManager: setupResponseManager,
37                 notificationType:     notificationType,
38         }
39 }
40
41 func (h SetupResponseNotificationHandler) Handle(logger *logger.Logger, request *models.NotificationRequest, messageChannel chan<- *models.NotificationResponse) {
42         logger.Infof("#SetupResponseNotificationHandler - RAN name: %s - Received %s notification", request.RanName, h.notificationType)
43
44         inventoryName := request.RanName
45
46         nodebInfo, rnibErr := h.rnibDataService.GetNodeb(inventoryName)
47
48         if rnibErr != nil {
49                 logger.Errorf("#SetupResponseNotificationHandler - RAN name: %s - Error fetching RAN from rNib: %v", request.RanName, rnibErr)
50                 return
51         }
52
53         if !isConnectionStatusValid(nodebInfo.ConnectionStatus) {
54                 logger.Errorf("#SetupResponseNotificationHandler - RAN name: %s - Invalid RAN connection status: %s", request.RanName, nodebInfo.ConnectionStatus)
55                 return
56         }
57
58         nodebInfo.ConnectionAttempts = 0
59         nbIdentity := &entities.NbIdentity{InventoryName: inventoryName}
60         err := h.setupResponseManager.PopulateNodebByPdu(logger, nbIdentity, nodebInfo, request.Payload)
61
62         if err != nil {
63                 return
64         }
65
66         rnibErr = h.rnibDataService.SaveNodeb(nbIdentity, nodebInfo)
67
68         if rnibErr != nil {
69                 logger.Errorf("#SetupResponseNotificationHandler - RAN name: %s - Error saving RAN to rNib: %v", request.RanName, rnibErr)
70                 return
71         }
72
73         logger.Infof("#SetupResponseNotificationHandler - RAN name: %s - Successfully saved RAN to rNib", request.RanName)
74 }
75
76 func isConnectionStatusValid(connectionStatus entities.ConnectionStatus) bool {
77         return connectionStatus == entities.ConnectionStatus_CONNECTING || connectionStatus == entities.ConnectionStatus_CONNECTED
78 }