[RIC-475] Enhance LostConnection flow + UTs
[ric-plt/e2mgr.git] / E2Manager / handlers / rmrmsghandlers / e2_term_init_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 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18 //  platform project (RICP).
19
20 package rmrmsghandlers
21
22 import (
23         "e2mgr/clients"
24         "e2mgr/logger"
25         "e2mgr/managers"
26         "e2mgr/models"
27         "encoding/json"
28         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
29         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
30 )
31
32 type E2TermInitNotificationHandler struct {
33         logger                  *logger.Logger
34         ranDisconnectionManager *managers.RanDisconnectionManager
35         e2tInstancesManager     managers.IE2TInstancesManager
36         routingManagerClient    clients.IRoutingManagerClient
37 }
38
39 func NewE2TermInitNotificationHandler(logger *logger.Logger, ranDisconnectionManager *managers.RanDisconnectionManager, e2tInstancesManager managers.IE2TInstancesManager, routingManagerClient clients.IRoutingManagerClient) E2TermInitNotificationHandler {
40         return E2TermInitNotificationHandler{
41                 logger:                  logger,
42                 ranDisconnectionManager: ranDisconnectionManager,
43                 e2tInstancesManager:     e2tInstancesManager,
44                 routingManagerClient:    routingManagerClient,
45         }
46 }
47
48 func (h E2TermInitNotificationHandler) Handle(request *models.NotificationRequest) {
49         unmarshalledPayload := models.E2TermInitPayload{}
50         err := json.Unmarshal(request.Payload, &unmarshalledPayload)
51
52         if err != nil {
53                 h.logger.Errorf("#E2TermInitNotificationHandler.Handle - Error unmarshaling E2 Term Init payload: %s", err)
54                 return
55         }
56
57         e2tAddress := unmarshalledPayload.Address
58
59         if len(e2tAddress) == 0 {
60                 h.logger.Errorf("#E2TermInitNotificationHandler.Handle - Empty E2T address received")
61                 return
62         }
63
64         h.logger.Infof("#E2TermInitNotificationHandler.Handle - E2T payload: %s - handling E2_TERM_INIT", unmarshalledPayload)
65
66         e2tInstance, err := h.e2tInstancesManager.GetE2TInstance(e2tAddress)
67
68         if err != nil {
69                 _, ok := err.(*common.ResourceNotFoundError)
70
71                 if !ok {
72                         h.logger.Errorf("#E2TermInitNotificationHandler.Handle - Failed retrieving E2TInstance. error: %s", err)
73                         return
74                 }
75
76                 h.HandleNewE2TInstance(e2tAddress, unmarshalledPayload.PodName)
77                 return
78         }
79
80         if len(e2tInstance.AssociatedRanList) == 0 {
81                 h.logger.Infof("#E2TermInitNotificationHandler.Handle - E2T Address: %s - E2T instance has no associated RANs", e2tInstance.Address)
82                 return
83         }
84
85         if e2tInstance.State == entities.ToBeDeleted {
86                 h.logger.Infof("#E2TermInitNotificationHandler.Handle - E2T Address: %s - E2T instance status is: %s, ignore", e2tInstance.Address, e2tInstance.State)
87                 return
88         }
89
90         h.HandleExistingE2TInstance(e2tInstance)
91
92         h.logger.Infof("#E2TermInitNotificationHandler.Handle - Completed handling of E2_TERM_INIT")
93 }
94
95 func (h E2TermInitNotificationHandler) HandleExistingE2TInstance(e2tInstance *entities.E2TInstance) {
96
97         for _, ranName := range e2tInstance.AssociatedRanList {
98
99                 if err := h.ranDisconnectionManager.DisconnectRan(ranName); err != nil {
100                         if _, ok := err.(*common.ResourceNotFoundError); !ok {
101                                 break
102                         }
103                 }
104         }
105 }
106
107 func (h E2TermInitNotificationHandler) HandleNewE2TInstance(e2tAddress string, podName string) {
108
109         err := h.routingManagerClient.AddE2TInstance(e2tAddress)
110
111         if err != nil {
112                 h.logger.Errorf("#E2TermInitNotificationHandler.HandleNewE2TInstance - e2t address: %s - routing manager failure", e2tAddress)
113                 return
114         }
115
116         _ = h.e2tInstancesManager.AddE2TInstance(e2tAddress, podName)
117 }