RIC-11565:Add support for Multiple E2 Nodes: CU/DU for the case having same GNBId
[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                 h.UpdateExistingE2TInstanceToRtmgr(e2tAddress)
83                 return
84         }
85
86         if e2tInstance.State == entities.ToBeDeleted {
87                 h.logger.Infof("#E2TermInitNotificationHandler.Handle - E2T Address: %s - E2T instance status is: %s, ignore", e2tInstance.Address, e2tInstance.State)
88                 return
89         }
90
91         h.HandleExistingE2TInstance(e2tInstance)
92
93         h.logger.Infof("#E2TermInitNotificationHandler.Handle - Completed handling of E2_TERM_INIT")
94 }
95
96 func (h E2TermInitNotificationHandler) HandleExistingE2TInstance(e2tInstance *entities.E2TInstance) {
97
98         for _, ranName := range e2tInstance.AssociatedRanList {
99
100                 if err := h.ranDisconnectionManager.DisconnectRan(ranName); err != nil {
101                         if _, ok := err.(*common.ResourceNotFoundError); !ok {
102                                 break
103                         }
104                 }
105         }
106 }
107
108 func (h E2TermInitNotificationHandler) HandleNewE2TInstance(e2tAddress string, podName string) {
109
110     for i := 1; i < 4; i++ {
111
112         err := h.routingManagerClient.AddE2TInstance(e2tAddress)
113
114             if err == nil{
115
116                    _ = h.e2tInstancesManager.AddE2TInstance(e2tAddress, podName)
117
118                    break
119             } else {
120
121                 h.logger.Errorf("#E2TermInitNotificationHandler.HandleNewE2TInstance - e2t address count - %d : %s - routing manager failure", i, e2tAddress)
122             }
123
124         }
125 }
126
127 func (h E2TermInitNotificationHandler) UpdateExistingE2TInstanceToRtmgr(e2tAddress string) {
128
129                 _ = h.e2tInstancesManager.ResetKeepAliveTimestamp(e2tAddress)
130     for i := 1; i < 4; i++ {
131
132         err := h.routingManagerClient.AddE2TInstance(e2tAddress)
133
134             if err == nil{
135                    break
136             } else {
137
138                 h.logger.Errorf("#E2TermInitNotificationHandler.UpdateExistingE2TInstanceToRtmgr - e2t address count - %d : %s - routing manager failure", i, e2tAddress)
139             }
140
141         }
142
143 }
144