X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Ftracker.go;h=df3d56e4bf4685feb98f8ee5bad7660c8ad99fd9;hb=HEAD;hp=e08f8dbc91110b7c24dbddafbc1b063db08763db;hpb=e00186861608731e2390055a0e1b1cf455670508;p=ric-plt%2Fsubmgr.git

diff --git a/pkg/control/tracker.go b/pkg/control/tracker.go
index e08f8db..df3d56e 100644
--- a/pkg/control/tracker.go
+++ b/pkg/control/tracker.go
@@ -21,73 +21,88 @@ package control
 
 import (
 	"fmt"
+	"gerrit.o-ran-sc.org/r/ric-plt/e2ap/pkg/e2ap"
+	"gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+	"sync"
 )
 
-/*
-Implements a record of ongoing transactions and helper functions to CRUD the records.
-*/
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
 type Tracker struct {
-	transaction_table map[Transaction_key]Transaction
+	mutex                sync.Mutex
+	transactionXappTable map[TransactionXappKey]*TransactionXapp
+	transSeq             uint64
 }
 
 func (t *Tracker) Init() {
-	t.transaction_table = make(map[Transaction_key]Transaction)
+	t.transactionXappTable = make(map[TransactionXappKey]*TransactionXapp)
 }
 
-/*
-Checks if a tranascation with similar type has been ongoing. If not then creates one.
-Returns error if there is similar transatcion ongoing.
-*/
-func (t *Tracker) Track_transaction(key Transaction_key, xact Transaction) error{
-	if _, ok := t.transaction_table[key]; ok {
-		// TODO: Implement merge related check here. If the key is same but the value is different.
-		err := fmt.Errorf("Transaction tracker: Similar transaction with sub id %d and type %s is ongoing", key.SubID, key.trans_type )
-		return err
-	}
-	t.transaction_table[key] = xact
-	return nil
+func (t *Tracker) initTransaction(transBase *Transaction) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	transBase.EventChan = make(chan interface{})
+	transBase.tracker = t
+	transBase.Seq = t.transSeq
+	t.transSeq++
 }
 
-/*
-Retreives the transaction table entry for the given request.
-Returns error in case the transaction cannot be found.
-*/
-func (t *Tracker) Update_transaction(SubID uint16, trans_type Action, xact Transaction) error{
-	key := Transaction_key{SubID, trans_type}
-	if _, ok := t.transaction_table[key]; ok {
-		// TODO: Implement merge related check here. If the key is same but the value is different.
-		err := fmt.Errorf("Transaction tracker: Similar transaction with sub id %d and type %s is ongoing", key.SubID, key.trans_type )
+func (t *Tracker) NewSubsTransaction(subs *Subscription) *TransactionSubs {
+	trans := &TransactionSubs{}
+	trans.Meid = subs.GetMeid()
+	t.initTransaction(&trans.Transaction)
+	xapp.Logger.Debug("CREATE %s", trans.String())
+	return trans
+}
+
+func (t *Tracker) NewXappTransaction(
+	endpoint *xapp.RmrEndpoint,
+	xid string,
+	requestId e2ap.RequestId,
+	meid *xapp.RMRMeid) *TransactionXapp {
+
+	trans := &TransactionXapp{}
+	trans.XappKey = &TransactionXappKey{requestId.Id, *endpoint, xid}
+	trans.Meid = meid
+	trans.RequestId = requestId
+	t.initTransaction(&trans.Transaction)
+	xapp.Logger.Debug("CREATE %s", trans.String())
+	return trans
+}
+
+func (t *Tracker) Track(trans *TransactionXapp) error {
+
+	if trans.GetEndpoint() == nil {
+		err := fmt.Errorf("Tracker: No valid endpoint given in %s", trans.String())
 		return err
 	}
-	t.transaction_table[key] = xact
-	return nil
-}
 
-/*
-Retreives the transaction table entry for the given request.
-Returns error in case the transaction cannot be found.
-*/
-func (t *Tracker) Retrive_transaction(subID uint16, act Action) (Transaction, error){
-	key := Transaction_key{subID, act}
-	var xact Transaction
-	if xact, ok := t.transaction_table[key]; ok {
-		return xact, nil
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+
+	theKey := *trans.XappKey
+
+	if othtrans, ok := t.transactionXappTable[theKey]; ok {
+		err := fmt.Errorf("Tracker: %s is ongoing, not tracking %s", othtrans, trans)
+		return err
 	}
-	err := fmt.Errorf("Tranaction record for Subscription ID %d and action %s does not exist", subID, act)
-	return xact, err
+
+	trans.tracker = t
+	t.transactionXappTable[theKey] = trans
+	xapp.Logger.Debug("Tracker: Append %s", trans.String())
+	//xapp.Logger.Debug("Tracker: transtable=%v", t.transactionXappTable)
+	return nil
 }
 
-/*
-Deletes the transaction table entry for the given request and returns the deleted xapp's address and port for reference.
-Returns error in case the transaction cannot be found.
-*/
-func (t *Tracker) complete_transaction(subID uint16, act Action) (Transaction, error){
-	key := Transaction_key{subID, act}
-	var empty_transaction Transaction
-	if xact, ok := t.transaction_table[key]; ok {
-		delete(t.transaction_table, key)
-		return xact, nil
+func (t *Tracker) UnTrackTransaction(xappKey TransactionXappKey) (*TransactionXapp, error) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	if trans, ok2 := t.transactionXappTable[xappKey]; ok2 {
+		xapp.Logger.Debug("Tracker: Remove %s", trans.String())
+		delete(t.transactionXappTable, xappKey)
+		//xapp.Logger.Debug("Tracker: transtable=%v", t.transactionXappTable)
+		return trans, nil
 	}
-	err := fmt.Errorf("Tranaction record for Subscription ID %d and action %s does not exist", subID, act)
-	return empty_transaction, err
+	return nil, fmt.Errorf("Tracker: No record %v", xappKey)
 }