Merge R4 branch to master
[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         err := m.rmClient.AssociateRanToE2TInstance(e2tAddress, ranName)
50         if err != nil {
51                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RoutingManager failure: Failed to associate RAN %s to E2T %s. Error: %s", ranName, e2tAddress, err)
52                 return err
53         }
54
55         nodebInfo.AssociatedE2TInstanceAddress = e2tAddress
56         nodebInfo.ConnectionAttempts = 0
57
58         rnibErr := m.rnibDataService.UpdateNodebInfo(nodebInfo)
59         if rnibErr != nil {
60                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to update RAN.AssociatedE2TInstanceAddress in rNib. Error: %s", ranName, rnibErr)
61                 return rnibErr
62         }
63
64         err = m.e2tInstanceManager.AddRansToInstance(e2tAddress, []string{ranName})
65         if err != nil {
66                 m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to add RAN to E2T instance %s. Error: %s", ranName, e2tAddress, err)
67                 return err
68         }
69         m.logger.Infof("#E2TAssociationManager.AssociateRan - successfully associated RAN %s with E2T %s", ranName, e2tAddress)
70         return nil
71 }
72
73 func (m *E2TAssociationManager) DissociateRan(e2tAddress string, ranName string) error {
74         m.logger.Infof("#E2TAssociationManager.DissociateRan - Dissociating RAN %s from E2T Instance address: %s", ranName, e2tAddress)
75
76         nodebInfo, rnibErr := m.rnibDataService.GetNodeb(ranName)
77         if rnibErr != nil {
78                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed fetching RAN from rNib. Error: %s", ranName, rnibErr)
79                 return rnibErr
80         }
81
82         nodebInfo.AssociatedE2TInstanceAddress = ""
83         rnibErr = m.rnibDataService.UpdateNodebInfo(nodebInfo)
84         if rnibErr != nil {
85                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed to update RAN.AssociatedE2TInstanceAddress in rNib. Error: %s", ranName, rnibErr)
86                 return rnibErr
87         }
88
89         err := m.e2tInstanceManager.RemoveRanFromInstance(ranName, e2tAddress)
90         if err != nil {
91                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RAN name: %s - Failed to remove RAN from E2T instance %s. Error: %s", ranName, e2tAddress, err)
92                 return err
93         }
94
95         err = m.rmClient.DissociateRanE2TInstance(e2tAddress, ranName)
96         if err != nil {
97                 m.logger.Errorf("#E2TAssociationManager.DissociateRan - RoutingManager failure: Failed to dissociate RAN %s from E2T %s. Error: %s", ranName, e2tAddress, err)
98         } else {
99                 m.logger.Infof("#E2TAssociationManager.DissociateRan - successfully dissociated RAN %s from E2T %s", ranName, e2tAddress)
100         }
101         return nil
102 }
103
104 func (m *E2TAssociationManager) RemoveE2tInstance(e2tInstance *entities.E2TInstance) error {
105         m.logger.Infof("#E2TAssociationManager.RemoveE2tInstance -  Removing E2T %s and dessociating its associated RANs.", e2tInstance.Address)
106
107         err := m.rmClient.DeleteE2TInstance(e2tInstance.Address, e2tInstance.AssociatedRanList)
108         if err != nil {
109                 m.logger.Warnf("#E2TAssociationManager.RemoveE2tInstance - RoutingManager failure: Failed to delete E2T %s. Error: %s", e2tInstance.Address, err)
110                 // log and continue
111         }
112
113         err = m.e2tInstanceManager.RemoveE2TInstance(e2tInstance.Address)
114         if err != nil {
115                 m.logger.Errorf("#E2TAssociationManager.RemoveE2tInstance - Failed to remove E2T %s. Error: %s", e2tInstance.Address, err)
116                 return err
117         }
118
119         m.logger.Infof("#E2TAssociationManager.RemoveE2tInstance -  E2T %s successfully removed.", e2tInstance.Address)
120         return nil
121 }