seaweedfs/weed/operation/grpc_client.go

28 lines
941 B
Go
Raw Normal View History

package operation
import (
"google.golang.org/grpc"
2020-03-04 08:39:47 +00:00
"github.com/chrislusf/seaweedfs/weed/pb"
"github.com/chrislusf/seaweedfs/weed/pb/master_pb"
2018-10-14 07:30:20 +00:00
"github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb"
)
func WithVolumeServerClient(volumeServer pb.ServerAddress, grpcDialOption grpc.DialOption, fn func(volume_server_pb.VolumeServerClient) error) error {
2020-03-04 08:39:47 +00:00
return pb.WithCachedGrpcClient(func(grpcConnection *grpc.ClientConn) error {
2018-12-07 09:25:01 +00:00
client := volume_server_pb.NewVolumeServerClient(grpcConnection)
return fn(client)
}, volumeServer.ToGrpcAddress(), grpcDialOption)
}
func WithMasterServerClient(masterServer pb.ServerAddress, grpcDialOption grpc.DialOption, fn func(masterClient master_pb.SeaweedClient) error) error {
2020-03-04 08:39:47 +00:00
return pb.WithCachedGrpcClient(func(grpcConnection *grpc.ClientConn) error {
2018-12-07 09:25:01 +00:00
client := master_pb.NewSeaweedClient(grpcConnection)
return fn(client)
}, masterServer.ToGrpcAddress(), grpcDialOption)
2020-04-05 07:51:16 +00:00
}