import (
"fmt"
-// "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+ "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]*Transaction
}
func (t *Tracker) Init() {
- t.transaction_table = make(map[Transaction_key]Transaction)
+ t.transactionXappTable = make(map[TransactionXappKey]*Transaction)
}
-/*
-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
+func (t *Tracker) TrackTransaction(
+ endpoint *RmrEndpoint,
+ mtype int,
+ xid string,
+ meid *xapp.RMRMeid,
+ respReceived bool,
+ forwardRespToXapp bool) (*Transaction, error) {
+
+ if endpoint == nil {
+ err := fmt.Errorf("Tracker: No valid endpoint given")
+ return nil, 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
+ trans := &Transaction{
+ tracker: nil,
+ Subs: nil,
+ RmrEndpoint: *endpoint,
+ Mtype: mtype,
+ Xid: xid,
+ Meid: meid,
+ RespReceived: respReceived,
+ ForwardRespToXapp: forwardRespToXapp,
+ }
+
+ t.mutex.Lock()
+ defer t.mutex.Unlock()
+
+ xappkey := TransactionXappKey{*endpoint, 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
}
- 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[xappkey] = trans
+ return trans, 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) (string, uint16, error){
- key := Transaction_key{subID, act}
- var empty_address string
- var empty_port uint16
- if xact, ok := t.transaction_table[key]; ok {
- delete(t.transaction_table, key)
- return xact.Xapp_instance_address, xact.Xapp_port, nil
+func (t *Tracker) UnTrackTransaction(xappKey TransactionXappKey) (*Transaction, error) {
+ t.mutex.Lock()
+ defer t.mutex.Unlock()
+ if trans, ok2 := t.transactionXappTable[xappKey]; ok2 {
+ delete(t.transactionXappTable, xappKey)
+ return trans, nil
}
- err := fmt.Errorf("Tranaction record for Subscription ID %d and action %s does not exist", subID, act)
- return empty_address, empty_port, err
+ return nil, fmt.Errorf("Tracker: No record for xappkey %s", xappKey)
}