stats

pkg/shreder/client.go

@@ -13,11 +13,13 @@ import (
 )
 
 type Client struct {
-	EnableClientStats bool
-	conn              *grpc.ClientConn
-	client            ShrederServiceClient
-	tableLoader       *AddressTables
-	subscription      map[string]*SubscribeRequestFilterTransactions
+	enableBlockStats bool
+	enableParseStats bool
+
+	conn         *grpc.ClientConn
+	client       ShrederServiceClient
+	tableLoader  *AddressTables
+	subscription map[string]*SubscribeRequestFilterTransactions
 
 	pool *ants.Pool
 
@@ -25,10 +27,37 @@ type Client struct {
 	lastSlotTime time.Time
 }
 
+type ClientOpts struct {
+	blockStats      bool
+	showTableLoaded bool
+	logParseStats   bool
+}
+
+type ClientOption func(*ClientOpts)
+
+func ShowTableLoaded(enable bool) ClientOption {
+	return func(opts *ClientOpts) {
+		opts.showTableLoaded = enable
+	}
+}
+
+func BlocksStats(enable bool) ClientOption {
+	return func(opts *ClientOpts) {
+		opts.blockStats = enable
+	}
+}
+
+func LogParsedStats(enable bool) ClientOption {
+	return func(opts *ClientOpts) {
+		opts.logParseStats = enable
+	}
+}
+
 func NewShrederClient(
 	url string,
 	rpcClient *rpc.Client,
 	subscription map[string]*SubscribeRequestFilterTransactions,
+	options ...ClientOption,
 ) (*Client, func(), error) {
 	if rpcClient == nil {
 		return nil, func() {}, fmt.Errorf("rpc client is nil")
@@ -45,13 +74,23 @@ func NewShrederClient(
 	if err != nil {
 		return nil, func() {}, err
 	}
-
+	o := &ClientOpts{
+		blockStats:      false,
+		showTableLoaded: true,
+		logParseStats:   false,
+	}
+	for _, option := range options {
+		option(o)
+	}
 	s := &Client{
 		conn:         conn,
 		client:       NewShrederServiceClient(conn),
 		subscription: subscription,
-		tableLoader:  NewAddressTables(rpcClient),
+		tableLoader:  NewAddressTables(rpcClient, o.showTableLoaded),
 		pool:         pool,
+
+		enableBlockStats: o.blockStats,
+		enableParseStats: o.logParseStats,
 	}
 
 	return s, func() {
@@ -95,7 +134,7 @@ func (c *Client) ReadSync(ctx context.Context, txCh chan<- TxSignalBatch) error
 			return err
 		}
 
-		if c.EnableClientStats {
+		if c.enableBlockStats {
 			slot := response.Transaction.Slot
 			now := time.Now()
 			if c.lastSlotTime.IsZero() || slot > c.lastSlot {
@@ -110,7 +149,7 @@ func (c *Client) ReadSync(ctx context.Context, txCh chan<- TxSignalBatch) error
 		txData := response.Transaction
 
 		err = c.pool.Submit(func() {
-			txBatch := ParseTransaction(txData, c.tableLoader, c.EnableClientStats)
+			txBatch := ParseTransaction(txData, c.tableLoader, c.enableParseStats)
 			if len(txBatch) == 0 {
 				return
 			}