X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmanagers%2Fe2t_association_manager.go;h=8ed5ec1acfe7c83196919f8e411e3db8f37871f6;hb=76709204dfd7968156b5fa81190f146a8991812d;hp=822c01adf33a2c4456219e0d402237751ed58d77;hpb=011bb9161ca8e7ad0d0cf282b818672a88a5dfbf;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/managers/e2t_association_manager.go b/E2Manager/managers/e2t_association_manager.go
index 822c01a..8ed5ec1 100644
--- a/E2Manager/managers/e2t_association_manager.go
+++ b/E2Manager/managers/e2t_association_manager.go
@@ -21,24 +21,27 @@ package managers
 
 import (
 	"e2mgr/clients"
+	"e2mgr/e2managererrors"
 	"e2mgr/logger"
 	"e2mgr/services"
 	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
 )
 
 type E2TAssociationManager struct {
-	logger             *logger.Logger
-	rnibDataService    services.RNibDataService
-	e2tInstanceManager IE2TInstancesManager
-	rmClient           clients.IRoutingManagerClient
+	logger                        *logger.Logger
+	rnibDataService               services.RNibDataService
+	e2tInstanceManager            IE2TInstancesManager
+	rmClient                      clients.IRoutingManagerClient
+	ranConnectStatusChangeManager IRanConnectStatusChangeManager
 }
 
-func NewE2TAssociationManager(logger *logger.Logger, rnibDataService services.RNibDataService, e2tInstanceManager IE2TInstancesManager, rmClient clients.IRoutingManagerClient) *E2TAssociationManager {
+func NewE2TAssociationManager(logger *logger.Logger, rnibDataService services.RNibDataService, e2tInstanceManager IE2TInstancesManager, rmClient clients.IRoutingManagerClient, ranConnectStatusChangeManager IRanConnectStatusChangeManager) *E2TAssociationManager {
 	return &E2TAssociationManager{
-		logger:             logger,
-		rnibDataService:    rnibDataService,
-		e2tInstanceManager: e2tInstanceManager,
-		rmClient:           rmClient,
+		logger:                        logger,
+		rnibDataService:               rnibDataService,
+		e2tInstanceManager:            e2tInstanceManager,
+		rmClient:                      rmClient,
+		ranConnectStatusChangeManager: ranConnectStatusChangeManager,
 	}
 }
 
@@ -46,27 +49,43 @@ func (m *E2TAssociationManager) AssociateRan(e2tAddress string, nodebInfo *entit
 	ranName := nodebInfo.RanName
 	m.logger.Infof("#E2TAssociationManager.AssociateRan - Associating RAN %s to E2T Instance address: %s", ranName, e2tAddress)
 
-	err := m.rmClient.AssociateRanToE2TInstance(e2tAddress, ranName)
+	err := m.associateRanAndUpdateNodeb(e2tAddress, nodebInfo)
 	if err != nil {
-		m.logger.Errorf("#E2TAssociationManager.AssociateRan - RoutingManager failure: Failed to associate RAN %s to E2T %s. Error: %s", ranName, e2tAddress, err)
+		m.logger.Errorf("#E2TAssociationManager.AssociateRan - RoutingManager failure: Failed to associate RAN %s to E2T %s. Error: %s", nodebInfo, e2tAddress, err)
 		return err
 	}
+	err = m.e2tInstanceManager.AddRansToInstance(e2tAddress, []string{ranName})
+	if err != nil {
+		m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to add RAN to E2T instance %s. Error: %s", ranName, e2tAddress, err)
+		return e2managererrors.NewRnibDbError()
+	}
+	m.logger.Infof("#E2TAssociationManager.AssociateRan - successfully associated RAN %s with E2T %s", ranName, e2tAddress)
+	return nil
+}
 
-	nodebInfo.AssociatedE2TInstanceAddress = e2tAddress
-	nodebInfo.ConnectionAttempts = 0
+func (m *E2TAssociationManager) associateRanAndUpdateNodeb(e2tAddress string, nodebInfo *entities.NodebInfo) error {
+
+	rmErr := m.rmClient.AssociateRanToE2TInstance(e2tAddress, nodebInfo.RanName)
+
+	if rmErr != nil {
+		_ =  m.ranConnectStatusChangeManager.ChangeStatus(nodebInfo, entities.ConnectionStatus_DISCONNECTED)
+		return e2managererrors.NewRoutingManagerError()
+	}
+
+	rnibErr := m.ranConnectStatusChangeManager.ChangeStatus(nodebInfo, entities.ConnectionStatus_CONNECTED)
 
-	rnibErr := m.rnibDataService.UpdateNodebInfo(nodebInfo)
 	if rnibErr != nil {
-		m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to update RAN.AssociatedE2TInstanceAddress in rNib. Error: %s", ranName, rnibErr)
-		return rnibErr
+		return e2managererrors.NewRnibDbError()
 	}
 
-	err = m.e2tInstanceManager.AddRansToInstance(e2tAddress, []string{ranName})
-	if err != nil {
-		m.logger.Errorf("#E2TAssociationManager.AssociateRan - RAN name: %s - Failed to add RAN to E2T instance %s. Error: %s", ranName, e2tAddress, err)
-		return err
+	nodebInfo.AssociatedE2TInstanceAddress = e2tAddress
+	rnibErr = m.rnibDataService.UpdateNodebInfo(nodebInfo)
+
+	if rnibErr != nil {
+		m.logger.Errorf("#E2TAssociationManager.associateRanAndUpdateNodeb - RAN name: %s - Failed updating nodeb. Error: %s", nodebInfo.RanName, rnibErr)
+		return e2managererrors.NewRnibDbError()
 	}
-	m.logger.Infof("#E2TAssociationManager.AssociateRan - successfully associated RAN %s with E2T %s", ranName, e2tAddress)
+
 	return nil
 }