X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fxapp%2Fdb.go;h=e422cfde69ebbcb034c4c1ce6ef9799a893720f2;hb=40bc000e6cafe3a7eea32e4361268574050c12c4;hp=25c11c56168b646c5e5377a99ae635cc660b35a5;hpb=e17b3da504c590621e01d40afa9613771f172a94;p=ric-plt%2Fxapp-frame.git

diff --git a/pkg/xapp/db.go b/pkg/xapp/db.go
index 25c11c5..e422cfd 100755
--- a/pkg/xapp/db.go
+++ b/pkg/xapp/db.go
@@ -23,9 +23,6 @@ import (
 	rnibentities "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
 	rnibreader "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
 	sdl "gerrit.o-ran-sc.org/r/ric-plt/sdlgo"
-	uenibprotobuf "gerrit.o-ran-sc.org/r/ric-plt/ue-nib/uenibprotobuf"
-	uenibreader "gerrit.o-ran-sc.org/r/ric-plt/ue-nib/uenibreader"
-	uenibwriter "gerrit.o-ran-sc.org/r/ric-plt/ue-nib/uenibwriter"
 	rnibwriter "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/rnib"
 	"sync"
 	"time"
@@ -46,19 +43,6 @@ type SDLClient struct {
 	ready bool
 }
 
-// Alias
-type EventCategory = uenibreader.EventCategory
-type EventCallback = uenibreader.EventCallback
-type MeasResultNR = uenibprotobuf.MeasResultNR
-type MeasQuantityResults = uenibprotobuf.MeasResultNR_MeasQuantityResults
-type MeasResultServMO = uenibprotobuf.MeasResults_MeasResultServMO
-type MeasResults = uenibprotobuf.MeasResults
-
-type UENIBClient struct {
-	reader *uenibreader.Reader
-	writer *uenibwriter.Writer
-}
-
 // Alias
 type RNIBNodeType = rnibentities.Node_Type
 type RNIBGlobalNbId = rnibentities.GlobalNbId
@@ -82,12 +66,16 @@ type RNIBServedNRCellInformation = rnibentities.ServedNRCellInformation
 type RNIBNrNeighbourInformation = rnibentities.NrNeighbourInformation
 
 type RNIBClient struct {
+	db     *sdl.SdlInstance
 	reader rnibreader.RNibReader
 	writer rnibwriter.RNibWriter
 }
 
 // NewSDLClient returns a new SDLClient.
 func NewSDLClient(ns string) *SDLClient {
+	if ns == "" {
+		ns = "sdl"
+	}
 	return &SDLClient{
 		db:    sdl.NewSdlInstance(ns, sdl.NewDatabase()),
 		stat:  Metric.RegisterCounterGroup(SDLCounterOpts, "SDL"),
@@ -180,74 +168,55 @@ func (c *SDLClient) GetStat() (t SDLStatistics) {
 	return
 }
 
-func NewUENIBClient() *UENIBClient {
-	return &UENIBClient{
-		reader: uenibreader.NewReader(),
-		writer: uenibwriter.NewWriter(),
+func NewRNIBClient() *RNIBClient {
+	s := sdl.NewSdlInstance("e2Manager", sdl.NewDatabase())
+	return &RNIBClient{
+		db:     s,
+		reader: nil,
+		writer: nil,
 	}
 }
 
-func (u *UENIBClient) StoreUeMeasurement(gNbId string, gNbUeX2ApId string, data *uenibprotobuf.MeasResults) error {
-	return u.writer.UpdateUeMeasurement(gNbId, gNbUeX2ApId, data)
-}
-
-func (u *UENIBClient) CreateUeContext(gNbId string, gNbUeX2ApId string) error {
-	return u.writer.UeContextAddComplete(gNbId, gNbUeX2ApId)
+func (r *RNIBClient) Subscribe(cb func(string, ...string), channel string) error {
+	return r.db.SubscribeChannel(cb, channel)
 }
 
-func (u *UENIBClient) ReleaseUeContext(gNbId string, gNbUeX2ApId string) error {
-	return u.writer.RemoveUeContext(gNbId, gNbUeX2ApId)
-}
-
-func (u *UENIBClient) ReadUeMeasurement(gNbId string, gNbUeX2ApId string) (*uenibprotobuf.MeasResults, error) {
-	return u.reader.GetUeMeasurement(gNbId, gNbUeX2ApId)
-}
-
-func (u *UENIBClient) SubscribeEvents(gNbIDs []string, eventCategories []EventCategory, cb EventCallback) error {
-	return u.reader.SubscribeEvents(gNbIDs, eventCategories, cb)
-}
-
-func NewRNIBClient(ns string) *RNIBClient {
-	rnibreader.Init("rnib", 1)
-	rnibwriter.InitWriter("rnib", 1)
-	return &RNIBClient{
-		reader: nil,
-		writer: nil,
-	}
+func (r *RNIBClient) StoreAndPublish(channel string, event string, pairs ...interface{}) error {
+	return r.db.SetAndPublish([]string{channel, event}, pairs...)
 }
 
 func (r *RNIBClient) GetNodeb(invName string) (*RNIBNodebInfo, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetNodeb(invName)
+	return rnibreader.GetRNibReader(r.db).GetNodeb(invName)
 }
 
 func (r *RNIBClient) GetNodebByGlobalNbId(t RNIBNodeType, gid *RNIBGlobalNbId) (*RNIBNodebInfo, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetNodebByGlobalNbId(t, gid)
+	return rnibreader.GetRNibReader(r.db).GetNodebByGlobalNbId(t, gid)
 }
 
 func (r *RNIBClient) GetCellList(invName string) (*RNIBCells, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetCellList(invName)
+	return rnibreader.GetRNibReader(r.db).GetCellList(invName)
 }
 
 func (r *RNIBClient) GetListGnbIds() ([]*RNIBNbIdentity, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetListGnbIds()
+	return rnibreader.GetRNibReader(r.db).GetListGnbIds()
 }
 
 func (r *RNIBClient) GetListEnbIds() ([]*RNIBNbIdentity, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetListEnbIds()
+	return rnibreader.GetRNibReader(r.db).GetListEnbIds()
 }
 
 func (r *RNIBClient) GetCountGnbList() (int, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetCountGnbList()
+	return rnibreader.GetRNibReader(r.db).GetCountGnbList()
 }
 
 func (r *RNIBClient) GetCell(invName string, pci uint32) (*RNIBCell, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetCell(invName, pci)
+	return rnibreader.GetRNibReader(r.db).GetCell(invName, pci)
 }
 
 func (r *RNIBClient) GetCellById(cellType RNIBCellType, cellId string) (*RNIBCell, RNIBIRNibError) {
-	return rnibreader.GetRNibReader().GetCellById(cellType, cellId)
+	return rnibreader.GetRNibReader(r.db).GetCellById(cellType, cellId)
 }
 
 func (r *RNIBClient) SaveNodeb(nbIdentity *RNIBNbIdentity, entity *RNIBNodebInfo) RNIBIRNibError {
-	return rnibwriter.GetRNibWriter().SaveNodeb(nbIdentity, entity)
+	return rnibwriter.GetRNibWriter(r.db).SaveNodeb(nbIdentity, entity)
 }