Merge "Automation of nodeb health check"
[ric-plt/e2mgr.git] / E2Manager / managers / ran_list_manager.go
1 //
2 // Copyright 2019 AT&T Intellectual Property
3 // Copyright 2019 Nokia
4 // Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved.
5 //
6 // Licensed under the Apache License, Version 2.0 (the "License");
7 // you may not use this file except in compliance with the License.
8 // You may obtain a copy of the License at
9 //
10 //      http://www.apache.org/licenses/LICENSE-2.0
11 //
12 // Unless required by applicable law or agreed to in writing, software
13 // distributed under the License is distributed on an "AS IS" BASIS,
14 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 // See the License for the specific language governing permissions and
16 // limitations under the License.
17
18 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
19 //  platform project (RICP).
20
21 package managers
22
23 import (
24         "e2mgr/e2managererrors"
25         "e2mgr/logger"
26         "e2mgr/services"
27         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
28         "sync"
29         "time"
30 )
31
32 type ranListManagerInstance struct {
33         logger          *logger.Logger
34         rnibDataService services.RNibDataService
35         mux             sync.Mutex
36         nbIdentityMap   map[string]*entities.NbIdentity
37 }
38
39 type RanListManager interface {
40         InitNbIdentityMap() error
41         AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error
42         UpdateNbIdentityConnectionStatus(nodeType entities.Node_Type, ranName string, connectionStatus entities.ConnectionStatus) error
43         RemoveNbIdentity(nodeType entities.Node_Type, ranName string) error
44         GetNbIdentityList() []*entities.NbIdentity
45         GetNbIdentity(ranName string) (*entities.NbIdentity, error)
46         UpdateHealthcheckTimeStampReceived(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity)
47         UpdateHealthcheckTimeStampSent(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity)
48         UpdateNbIdentities(nodeType entities.Node_Type, oldNbIdentities []*entities.NbIdentity, newNbIdentities []*entities.NbIdentity) error
49 }
50
51 func NewRanListManager(logger *logger.Logger, rnibDataService services.RNibDataService) RanListManager {
52         return &ranListManagerInstance{
53                 logger:          logger,
54                 rnibDataService: rnibDataService,
55                 nbIdentityMap:   make(map[string]*entities.NbIdentity),
56         }
57 }
58
59 func (m *ranListManagerInstance) InitNbIdentityMap() error {
60         nbIds, err := m.rnibDataService.GetListNodebIds()
61
62         if err != nil {
63                 m.logger.Errorf("#ranListManagerInstance.InitNbIdentityMap - Failed fetching RAN list from DB. error: %s", err)
64                 return err
65         }
66
67         for _, v := range nbIds {
68                 m.nbIdentityMap[v.InventoryName] = v
69         }
70
71         m.logger.Infof("#ranListManagerInstance.InitNbIdentityMap - Successfully initiated nodeb identity map")
72         m.logger.Debugf("#ranListManagerInstance.InitNbIdentityMap - nodeb Identity map: %s", m.nbIdentityMap)
73         return nil
74 }
75
76 func (m *ranListManagerInstance) AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error {
77         m.mux.Lock()
78         defer m.mux.Unlock()
79
80         m.nbIdentityMap[nbIdentity.InventoryName] = nbIdentity
81
82         err := m.rnibDataService.AddNbIdentity(nodeType, nbIdentity)
83
84         if err != nil {
85                 m.logger.Errorf("#ranListManagerInstance.AddNbIdentity - RAN name: %s - Failed adding nodeb identity to DB. error: %s", nbIdentity.InventoryName, err)
86                 return err
87         }
88
89         m.logger.Infof("#ranListManagerInstance.AddNbIdentity - RAN name: %s - Successfully added nodeb identity", nbIdentity.InventoryName)
90         m.logger.Debugf("#ranListManagerInstance.AddNbIdentity - nodeb Identity map: %s", m.nbIdentityMap)
91         return nil
92 }
93
94 func (m *ranListManagerInstance) UpdateNbIdentityConnectionStatus(nodeType entities.Node_Type, ranName string, connectionStatus entities.ConnectionStatus) error {
95         m.mux.Lock()
96         defer m.mux.Unlock()
97
98         m.logger.Infof("#ranListManagerInstance.UpdateNbIdentityConnectionStatus - RAN name: %s - updating nodeb identity connection status", ranName)
99
100         oldNbIdentity, ok := m.nbIdentityMap[ranName]
101         if !ok {
102                 m.logger.Errorf("#ranListManagerInstance.UpdateNbIdentityConnectionStatus - RAN name: %s - nodeb identity not found in nbIdentityMap", ranName)
103                 return e2managererrors.NewInternalError()
104         }
105
106         newNbIdentity := &entities.NbIdentity{
107                 GlobalNbId:       oldNbIdentity.GlobalNbId,
108                 InventoryName:    ranName,
109                 ConnectionStatus: connectionStatus,
110                 HealthCheckTimestampSent: oldNbIdentity.HealthCheckTimestampSent,
111                 HealthCheckTimestampReceived: oldNbIdentity.HealthCheckTimestampReceived,
112         }
113         m.nbIdentityMap[ranName] = newNbIdentity
114
115         err := m.rnibDataService.UpdateNbIdentity(nodeType, oldNbIdentity, newNbIdentity)
116         if err != nil {
117                 m.logger.Errorf("#ranListManagerInstance.UpdateNbIdentityConnectionStatus - RAN name: %s - Failed updating nodeb identity in DB. error: %s", ranName, err)
118                 return err
119         }
120         m.logger.Infof("#ranListManagerInstance.UpdateNbIdentityConnectionStatus - RAN name: %s - Successfully updated nodeb identity", ranName)
121         return nil
122 }
123
124 func (m *ranListManagerInstance) RemoveNbIdentity(nodeType entities.Node_Type, ranName string) error {
125         m.mux.Lock()
126         defer m.mux.Unlock()
127
128         m.logger.Infof("#ranListManagerInstance.RemoveNbIdentity - RAN name: %s - deleting nodeb identity from memory and db...", ranName)
129
130         nbIdentity, ok := m.nbIdentityMap[ranName]
131         if !ok {
132                 m.logger.Infof("#ranListManagerInstance.RemoveNbIdentity - RAN name: %s - nodeb identity not found", ranName)
133                 return nil
134         }
135
136         delete(m.nbIdentityMap, ranName)
137
138         err := m.rnibDataService.RemoveNbIdentity(nodeType, nbIdentity)
139         if err != nil {
140                 m.logger.Errorf("#ranListManagerInstance.RemoveNbIdentity - RAN name: %s - Failed removing nodeb identity from DB. error: %s", ranName, err)
141                 return err
142         }
143
144         m.logger.Infof("#ranListManagerInstance.RemoveNbIdentity - RAN name: %s - Successfully deleted nodeb identity", ranName)
145         return nil
146 }
147
148 func (m *ranListManagerInstance) GetNbIdentityList() []*entities.NbIdentity {
149         nbIds := make([]*entities.NbIdentity, 0, len(m.nbIdentityMap))
150         for _, v := range m.nbIdentityMap {
151                 nbIds = append(nbIds, v)
152         }
153
154         m.logger.Infof("#ranListManagerInstance.GetNbIdentityList - %d identity returned", len(nbIds))
155
156         return nbIds
157 }
158
159 func (m *ranListManagerInstance) GetNbIdentity(ranName string) (*entities.NbIdentity, error) {
160         nbIdentity, ok := m.nbIdentityMap[ranName]
161         if !ok {
162                 m.logger.Infof("#ranListManagerInstance.GetNbIdentity - RAN name: %s - nodeb identity not found", ranName)
163                 return nil , e2managererrors.NewResourceNotFoundError()
164         }
165
166         m.logger.Infof("#ranListManagerInstance.GetNbIdentity - RAN name: %s - nodeb identity returned", ranName)
167
168         return nbIdentity, nil
169 }
170
171 func (m *ranListManagerInstance) UpdateHealthcheckTimeStampSent(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity){
172         currentTimeStamp := time.Now().UnixNano()
173         oldNbIdentity := m.nbIdentityMap[oldRRanName]
174
175         newNbIdentity := &entities.NbIdentity{
176                 GlobalNbId:       oldNbIdentity.GlobalNbId,
177                 InventoryName:    oldNbIdentity.InventoryName,
178                 ConnectionStatus: oldNbIdentity.ConnectionStatus,
179                 HealthCheckTimestampSent: currentTimeStamp,
180                 HealthCheckTimestampReceived: oldNbIdentity.HealthCheckTimestampReceived,
181         }
182
183         m.nbIdentityMap[oldNbIdentity.InventoryName] = newNbIdentity
184         return oldNbIdentity, newNbIdentity
185 }
186
187 func (m *ranListManagerInstance) UpdateHealthcheckTimeStampReceived(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity){
188         currentTimeStamp := time.Now().UnixNano()
189         oldNbIdentity := m.nbIdentityMap[oldRRanName]
190
191         newNbIdentity := &entities.NbIdentity{
192                 GlobalNbId:       oldNbIdentity.GlobalNbId,
193                 InventoryName:    oldNbIdentity.InventoryName,
194                 ConnectionStatus: oldNbIdentity.ConnectionStatus,
195                 HealthCheckTimestampSent: oldNbIdentity.HealthCheckTimestampSent,
196                 HealthCheckTimestampReceived: currentTimeStamp,
197         }
198
199         m.nbIdentityMap[oldNbIdentity.InventoryName] = newNbIdentity
200         return oldNbIdentity, newNbIdentity
201 }
202
203 func (m *ranListManagerInstance) UpdateNbIdentities(nodeType entities.Node_Type, oldNbIdentities []*entities.NbIdentity, newNbIdentities []*entities.NbIdentity) error {
204         m.mux.Lock()
205         defer m.mux.Unlock()
206
207         err := m.rnibDataService.UpdateNbIdentities(nodeType, oldNbIdentities, newNbIdentities)
208
209         if err != nil {
210                 m.logger.Errorf("#ranListManagerInstance.UpdateNbIdentities not completed for %d nbIdentities of nodetype - %s", len(newNbIdentities), nodeType.String())
211         } else{
212                 m.logger.Infof("#ranListManagerInstance.UpdateNbIdentities completed successfully for %d nbIdentities of nodetype - %s", len(newNbIdentities), nodeType.String())
213         }
214
215         return err
216 }