X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fservices%2Frnib_data_service.go;h=c053171ca340d9bca790a5eb90b158e84e4df42f;hb=refs%2Fchanges%2F20%2F4520%2F2;hp=e668da66ba82a5cadec5f2b59afe82514c4d0e42;hpb=5830d4db63151dd466533673fd76d842302f0034;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/services/rnib_data_service.go b/E2Manager/services/rnib_data_service.go
index e668da6..c053171 100644
--- a/E2Manager/services/rnib_data_service.go
+++ b/E2Manager/services/rnib_data_service.go
@@ -31,8 +31,9 @@ import (
 )
 
 type RNibDataService interface {
-	SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error
+	SaveNodeb(nodebInfo *entities.NodebInfo) error
 	UpdateNodebInfo(nodebInfo *entities.NodebInfo) error
+	UpdateNodebInfoAndPublish(nodebInfo *entities.NodebInfo) error
 	SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) error
 	GetNodeb(ranName string) (*entities.NodebInfo, error)
 	GetListNodebIds() ([]*entities.NbIdentity, error)
@@ -55,26 +56,52 @@ type RNibDataService interface {
 	RemoveEnb(nodebInfo *entities.NodebInfo) error
 	RemoveServedCells(inventoryName string, servedCells []*entities.ServedCellInfo) error
 	UpdateEnb(nodebInfo *entities.NodebInfo, servedCells []*entities.ServedCellInfo) error
+	AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error
+	RemoveNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error
+	AddEnb(nodebInfo *entities.NodebInfo) error
+	UpdateNbIdentity(nodeType entities.Node_Type, oldNbIdentities *entities.NbIdentity, newNbIdentities *entities.NbIdentity) error
 }
 
 type rNibDataService struct {
-	logger           *logger.Logger
-	rnibReader       reader.RNibReader
-	rnibWriter       rNibWriter.RNibWriter
-	maxAttempts      int
-	retryInterval    time.Duration
+	logger        *logger.Logger
+	rnibReader    reader.RNibReader
+	rnibWriter    rNibWriter.RNibWriter
+	maxAttempts   int
+	retryInterval time.Duration
 }
 
 func NewRnibDataService(logger *logger.Logger, config *configuration.Configuration, rnibReader reader.RNibReader, rnibWriter rNibWriter.RNibWriter) *rNibDataService {
 	return &rNibDataService{
-		logger:           logger,
-		rnibReader:       rnibReader,
-		rnibWriter:       rnibWriter,
-		maxAttempts:      config.MaxRnibConnectionAttempts,
-		retryInterval:    time.Duration(config.RnibRetryIntervalMs) * time.Millisecond,
+		logger:        logger,
+		rnibReader:    rnibReader,
+		rnibWriter:    rnibWriter,
+		maxAttempts:   config.MaxRnibConnectionAttempts,
+		retryInterval: time.Duration(config.RnibRetryIntervalMs) * time.Millisecond,
 	}
 }
 
