refactor: ♻️ mev: update interface (#97)

Signed-off-by: thanhpp <[email protected]>
KyberNetwork · Nov 26, 2024 · 5f3cf28 · 5f3cf28
1 parent a4f5a8b
commit 5f3cf28
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/pkg/mev/blxr_bundle_sender.go b/pkg/mev/blxr_bundle_sender.go
@@ -123,6 +123,13 @@ func (s *BloxrouteClient) SendBundle(
 	return SendBundleResponse(resp), nil
 }
 
+func (s *BloxrouteClient) SendPrivateRawTransaction(
+	ctx context.Context,
+	tx *types.Transaction,
+) (SendPrivateRawTransactionResponse, error) {
+	return SendPrivateRawTransactionResponse{}, nil
+}
+
 func (s *BloxrouteClient) CancelBundle(
 	ctx context.Context, bundleUUID string,
 ) error {

diff --git a/pkg/mev/bundle_sender.go b/pkg/mev/bundle_sender.go
@@ -190,10 +190,6 @@ func (s *Client) sendBundle(
 	blockNumber uint64,
 	txs ...*types.Transaction,
 ) (SendBundleResponse, error) {
-	if !s.enableSendPrivateRaw {
-		return SendBundleResponse{}, nil
-	}
-
 	req := SendRequest{
 		ID:      SendBundleID,
 		JSONRPC: JSONRPC2,
@@ -244,6 +240,10 @@ func (s *Client) SendPrivateRawTransaction(
 	ctx context.Context,
 	tx *types.Transaction,
 ) (SendPrivateRawTransactionResponse, error) {
+	if !s.enableSendPrivateRaw {
+		return SendPrivateRawTransactionResponse{}, nil
+	}
+
 	req := SendRequest{
 		ID:      SendBundleID,
 		JSONRPC: JSONRPC2,

diff --git a/pkg/mev/pkg.go b/pkg/mev/pkg.go
@@ -82,6 +82,10 @@ type IBundleSender interface {
 	CancelBundle(
 		ctx context.Context, bundleUUID string,
 	) error
+	SendPrivateRawTransaction(
+		ctx context.Context,
+		tx *types.Transaction,
+	) (SendPrivateRawTransactionResponse, error)
 	SimulateBundle(ctx context.Context, blockNumber uint64, txs ...*types.Transaction) (SendBundleResponse, error)
 	GetSenderType() BundleSenderType
 	GetBundleStats(