X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Fregistry.go;h=29d82a9220447caadcd708d26ce87d86175cb5bb;hb=d13ba63c1c237882202fc975d70ed5ad21a48fcf;hp=8c795406aa4d1935581f6c71845817c8b0b156a3;hpb=c7da4ee19c2b2a96c1eeccd8da944902fcd9323f;p=ric-plt%2Fsubmgr.git diff --git a/pkg/control/registry.go b/pkg/control/registry.go index 8c79540..29d82a9 100644 --- a/pkg/control/registry.go +++ b/pkg/control/registry.go @@ -36,6 +36,8 @@ import ( //----------------------------------------------------------------------------- type RESTSubscription struct { + Created string + xAppServiceName string xAppRmrEndPoint string Meid string InstanceIds []uint32 @@ -89,7 +91,7 @@ func (r *RESTSubscription) SetProcessed(err error) { } type Registry struct { - mutex sync.Mutex + mutex *sync.Mutex register map[uint32]*Subscription subIds []uint32 rtmgrClient *RtmgrClient @@ -97,6 +99,7 @@ type Registry struct { } func (r *Registry) Initialize() { + r.mutex = new(sync.Mutex) r.register = make(map[uint32]*Subscription) r.restSubscriptions = make(map[string]*RESTSubscription) @@ -106,20 +109,132 @@ func (r *Registry) Initialize() { } } -func (r *Registry) GetAllRestSubscriptions() []byte { +func (r *Registry) GetAllRestSubscriptionsJson() []byte { + r.mutex.Lock() defer r.mutex.Unlock() restSubscriptionsJson, err := json.Marshal(r.restSubscriptions) if err != nil { - xapp.Logger.Error("GetAllRestSubscriptions(): %v", err) + xapp.Logger.Error("GetAllRestSubscriptions() json.Marshal error: %v", err) } return restSubscriptionsJson } -func (r *Registry) CreateRESTSubscription(restSubId *string, xAppRmrEndPoint *string, maid *string) *RESTSubscription { +func (r *Registry) GetAllE2NodeRestSubscriptionsJson(ranName string) []byte { + + restSubscriptions := r.GetAllE2NodeRestSubscriptions(ranName) + e2NodeRestSubscriptionsJson, err := json.Marshal(restSubscriptions) + if err != nil { + xapp.Logger.Error("GetE2NodeRestSubscriptions() json.Marshal error: %v", err) + } + return e2NodeRestSubscriptionsJson +} + +func (r *Registry) GetAllE2NodeRestSubscriptions(ranName string) map[string]RESTSubscription { + + r.mutex.Lock() + defer r.mutex.Unlock() + var restSubscriptions map[string]RESTSubscription + restSubscriptions = make(map[string]RESTSubscription) + for restSubsId, restSubscription := range r.restSubscriptions { + if restSubscription.Meid == ranName { + restSubscriptions[restSubsId] = *restSubscription + } + } + return restSubscriptions +} + +func (r *Registry) GetAllXappsJson() []byte { + + r.mutex.Lock() + var xappList []string + var xappsMap map[string]string + xappsMap = make(map[string]string) + for _, restSubscription := range r.restSubscriptions { + _, ok := xappsMap[restSubscription.xAppServiceName] + if !ok { + xappsMap[restSubscription.xAppServiceName] = restSubscription.xAppServiceName + xappList = append(xappList, restSubscription.xAppServiceName) + } + } + r.mutex.Unlock() + + xappsJson, err := json.Marshal(xappList) + if err != nil { + xapp.Logger.Error("GetXapps() json.Marshal error: %v", err) + } + return xappsJson +} + +func (r *Registry) GetAllXapps() map[string]string { + + r.mutex.Lock() + defer r.mutex.Unlock() + var xappsMap map[string]string + xappsMap = make(map[string]string) + for _, restSubscription := range r.restSubscriptions { + _, ok := xappsMap[restSubscription.xAppServiceName] + if !ok { + xappsMap[restSubscription.xAppServiceName] = restSubscription.xAppServiceName + } + } + return xappsMap +} + +func (r *Registry) GetAllXappRestSubscriptionsJson(xAppServiceName string) []byte { + + xappRestSubscriptions := r.GetAllXappRestSubscriptions(xAppServiceName) + xappRestSubscriptionsJson, err := json.Marshal(xappRestSubscriptions) + if err != nil { + xapp.Logger.Error("GetXappRestSubscriptions() json.Marshal error: %v", err) + } + return xappRestSubscriptionsJson +} + +func (r *Registry) GetAllXappRestSubscriptions(xAppServiceName string) map[string]RESTSubscription { + + r.mutex.Lock() + defer r.mutex.Unlock() + var xappRestSubscriptions map[string]RESTSubscription + xappRestSubscriptions = make(map[string]RESTSubscription) + for restSubsId, xappRestSubscription := range r.restSubscriptions { + if xappRestSubscription.xAppServiceName == xAppServiceName { + xappRestSubscriptions[restSubsId] = *xappRestSubscription + } + } + return xappRestSubscriptions +} + +func (r *Registry) GetE2SubscriptionsJson(restSubsId string) ([]byte, error) { + + // Get all E2 subscriptions of a REST subscription + restSubs, err := r.GetRESTSubscription(restSubsId, false) + if err != nil { + return nil, err + } + + r.mutex.Lock() + var e2Subscriptions []Subscription + for _, e2SubId := range restSubs.InstanceIds { + e2Subscription, ok := r.register[e2SubId] + if ok { + e2Subscriptions = append(e2Subscriptions, *e2Subscription) + } + } + r.mutex.Unlock() + e2SubscriptionsJson, err := json.Marshal(e2Subscriptions) + if err != nil { + xapp.Logger.Error("GetE2Subscriptions() json.Marshal error: %v", err) + } + return e2SubscriptionsJson, nil +} + +func (r *Registry) CreateRESTSubscription(restSubId *string, xappServiceName *string, xAppRmrEndPoint *string, maid *string) *RESTSubscription { r.mutex.Lock() defer r.mutex.Unlock() newRestSubscription := RESTSubscription{} + newRestSubscription.Created = time.Now().Format("2006-01-02 15:04:05.000") + newRestSubscription.xAppServiceName = *xappServiceName newRestSubscription.xAppRmrEndPoint = *xAppRmrEndPoint newRestSubscription.Meid = *maid newRestSubscription.SubReqOngoing = true @@ -396,9 +511,9 @@ func (r *Registry) CheckActionTypes(subReqMsg *e2ap.E2APSubscriptionRequest) (ui return e2ap.E2AP_ActionTypeInvalid, fmt.Errorf("Invalid action type in RICactions-ToBeSetup-List") } -// TODO: Works with concurrent calls, but check if can be improved func (r *Registry) RemoveFromSubscription(subs *Subscription, trans *TransactionXapp, waitRouteClean time.Duration, c *Control) error { + xapp.Logger.Debug("RemoveFromSubscription %s", idstring(nil, trans, subs, trans)) r.mutex.Lock() defer r.mutex.Unlock() subs.mutex.Lock() @@ -406,51 +521,55 @@ func (r *Registry) RemoveFromSubscription(subs *Subscription, trans *Transaction delStatus := subs.EpList.DelEndpoint(trans.GetEndpoint()) epamount := subs.EpList.Size() + subId := subs.ReqId.InstanceId if delStatus == false { return nil } - go func() { - if waitRouteClean > 0 { - xapp.Logger.Debug("Pending %v in order to wait route cleanup", waitRouteClean) - time.Sleep(waitRouteClean) - } + if waitRouteClean > 0 { + // Wait here that response is delivered to xApp via RMR before route is cleaned + xapp.Logger.Debug("Pending %v in order to wait route cleanup", waitRouteClean) + time.Sleep(waitRouteClean) + } - subs.mutex.Lock() - defer subs.mutex.Unlock() - xapp.Logger.Debug("CLEAN %s", subs.String()) + xapp.Logger.Debug("CLEAN %s", subs.String()) - if epamount == 0 { - // - // Subscription route delete - // - if subs.RMRRouteCreated == true { - r.RouteDelete(subs, trans, c) - } + if epamount == 0 { + // + // Subscription route delete + // + if subs.RMRRouteCreated == true { + r.RouteDelete(subs, trans, c) + } - // - // Subscription release - // - r.mutex.Lock() - defer r.mutex.Unlock() + // Not merged subscription is being deleted + xapp.Logger.Debug("Subscription route delete RemoveSubscriptionFromDb") + c.RemoveSubscriptionFromDb(subs) - if _, ok := r.register[subId]; ok { - xapp.Logger.Debug("RELEASE %s", subs.String()) - delete(r.register, subId) - xapp.Logger.Debug("Registry: substable=%v", r.register) - } - r.subIds = append(r.subIds, subId) - } else if subs.EpList.Size() > 0 { - // - // Subscription route update - // - if subs.RMRRouteCreated == true { - r.RouteDeleteUpdate(subs, c) - } + // + // Subscription release + // + + if _, ok := r.register[subId]; ok { + xapp.Logger.Debug("RELEASE %s", subs.String()) + delete(r.register, subId) + xapp.Logger.Debug("Registry: substable=%v", r.register) + } + r.subIds = append(r.subIds, subId) + } else if subs.EpList.Size() > 0 { + // + // Subscription route update + // + if subs.RMRRouteCreated == true { + r.RouteDeleteUpdate(subs, c) } - }() + // Endpoint of merged subscription is being deleted + xapp.Logger.Debug("Subscription route update WriteSubscriptionToDb") + c.WriteSubscriptionToDb(subs) + c.UpdateCounter(cUnmergedSubscriptions) + } return nil } @@ -470,26 +589,6 @@ func (r *Registry) RouteDeleteUpdate(subs *Subscription, c *Control) { } } -func (r *Registry) UpdateSubscriptionToDb(subs *Subscription, c *Control) { - r.mutex.Lock() - defer r.mutex.Unlock() - subs.mutex.Lock() - defer subs.mutex.Unlock() - - epamount := subs.EpList.Size() - if epamount == 0 { - if _, ok := r.register[subs.ReqId.InstanceId]; ok { - // Not merged subscription is being deleted - c.RemoveSubscriptionFromDb(subs) - - } - } else if subs.EpList.Size() > 0 { - // Endpoint of merged subscription is being deleted - c.WriteSubscriptionToDb(subs) - c.UpdateCounter(cUnmergedSubscriptions) - } -} - func (r *Registry) GetSubscription(subId uint32) *Subscription { r.mutex.Lock() defer r.mutex.Unlock()