X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Ftracker.go;h=df3d56e4bf4685feb98f8ee5bad7660c8ad99fd9;hb=HEAD;hp=0e6941d7d37ce31e7d99d9152c75d2d56d7051eb;hpb=0d064ecdb5239a875857b5910f7f6e83f827d7a6;p=ric-plt%2Fsubmgr.git diff --git a/pkg/control/tracker.go b/pkg/control/tracker.go index 0e6941d..df3d56e 100644 --- a/pkg/control/tracker.go +++ b/pkg/control/tracker.go @@ -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) }