add: batch rpc calls

internal/server/handlers.go

@@ -1,52 +1,51 @@
 package server
 
 import (
-	"net/http"
-
-	"github.com/gin-gonic/gin"
-	"github.com/rs/zerolog/log"
-
 	"git.grassecon.org/grassrootseconomics/openethereum-node-status/pkg/rpc"
-)
+	"github.com/rs/zerolog/log"
-
+	"github.com/ybbus/jsonrpc/v2"
-type (
-	R struct {
-		Result string `json:"result"`
-	}
-
-	P map[string]interface{}
 )
 
 var (
 	RpcClient rpc.RpcClient
 )
 
-func metricsHandler(c *gin.Context) {
+type BatchResponse struct {
-	// pllholder rpc call
+	SyncComplete bool
-	rpcResponse, err := RpcClient.EthRpcCall("eth_gasPrice")
+	ChainName    string
+}
+
+func batchCall() (bool, BatchResponse) {
+	var batchResponse BatchResponse
+
+	rpcResponses, err := RpcClient.EthBatchRpcCall(jsonrpc.RPCRequests{
+		jsonrpc.NewRequest("eth_syncing"),
+		jsonrpc.NewRequest("parity_chain"),
+	})
+
 	if err != nil {
 		log.Error().
 			Err(err).
 			Str("module", "server").
-			Msg("rpc call failed")
+			Msg("error in batch responses")
+		return false, batchResponse
 	}
 
-	value, err := rpcResponse.GetString()
+	if syncStatus, ok := rpcResponses[0]; ok {
+		val, err := syncStatus.GetBool()
 		if err != nil {
-		log.Error().
+			return false, batchResponse
-			Err(err).
+		}
-			Str("module", "server").
+		batchResponse.SyncComplete = !val
-			Msg("rpc response to string conversion failed")
 	}
 
-	c.JSON(http.StatusOK, gin.H{
+	if chainName, ok := rpcResponses[1]; ok {
-		"ok":   true,
+		val, err := chainName.GetString()
-		"data": value,
+		if err != nil {
-	})
+			return false, batchResponse
-}
+		}
+		batchResponse.ChainName = val
+	}
 
-func healthHandler(c *gin.Context) {
+	return true, batchResponse
-	c.JSON(http.StatusOK, gin.H{
-		"ok": true,
-	})
 }

internal/server/router.go

@@ -1,6 +1,9 @@
 package server
 
 import (
+	"fmt"
+	"net/http"
+
 	"github.com/gin-gonic/gin"
 )
 
@@ -15,3 +18,28 @@ func Start(port string, ginMode string) error {
 
 	return router.Run(port)
 }
+
+func healthHandler(c *gin.Context) {
+	allOk, data := batchCall()
+
+	fmt.Println(data.SyncComplete)
+	if !allOk {
+		c.JSON(http.StatusExpectationFailed, gin.H{
+			"ok":      false,
+			"message": "could not get node health status",
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"ok":           true,
+		"syncComplete": data.SyncComplete,
+		"chainName":    data.ChainName,
+	})
+}
+
+func metricsHandler(c *gin.Context) {
+	c.JSON(http.StatusOK, gin.H{
+		"ok": true,
+	})
+}

pkg/rpc/client.go

@@ -19,12 +19,24 @@ func NewRpcClient(endpoint string) RpcClient {
 func (r *RpcClient) EthRpcCall(method string) (*jsonrpc.RPCResponse, error) {
 	response, err := r.ethClient.Call(method)
 	if err != nil {
-		return response, err
+		return nil, err
 	}
 
 	if response.Error != nil {
-		return response, fmt.Errorf("rpc client error: (code %d) %s", response.Error.Code, response.Error.Message)
+		return nil, fmt.Errorf("rpc client error: (code %d) %s", response.Error.Code, response.Error.Message)
 	}
 
 	return response, nil
 }
+
+func (r *RpcClient) EthBatchRpcCall(batchedMethods jsonrpc.RPCRequests) (map[int]*jsonrpc.RPCResponse, error) {
+	response, _ := r.ethClient.CallBatch(batchedMethods)
+
+	if response.HasError() {
+		return nil, fmt.Errorf("batch rpc request failed")
+	}
+
+	results := response.AsMap()
+
+	return results, nil
+}