diff options
| author | chrislu <chris.lu@gmail.com> | 2022-04-01 16:37:06 -0700 |
|---|---|---|
| committer | chrislu <chris.lu@gmail.com> | 2022-04-01 16:37:06 -0700 |
| commit | 800cbc004c84a5113076fdbb70e93d53e8434073 (patch) | |
| tree | 1e3076f96a81d6a17a57a344657996ca9b34ae0e /weed/pb/grpc_client_server.go | |
| parent | 7fdd29c234390c09eb48e2b59200eaeb0a41637b (diff) | |
| download | seaweedfs-800cbc004c84a5113076fdbb70e93d53e8434073.tar.xz seaweedfs-800cbc004c84a5113076fdbb70e93d53e8434073.zip | |
volume server adds ping function
Diffstat (limited to 'weed/pb/grpc_client_server.go')
| -rw-r--r-- | weed/pb/grpc_client_server.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/weed/pb/grpc_client_server.go b/weed/pb/grpc_client_server.go index 50feb2e23..d89e61433 100644 --- a/weed/pb/grpc_client_server.go +++ b/weed/pb/grpc_client_server.go @@ -4,6 +4,7 @@ import ( "context" "fmt" "github.com/chrislusf/seaweedfs/weed/glog" + "github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb" "github.com/chrislusf/seaweedfs/weed/util" "math/rand" "net/http" @@ -206,6 +207,14 @@ func WithMasterClient(streamingMode bool, master ServerAddress, grpcDialOption g } +func WithVolumeServerClient(streamingMode bool, volumeServer ServerAddress, grpcDialOption grpc.DialOption, fn func(client volume_server_pb.VolumeServerClient) error) error { + return WithGrpcClient(streamingMode, func(grpcConnection *grpc.ClientConn) error { + client := volume_server_pb.NewVolumeServerClient(grpcConnection) + return fn(client) + }, volumeServer.ToGrpcAddress(), grpcDialOption) + +} + func WithOneOfGrpcMasterClients(streamingMode bool, masterGrpcAddresses map[string]ServerAddress, grpcDialOption grpc.DialOption, fn func(client master_pb.SeaweedClient) error) (err error) { for _, masterGrpcAddress := range masterGrpcAddresses { |