Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / hashicorp / go-plugin / examples / bidirectional / shared / grpc.go

diff --git a/vendor/github.com/hashicorp/go-plugin/examples/bidirectional/shared/grpc.go b/vendor/github.com/hashicorp/go-plugin/examples/bidirectional/shared/grpc.go
deleted file mode 100644 (file)
index c2e54ba..0000000
--- a/vendor/github.com/hashicorp/go-plugin/examples/bidirectional/shared/grpc.go
+++ /dev/null
@@ -1,103 +0,0 @@
-package shared
-
-import (
-       hclog "github.com/hashicorp/go-hclog"
-       plugin "github.com/hashicorp/go-plugin"
-       "github.com/hashicorp/go-plugin/examples/bidirectional/proto"
-       "golang.org/x/net/context"
-       "google.golang.org/grpc"
-)
-
-// GRPCClient is an implementation of KV that talks over RPC.
-type GRPCClient struct {
-       broker *plugin.GRPCBroker
-       client proto.CounterClient
-}
-
-func (m *GRPCClient) Put(key string, value int64, a AddHelper) error {
-       addHelperServer := &GRPCAddHelperServer{Impl: a}
-
-       var s *grpc.Server
-       serverFunc := func(opts []grpc.ServerOption) *grpc.Server {
-               s = grpc.NewServer(opts...)
-               proto.RegisterAddHelperServer(s, addHelperServer)
-
-               return s
-       }
-
-       brokerID := m.broker.NextId()
-       go m.broker.AcceptAndServe(brokerID, serverFunc)
-
-       _, err := m.client.Put(context.Background(), &proto.PutRequest{
-               AddServer: brokerID,
-               Key:       key,
-               Value:     value,
-       })
-
-       s.Stop()
-       return err
-}
-
-func (m *GRPCClient) Get(key string) (int64, error) {
-       resp, err := m.client.Get(context.Background(), &proto.GetRequest{
-               Key: key,
-       })
-       if err != nil {
-               return 0, err
-       }
-
-       return resp.Value, nil
-}
-
-// Here is the gRPC server that GRPCClient talks to.
-type GRPCServer struct {
-       // This is the real implementation
-       Impl Counter
-
-       broker *plugin.GRPCBroker
-}
-
-func (m *GRPCServer) Put(ctx context.Context, req *proto.PutRequest) (*proto.Empty, error) {
-       conn, err := m.broker.Dial(req.AddServer)
-       if err != nil {
-               return nil, err
-       }
-       defer conn.Close()
-
-       a := &GRPCAddHelperClient{proto.NewAddHelperClient(conn)}
-       return &proto.Empty{}, m.Impl.Put(req.Key, req.Value, a)
-}
-
-func (m *GRPCServer) Get(ctx context.Context, req *proto.GetRequest) (*proto.GetResponse, error) {
-       v, err := m.Impl.Get(req.Key)
-       return &proto.GetResponse{Value: v}, err
-}
-
-// GRPCClient is an implementation of KV that talks over RPC.
-type GRPCAddHelperClient struct{ client proto.AddHelperClient }
-
-func (m *GRPCAddHelperClient) Sum(a, b int64) (int64, error) {
-       resp, err := m.client.Sum(context.Background(), &proto.SumRequest{
-               A: a,
-               B: b,
-       })
-       if err != nil {
-               hclog.Default().Info("add.Sum", "client", "start", "err", err)
-               return 0, err
-       }
-       return resp.R, err
-}
-
-// Here is the gRPC server that GRPCClient talks to.
-type GRPCAddHelperServer struct {
-       // This is the real implementation
-       Impl AddHelper
-}
-
-func (m *GRPCAddHelperServer) Sum(ctx context.Context, req *proto.SumRequest) (resp *proto.SumResponse, err error) {
-       r, err := m.Impl.Sum(req.A, req.B)
-       if err != nil {
-               return nil, err
-       }
-       return &proto.SumResponse{R: r}, err
-}