[RICPLT-1740] - Get Ran Load Information API

[ric-plt/nodeb-rnib.git] / reader / rNibReader.go

diff --git a/reader/rNibReader.go b/reader/rNibReader.go
index ae9dcf3..0509326 100644 (file)
--- a/reader/rNibReader.go
+++ b/reader/rNibReader.go
@@ -22,6 +22,7 @@ import (
        "gerrit.o-ran-sc.org/r/ric-plt/sdlgo"
        "github.com/golang/protobuf/proto"
        "github.com/pkg/errors"
+       "reflect"
 )
 
 var readerPool *common.Pool
@@ -53,6 +54,8 @@ type RNibReader interface {
        GetCellById(cellType entities.Cell_Type, cellId string) (*entities.Cell, common.IRNibError)
        // GetListNodebIds returns the full list of Nodeb identity entities
        GetListNodebIds()([]*entities.NbIdentity, common.IRNibError)
+       // GetRanLoadInformation retrieves nodeb load information entity from redis DB by nodeb inventory name
+       GetRanLoadInformation(inventoryName string) (*entities.RanLoadInformation, common.IRNibError)
 }
 
 const(
@@ -180,6 +183,34 @@ func (w *rNibReaderInstance) GetListNodebIds()([]*entities.NbIdentity, common.IR
        return *data, rnibErr
 }
 
+func (w *rNibReaderInstance) GetRanLoadInformation(inventoryName string) (*entities.RanLoadInformation, common.IRNibError){
+       key, rNibErr := common.ValidateAndBuildRanLoadInformationKey(inventoryName)
+       if rNibErr != nil {
+               return nil, rNibErr
+       }
+       loadInfo := &entities.RanLoadInformation{}
+       err := w.getByKeyAndUnmarshal(key, loadInfo)
+       if err!= nil{
+               return nil, err
+       }
+       return loadInfo, err
+}
+
+func (w *rNibReaderInstance) getByKeyAndUnmarshal(key string, entity proto.Message)common.IRNibError{
+       data, err := (*w.sdl).Get([]string{key})
+       if err != nil {
+               return common.NewInternalError(err)
+       }
+       if data != nil && data[key] != nil {
+               err = proto.Unmarshal([]byte(data[key].(string)), entity)
+               if err != nil {
+                       return common.NewInternalError(err)
+               }
+               return nil
+       }
+       return common.NewResourceNotFoundError(errors.Errorf("#rNibReader.getByKeyAndUnmarshal - entity of type %s not found. Key: %s", reflect.TypeOf(entity).String(), key))
+}
+
 func (w *rNibReaderInstance) getNodeb(key string) (*entities.NodebInfo, common.IRNibError) {
        data, err := (*w.sdl).Get([]string{key})
        if err != nil {