855ac05094788e62d289c18a8273748151146691
[ric-plt/e2mgr.git] / E2Manager / managers / e2t_association_manager.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 managers
21
22 import (
23         "e2mgr/clients"
24         "e2mgr/logger"
25         "e2mgr/services"
26         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
27 )
28
29 type E2TAssociationManager struct {
30         logger             *logger.Logger
31         rnibDataService    services.RNibDataService
32         e2tInstanceManager IE2TInstancesManager
33         rmClient           clients.IRoutingManagerClient
34 }
35
36 func NewE2TAssociationManager(logger *logger.Logger, rnibDataService services.RNibDataService, e2tInstanceManager IE2TInstancesManager, rmClient clients.IRoutingManagerClient) *E2TAssociationManager {
37         return &E2TAssociationManager{
38                 logger:             logger,
39                 rnibDataService:    rnibDataService,
40                 e2tInstanceManager: e2tInstanceManager,
41                 rmClient:           rmClient,
42         }
43 }
44
45 func (m *E2TAssociationManager) AssociateRan(e2tAddress string, nodebInfo *entities.NodebInfo) error {
46         ranName := nodebInfo.RanName
47         m.logger.Infof("#E2TAssociationManager.AssociateRan - Associating RAN %s to E2T Instance address: %s", ranName, e2tAddress)
48
49         nodebInfo.AssociatedE2TInstanceAddress = e2tAddress
50         nodebInfo.ConnectionAttempts = 0
51
52         rnibErr := m.rnibDataService.UpdateNodebInfo(nodebInfo)
53         if rnibErr != nil {
54                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to update RAN.AssociatedE2TInstanceAddress in rNib. Error: %s", ranName, rnibErr)
55                 return rnibErr
56         }
57
58         err := m.e2tInstanceManager.AddRansToInstance(e2tAddress, []string{ranName})
59         if err != nil {
60                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to add RAN to E2T instance %s. Error: %s", ranName, e2tAddress, err)
61                 return err
62         }
63         err = m.rmClient.AssociateRanToE2TInstance(e2tAddress, ranName)
64         if err != nil {
65                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RoutingManager failure: Failed to associate RAN %s to E2T %s. Error: %s", ranName, e2tAddress, err)
66         }
67         m.logger.Infof("#E2TAssociationManager.AssociateRan - successfully associated RAN %s with E2T %s", ranName, e2tAddress)
68         return nil
69 }
70
71 func (m *E2TAssociationManager) DissociateRan(e2tAddress string, ranName string) error {
72         m.logger.Infof("#E2TAssociationManager.DissociateRan - Dissociating RAN %s from E2T Instance address: %s", ranName, e2tAddress)
73
74         nodebInfo, rnibErr := m.rnibDataService.GetNodeb(ranName)
75         if rnibErr != nil {
76                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed fetching RAN from rNib. Error: %s", ranName, rnibErr)
77                 return rnibErr
78         }
79
80         nodebInfo.AssociatedE2TInstanceAddress = ""
81         rnibErr = m.rnibDataService.UpdateNodebInfo(nodebInfo)
82         if rnibErr != nil {
83                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed to update RAN.AssociatedE2TInstanceAddress in rNib. Error: %s", ranName, rnibErr)
84                 return rnibErr
85         }
86
87         err := m.e2tInstanceManager.RemoveRanFromInstance(ranName, e2tAddress)
88         if err != nil {
89                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed to remove RAN from E2T instance %s. Error: %s", ranName, e2tAddress, err)
90                 return err
91         }
92
93         err = m.rmClient.DissociateRanE2TInstance(e2tAddress, ranName)
94         if err != nil {
95                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RoutingManager failure: Failed to dissociate RAN %s from E2T %s. Error: %s", ranName, e2tAddress, err)
96         } else {
97                 m.logger.Infof("#E2TAssociationManager.DissociateRan - successfully dissociated RAN %s from E2T %s", ranName, e2tAddress)
98         }
99         return nil
100 }
101
102 func (m *E2TAssociationManager) RemoveE2tInstance(e2tInstance *entities.E2TInstance) error {
103         m.logger.Infof("#E2TAssociationManager.RemoveE2tInstance -  Removing E2T %s and dessociating its associated RANs.", e2tInstance.Address)
104
105         err := m.rmClient.DeleteE2TInstance(e2tInstance.Address, e2tInstance.AssociatedRanList)
106         if err != nil {
107                 m.logger.Warnf("#E2TAssociationManager.RemoveE2tInstance - RoutingManager failure: Failed to delete E2T %s. Error: %s", e2tInstance.Address, err)
108                 // log and continue
109         }
110
111         err = m.e2tInstanceManager.RemoveE2TInstance(e2tInstance.Address)
112         if err != nil {
113                 m.logger.Errorf("#E2TAssociationManager.RemoveE2tInstance - Failed to remove E2T %s. Error: %s", e2tInstance.Address, err)
114                 return err
115         }
116
117         m.logger.Infof("#E2TAssociationManager.RemoveE2tInstance -  E2T %s successfully removed.", e2tInstance.Address)
118         return nil
119 }