format

berachain · Feb 15, 2024 · a331cc1 · a331cc1
1 parent af3712d
commit a331cc1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/core/transactor/types/types.go b/core/transactor/types/types.go
@@ -77,7 +77,7 @@ func (TxRequest) New() types.Marshallable {
 	return &TxRequest{}
 }
 
-// String() implements fmt.Stringer
+// String() implements fmt.Stringer.
 func (tx *TxRequest) String() string {
 	return tx.MsgID
 }

diff --git a/types/queue/mem/mem.go b/types/queue/mem/mem.go
@@ -4,6 +4,7 @@ import (
 	"container/list"
 	"sync"
 
+	"github.com/berachain/go-utils/utils"
 	"github.com/berachain/offchain-sdk/types/queue/types"
 )
 
@@ -45,7 +46,7 @@ func (q *Queue[T]) Receive() (string, T, bool) {
 	}
 
 	q.queuedItems.Remove(element)
-	val := element.Value.(T)
+	val := utils.MustGetAs[T](element.Value)
 	msgID := val.String()
 	q.inProgressItems[msgID] = element
 
@@ -62,7 +63,7 @@ func (q *Queue[T]) ReceiveMany(num int32) ([]string, []T, error) {
 	for i := int32(0); i < num; i++ {
 		if element := q.queuedItems.Front(); element != nil {
 			q.queuedItems.Remove(element)
-			val := element.Value.(T)
+			val := utils.MustGetAs[T](element.Value)
 			msgID := val.String()
 			q.inProgressItems[msgID] = element
 			msgIDs = append(msgIDs, msgID)