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