+func (w *rNibDataService) AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error {
+	w.logger.Infof("#RnibDataService.AddNbIdentity - nbIdentity: %s", nbIdentity)
+
+	err := w.retry("AddNbIdentity", func() (err error) {
+		err = w.rnibWriter.AddNbIdentity(nodeType, nbIdentity)
+		return
+	})
+
+	return err
+}
+
+func (w *rNibDataService) RemoveNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error{
+	w.logger.Infof("#RnibDataService.RemoveNbIdentity - nbIdentity: %s", nbIdentity)
+
+	err := w.retry("RemoveNbIdentity", func() (err error) {
+		err = w.rnibWriter.RemoveNbIdentity(nodeType, nbIdentity)
+		return
+	})
+
+	return err
+}
+
 func (w *rNibDataService) RemoveServedNrCells(inventoryName string, servedNrCells []*entities.ServedNRCell) error {
 	err := w.retry("RemoveServedNrCells", func() (err error) {
 		err = w.rnibWriter.RemoveServedNrCells(inventoryName, servedNrCells)
@@ -85,6 +112,8 @@ func (w *rNibDataService) RemoveServedNrCells(inventoryName string, servedNrCell
 }
 
 func (w *rNibDataService) RemoveEnb(nodebInfo *entities.NodebInfo) error {
+	w.logger.Infof("#RnibDataService.RemoveEnb - nodebInfo: %s", nodebInfo)
+
 	err := w.retry("RemoveEnb", func() (err error) {
 		err = w.rnibWriter.RemoveEnb(nodebInfo)
 		return
@@ -115,11 +144,22 @@ func (w *rNibDataService) UpdateNodebInfo(nodebInfo *entities.NodebInfo) error {
 	return err
 }
 
-func (w *rNibDataService) SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error {
-	w.logger.Infof("#RnibDataService.SaveNodeb - nbIdentity: %s, nodebInfo: %s", nbIdentity, nb)
+func (w *rNibDataService) UpdateNodebInfoAndPublish(nodebInfo *entities.NodebInfo) error {
+	w.logger.Infof("#RnibDataService.UpdateNodebInfoAndPublish - nodebInfo: %s", nodebInfo)
+
+	err := w.retry("UpdateNodebInfoAndPublish", func() (err error) {
+		err = w.rnibWriter.UpdateNodebInfoAndPublish(nodebInfo)
+		return
+	})
+
+	return err
+}
+
+func (w *rNibDataService) SaveNodeb(nodebInfo *entities.NodebInfo) error {
+	w.logger.Infof("#RnibDataService.SaveNodeb - nodebInfo: %s", nodebInfo)
 
 	err := w.retry("SaveNodeb", func() (err error) {
-		err = w.rnibWriter.SaveNodeb(nbIdentity, nb)
+		err = w.rnibWriter.SaveNodeb(nodebInfo)
 		return
 	})
 
@@ -147,7 +187,7 @@ func (w *rNibDataService) GetNodeb(ranName string) (*entities.NodebInfo, error)
 	})
 
 	if err == nil {
-		w.logger.Infof("#RnibDataService.GetNodeb - RAN name: %s, connection status: %s, associated E2T: %s", nodeb.RanName, nodeb.ConnectionStatus, nodeb.AssociatedE2TInstanceAddress)
+		w.logger.Infof("#RnibDataService.GetNodeb - RAN name: %s, connection status: %s, associated E2T: %s, setup from network: %t", nodeb.RanName, nodeb.ConnectionStatus, nodeb.AssociatedE2TInstanceAddress, nodeb.SetupFromNetwork)
 	}
 
 	return nodeb, err
@@ -342,6 +382,26 @@ func (w *rNibDataService) UpdateNodebInfoOnConnectionStatusInversion(nodebInfo *
 	return err
 }
 
+func (w *rNibDataService) AddEnb(nodebInfo *entities.NodebInfo) error {
+	w.logger.Infof("#RnibDataService.AddEnb - nodebInfo: %s", nodebInfo)
+
+	err := w.retry("AddEnb", func() (err error) {
+		err = w.rnibWriter.AddEnb(nodebInfo)
+		return
+	})
+
+	return err
+}
+
+func (w *rNibDataService) UpdateNbIdentity(nodeType entities.Node_Type, oldNbIdentity *entities.NbIdentity, newNbIdentity *entities.NbIdentity) error {
+	err := w.retry("UpdateNbIdentities", func() (err error) {
+		err = w.rnibWriter.UpdateNbIdentities(nodeType, []*entities.NbIdentity{oldNbIdentity}, []*entities.NbIdentity{newNbIdentity})
+		return
+	})
+
+	return err
+}
+
 func (w *rNibDataService) retry(rnibFunc string, f func() error) (err error) {
 	attempts := w.maxAttempts