xapp-frame v0.8.2 integration to submgr
[ric-plt/submgr.git] / pkg / control / tracker.go
index 0e6941d..df3d56e 100644 (file)
@@ -21,133 +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"
-       "strconv"
        "sync"
 )
 
 //-----------------------------------------------------------------------------
 //
 //-----------------------------------------------------------------------------
-type TransactionXappKey struct {
-       RmrEndpoint
-       Xid string // xapp xid in req
-}
-
-func (key *TransactionXappKey) String() string {
-       return key.RmrEndpoint.String() + "/" + key.Xid
-}
-
-//-----------------------------------------------------------------------------
-//
-//-----------------------------------------------------------------------------
-type Transaction struct {
-       tracker           *Tracker // tracker instance
-       Subs              *Subscription
-       RmrEndpoint       RmrEndpoint
-       Xid               string          // xapp xid in req
-       OrigParams        *xapp.RMRParams // request orginal params
-       RespReceived      bool
-       ForwardRespToXapp bool
-       mutex             sync.Mutex
-}
-
-func (t *Transaction) String() string {
-       t.mutex.Lock()
-       defer t.mutex.Unlock()
-       var subId string = "?"
-       if t.Subs != nil {
-               subId = strconv.FormatUint(uint64(t.Subs.Seq), 10)
-       }
-       return subId + "/" + t.RmrEndpoint.String() + "/" + t.Xid
-}
-
-func (t *Transaction) CheckResponseReceived() bool {
-       t.mutex.Lock()
-       defer t.mutex.Unlock()
-       if t.RespReceived == false {
-               t.RespReceived = true
-               return false
-       }
-       return true
+type Tracker struct {
+       mutex                sync.Mutex
+       transactionXappTable map[TransactionXappKey]*TransactionXapp
+       transSeq             uint64
 }
 
-func (t *Transaction) RetryTransaction() {
-       t.mutex.Lock()
-       defer t.mutex.Unlock()
-       t.RespReceived = false
+func (t *Tracker) Init() {
+       t.transactionXappTable = make(map[TransactionXappKey]*TransactionXapp)
 }
 
-func (t *Transaction) Release() {
+func (t *Tracker) initTransaction(transBase *Transaction) {
        t.mutex.Lock()
        defer t.mutex.Unlock()
-       if t.Subs != nil {
-               t.Subs.UnSetTransaction(t)
-       }
-       if t.tracker != nil {
-               xappkey := TransactionXappKey{t.RmrEndpoint, t.Xid}
-               t.tracker.UnTrackTransaction(xappkey)
-       }
-       t.Subs = nil
-       t.tracker = nil
+       transBase.EventChan = make(chan interface{})
+       transBase.tracker = t
+       transBase.Seq = t.transSeq
+       t.transSeq++
 }
 
-//-----------------------------------------------------------------------------
-//
-//-----------------------------------------------------------------------------
-type Tracker struct {
-       transactionXappTable map[TransactionXappKey]*Transaction
-       mutex                sync.Mutex
+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) Init() {
-       t.transactionXappTable = make(map[TransactionXappKey]*Transaction)
+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) TrackTransaction(subs *Subscription, endpoint RmrEndpoint, params *xapp.RMRParams, respReceived bool, forwardRespToXapp bool) (*Transaction, error) {
+func (t *Tracker) Track(trans *TransactionXapp) error {
 
-       trans := &Transaction{
-               tracker:           nil,
-               Subs:              nil,
-               RmrEndpoint:       endpoint,
-               Xid:               params.Xid,
-               OrigParams:        params,
-               RespReceived:      respReceived,
-               ForwardRespToXapp: forwardRespToXapp,
+       if trans.GetEndpoint() == nil {
+               err := fmt.Errorf("Tracker: No valid endpoint given in %s", trans.String())
+               return err
        }
 
        t.mutex.Lock()
        defer t.mutex.Unlock()
 
-       xappkey := TransactionXappKey{endpoint, params.Xid}
-       if _, ok := t.transactionXappTable[xappkey]; ok {
-               err := fmt.Errorf("Tracker: Similar transaction with xappkey %s is ongoing, transaction %s not created ", xappkey, trans)
-               return nil, err
-       }
+       theKey := *trans.XappKey
 
-       if subs.SetTransaction(trans) == false {
-               othTrans := subs.GetTransaction()
-               err := fmt.Errorf("Tracker: Subscription %s got already transaction ongoing: %s, transaction %s not created", subs, othTrans, trans)
-               return nil, err
-       }
-       trans.Subs = subs
-       if (trans.Subs.RmrEndpoint.Addr != trans.RmrEndpoint.Addr) || (trans.Subs.RmrEndpoint.Port != trans.RmrEndpoint.Port) {
-               err := fmt.Errorf("Tracker: Subscription endpoint %s mismatch with trans: %s", subs, trans)
-               trans.Subs.UnSetTransaction(nil)
-               return nil, err
+       if othtrans, ok := t.transactionXappTable[theKey]; ok {
+               err := fmt.Errorf("Tracker: %s is ongoing, not tracking %s", othtrans, trans)
+               return err
        }
 
        trans.tracker = t
-       t.transactionXappTable[xappkey] = trans
-       return trans, nil
+       t.transactionXappTable[theKey] = trans
+       xapp.Logger.Debug("Tracker: Append %s", trans.String())
+       //xapp.Logger.Debug("Tracker: transtable=%v", t.transactionXappTable)
+       return nil
 }
 
-func (t *Tracker) UnTrackTransaction(xappKey TransactionXappKey) (*Transaction, error) {
+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
        }
-       return nil, fmt.Errorf("Tracker: No record for xappkey %s", xappKey)
+       return nil, fmt.Errorf("Tracker: No record %v", xappKey)
 }