[RICPLT-2146] Remove E2Sessions......

[ric-plt/e2mgr.git] / E2Manager / managers / ran_setup_manager.go

diff --git a/E2Manager/managers/ran_setup_manager.go b/E2Manager/managers/ran_setup_manager.go
index ce84468..89bdcc8 100644 (file)
--- a/E2Manager/managers/ran_setup_manager.go
+++ b/E2Manager/managers/ran_setup_manager.go
@@ -18,28 +18,101 @@
 package managers
 
 import (
+       "e2mgr/e2managererrors"
+       "e2mgr/e2pdus"
        "e2mgr/logger"
+       "e2mgr/models"
        "e2mgr/rNibWriter"
+       "e2mgr/rmrCgo"
        "e2mgr/services"
        "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
-       "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
 )
 
 type RanSetupManager struct {
        logger             *logger.Logger
-       rnibReaderProvider func() reader.RNibReader
        rnibWriterProvider func() rNibWriter.RNibWriter
        rmrService         *services.RmrService
 }
 
-func NewRanSetupManager(logger *logger.Logger, rmrService *services.RmrService, rnibReaderProvider func() reader.RNibReader, rnibWriterProvider func() rNibWriter.RNibWriter) *RanSetupManager {
+func NewRanSetupManager(logger *logger.Logger, rmrService *services.RmrService, rnibWriterProvider func() rNibWriter.RNibWriter) *RanSetupManager {
        return &RanSetupManager{
                logger:             logger,
-               rnibReaderProvider: rnibReaderProvider,
                rnibWriterProvider: rnibWriterProvider,
+               rmrService:         rmrService,
        }
 }
 
-func (m *RanSetupManager) ExecuteSetup(nodebInfo *entities.NodebInfo) error {
+// Update retries and connection status 
+func (m *RanSetupManager) updateConnectionStatus(nodebInfo *entities.NodebInfo, status entities.ConnectionStatus) error {
+       // Update retries and connection status
+       nodebInfo.ConnectionStatus = status
+       nodebInfo.ConnectionAttempts++
+       err := m.rnibWriterProvider().UpdateNodebInfo(nodebInfo)
+       if err != nil {
+               m.logger.Errorf("#RanSetupManager.updateConnectionStatus - Ran name: %s - Failed updating RAN's connection status to %v : %s", nodebInfo.RanName, status.String(), err)
+       } else {
+               m.logger.Infof("#RanSetupManager.updateConnectionStatus - Ran name: %s - Successfully updated rNib. RAN's current connection status: %v, RAN's current connection attempts: %d", nodebInfo.RanName, status.String(), nodebInfo.ConnectionAttempts)
+       }
+       return err
+}
+
+// Decrement retries and connection status (disconnected)
+func (m *RanSetupManager) updateConnectionStatusDisconnected(nodebInfo *entities.NodebInfo) error {
+       // Update retries and connection status
+       nodebInfo.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
+       nodebInfo.ConnectionAttempts--
+       err := m.rnibWriterProvider().UpdateNodebInfo(nodebInfo)
+       if err != nil {
+               m.logger.Errorf("#RanSetupManager.updateConnectionStatusDisconnected - Ran name: %s - Failed updating RAN's connection status to DISCONNECTED : %s", nodebInfo.RanName, err)
+       } else {
+               m.logger.Infof("#RanSetupManager.updateConnectionStatusDisconnected - Ran name: %s - Successfully updated rNib. RAN's current connection status: DISCONNECTED, RAN's current connection attempts: %d", nodebInfo.RanName, nodebInfo.ConnectionAttempts)
+       }
+       return err
+}
+
+func (m *RanSetupManager) prepareSetupRequest(nodebInfo *entities.NodebInfo) (int, *models.E2RequestMessage, error) {
+       // Build the endc/x2 setup request
+       switch nodebInfo.E2ApplicationProtocol {
+       case entities.E2ApplicationProtocol_X2_SETUP_REQUEST:
+               rmrMsgType := rmrCgo.RIC_X2_SETUP_REQ
+               request := models.NewE2RequestMessage(nodebInfo.RanName /*tid*/, nodebInfo.Ip, uint16(nodebInfo.Port), nodebInfo.RanName, e2pdus.PackedX2setupRequest)
+               return rmrMsgType, request, nil
+       case entities.E2ApplicationProtocol_ENDC_X2_SETUP_REQUEST:
+               rmrMsgType := rmrCgo.RIC_ENDC_X2_SETUP_REQ
+               request := models.NewE2RequestMessage(nodebInfo.RanName /*tid*/, nodebInfo.Ip, uint16(nodebInfo.Port), nodebInfo.RanName, e2pdus.PackedEndcX2setupRequest)
+               return rmrMsgType, request, nil
+       }
+
+       m.logger.Errorf("#RanSetupManager.prepareSetupRequest - Unsupported nodebInfo.E2ApplicationProtocol %d ", nodebInfo.E2ApplicationProtocol)
+       return 0, nil, e2managererrors.NewInternalError()
+}
+
+// ExecuteSetup updates the connection status and number of attempts in the nodebInfo and send an endc/x2 setup request to establish a connection with the RAN
+func (m *RanSetupManager) ExecuteSetup(nodebInfo *entities.NodebInfo, status entities.ConnectionStatus) error {
+
+       // Update retries and connection status
+       if err := m.updateConnectionStatus(nodebInfo, status); err != nil {
+               return e2managererrors.NewRnibDbError()
+       }
+
+       // Build the endc/x2 setup request
+       rmrMsgType, request, err := m.prepareSetupRequest(nodebInfo)
+       if err != nil {
+               return err
+       }
+
+       // Send the endc/x2 setup request
+       response := &models.NotificationResponse{MgsType: rmrMsgType, RanName: nodebInfo.RanName, Payload: request.GetMessageAsBytes(m.logger)}
+       if err := m.rmrService.SendRmrMessage(response); err != nil {
+               m.logger.Errorf("#RanSetupManager.ExecuteSetup - failed sending setup request to RMR: %s", err)
+
+               // Decrement retries and connection status (disconnected)
+               if err := m.updateConnectionStatusDisconnected(nodebInfo); err != nil {
+                       return e2managererrors.NewRnibDbError()
+               }
+
+               return e2managererrors.NewRmrError()
+       }
+
        return nil
 }