Skip to content

Commit

Permalink
feat: agglayer grpc client
Browse files Browse the repository at this point in the history
  • Loading branch information
goran-ethernal committed Feb 19, 2025
1 parent 6441441 commit a6b2914
Show file tree
Hide file tree
Showing 3 changed files with 34 additions and 3 deletions.
25 changes: 25 additions & 0 deletions agglayer/grpc/agglayer_grpc_client.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
package grpc

import (
"github.com/agglayer/aggkit/agglayer/proto/node"
"github.com/agglayer/aggkit/common"
)

type AgglayerGRPCClient struct {
cfgService node.AgglayerConfigurationServiceClient
networkStateService node.AgglayerNetworkStateServiceClient
submissionService node.AgglayerCertificateSubmissionServiceClient
}

// NewAggchainProofClient initializes a new AggchainProof instance
func NewAgglayerGRPCClient(serverAddr string) (*AgglayerGRPCClient, error) {
grpcClient, err := common.NewClient(serverAddr)
if err != nil {
return nil, err
}
return &AgglayerGRPCClient{
cfgService: node.NewAgglayerConfigurationServiceClient(grpcClient.Conn()),
networkStateService: node.NewAgglayerNetworkStateServiceClient(grpcClient.Conn()),
submissionService: node.NewAgglayerCertificateSubmissionServiceClient(grpcClient.Conn()),
}, nil
}
5 changes: 3 additions & 2 deletions aggsender/grpc/aggchain_proof_client.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (

agglayer "github.com/agglayer/aggkit/agglayer"
"github.com/agglayer/aggkit/aggsender/types"
aggkitCommon "github.com/agglayer/aggkit/common"
"github.com/agglayer/aggkit/l1infotreesync"
treeTypes "github.com/agglayer/aggkit/tree/types"
"github.com/ethereum/go-ethereum/common"
Expand Down Expand Up @@ -33,12 +34,12 @@ type AggchainProofClient struct {

// NewAggchainProofClient initializes a new AggchainProof instance
func NewAggchainProofClient(serverAddr string) (*AggchainProofClient, error) {
grpcClient, err := NewClient(serverAddr)
grpcClient, err := aggkitCommon.NewClient(serverAddr)
if err != nil {
return nil, err
}
return &AggchainProofClient{
client: types.NewAggchainProofServiceClient(grpcClient.conn),
client: types.NewAggchainProofServiceClient(grpcClient.Conn()),
}, nil
}

Expand Down
7 changes: 6 additions & 1 deletion aggsender/grpc/client.go → common/grpc_client.go
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package grpc
package common

import (
"google.golang.org/grpc"
Expand All @@ -23,6 +23,11 @@ func NewClient(serverAddr string) (*Client, error) {
return &Client{conn: conn}, nil
}

// Conn returns the gRPC connection
func (c *Client) Conn() *grpc.ClientConn {
return c.conn
}

// Close closes the gRPC connection
func (c *Client) Close() error {
return c.conn.Close()
Expand Down

0 comments on commit a6b2914

Please sign in to comment.