feat: support customized trace config and block hash (#126)

* feat: support customized trace config and block hash Signed-off-by: Jingfu Wang <[email protected]> * test: fix CI Signed-off-by: Jingfu Wang <[email protected]> --------- Signed-off-by: Jingfu Wang <[email protected]>
coinbase · Jul 29, 2024 · 3b6a8e4 · 3b6a8e4
1 parent 0e198c3
commit 3b6a8e4
Show file tree

Hide file tree

Showing 8 changed files with 393 additions and 35 deletions.
diff --git a/client/client.go b/client/client.go
@@ -44,8 +44,9 @@ import (
 )
 
 type SDKClient struct {
-	P  *params.ChainConfig
-	tc *tracers.TraceConfig
+	P            *params.ChainConfig
+	tc           *tracers.TraceConfig
+	customizedTc interface{}
 
 	rosettaConfig configuration.RosettaConfig
 
@@ -79,9 +80,15 @@ func NewClient(cfg *configuration.Configuration, rpcClient *RPCClient) (*SDKClie
 		return nil, fmt.Errorf("unable to load trace config: %w", err)
 	}
 
+	var customizedTc interface{}
+	if cfg.RosettaCfg.SupportCustomizedTraceConfig {
+		customizedTc = cfg.RosettaCfg.CustomizedTraceConfig
+	}
+
 	return &SDKClient{
 		P:              cfg.ChainConfig,
 		tc:             tc,
+		customizedTc:   customizedTc,
 		rosettaConfig:  cfg.RosettaCfg,
 		RPCClient:      c,
 		EthClient:      ec,
@@ -411,7 +418,12 @@ func (ec *SDKClient) TraceBlockByHash(
 
 	var calls []*rpcCall
 	var raw json.RawMessage
-	err := ec.CallContext(ctx, &raw, "debug_traceBlockByHash", blockHash, ec.tc)
+	var err error
+	if ec.rosettaConfig.SupportCustomizedTraceConfig {
+		err = ec.CallContext(ctx, &raw, "debug_traceBlockByHash", blockHash, ec.customizedTc)
+	} else {
+		err = ec.CallContext(ctx, &raw, "debug_traceBlockByHash", blockHash, ec.tc)
+	}
 	if err != nil {
 		return nil, err
 	}
@@ -938,6 +950,14 @@ func (ec *SDKClient) GetLoadedTransaction(
 	return loadedTx, nil
 }
 
+func (ec *SDKClient) GetBlockHash(ctx context.Context, blockIdentifier RosettaTypes.BlockIdentifier) (string, error) {
+	return blockIdentifier.Hash, nil
+}
+
+func (ec *SDKClient) SkipTxReceiptParsing(contractAddress string) bool {
+	return false
+}
+
 ///////////////////////////////////////////////////////////////////////////
 // Below are functions that should be implemented by chain specific Rosetta
 ///////////////////////////////////////////////////////////////////////////

diff --git a/configuration/configuration.go b/configuration/configuration.go
@@ -112,6 +112,12 @@ type RosettaConfig struct {
 
 	// PriorityFeeDivisor is the divisor of priority fee for EIP-1559
 	PriorityFeeDivisor *big.Int
+
+	// SupportCustomizedTraceConfig indicates if the blockchain supports customized trace config
+	SupportCustomizedTraceConfig bool
+
+	// CustomizedTraceConfig is the blockchain customized trace config
+	CustomizedTraceConfig interface{}
 }
 
 type Token struct {

diff --git a/mocks/client/graph_ql.go b/mocks/client/graph_ql.go
diff --git a/mocks/client/jsonrpc.go b/mocks/client/jsonrpc.go