forked from yarpc/yarpc-go
/
server.go
74 lines (61 loc) · 1.94 KB
/
server.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
// Code generated by thriftrw-plugin-yarpc
// @generated
package readonlystoreserver
import (
"context"
"go.uber.org/thriftrw/wire"
"go.uber.org/yarpc/api/transport"
"go.uber.org/yarpc/encoding/thrift"
"go.uber.org/yarpc/encoding/thrift/thriftrw-plugin-yarpc/internal/tests/atomic"
"go.uber.org/yarpc/encoding/thrift/thriftrw-plugin-yarpc/internal/tests/common/baseserviceserver"
)
// Interface is the server-side interface for the ReadOnlyStore service.
type Interface interface {
baseserviceserver.Interface
Integer(
ctx context.Context,
Key *string,
) (int64, error)
}
// New prepares an implementation of the ReadOnlyStore service for
// registration.
//
// handler := ReadOnlyStoreHandler{}
// dispatcher.Register(readonlystoreserver.New(handler))
func New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {
h := handler{impl}
service := thrift.Service{
Name: "ReadOnlyStore",
Methods: []thrift.Method{
thrift.Method{
Name: "integer",
HandlerSpec: thrift.HandlerSpec{
Type: transport.Unary,
Unary: thrift.UnaryHandler(h.Integer),
},
Signature: "Integer(Key *string) (int64)",
ThriftModule: atomic.ThriftModule,
},
},
}
procedures := make([]transport.Procedure, 0, 1)
procedures = append(procedures, baseserviceserver.New(impl, opts...)...)
procedures = append(procedures, thrift.BuildProcedures(service, opts...)...)
return procedures
}
type handler struct{ impl Interface }
func (h handler) Integer(ctx context.Context, body wire.Value) (thrift.Response, error) {
var args atomic.ReadOnlyStore_Integer_Args
if err := args.FromWire(body); err != nil {
return thrift.Response{}, err
}
success, err := h.impl.Integer(ctx, args.Key)
hadError := err != nil
result, err := atomic.ReadOnlyStore_Integer_Helper.WrapResponse(success, err)
var response thrift.Response
if err == nil {
response.IsApplicationError = hadError
response.Body = result
}
return response, err
}