/
grpc.go
107 lines (86 loc) · 2.53 KB
/
grpc.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package shared
import (
"context"
hclog "github.com/hashicorp/go-hclog"
plugin "github.com/hashicorp/go-plugin"
"github.com/hashicorp/go-plugin/examples/bidirectional/proto"
"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
}