-
Notifications
You must be signed in to change notification settings - Fork 45
/
service.connect.go
187 lines (165 loc) · 8.08 KB
/
service.connect.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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
// Code generated by protoc-gen-connect-go. DO NOT EDIT.
//
// Source: sf/substreams/rpc/v2/service.proto
package pbsubstreamsrpcconnect
import (
connect "connectrpc.com/connect"
context "context"
errors "errors"
v21 "github.com/streamingfast/pbgo/sf/firehose/v2"
v2 "github.com/streamingfast/substreams/pb/sf/substreams/rpc/v2"
http "net/http"
strings "strings"
)
// This is a compile-time assertion to ensure that this generated file and the connect package are
// compatible. If you get a compiler error that this constant is not defined, this code was
// generated with a version of connect newer than the one compiled into your binary. You can fix the
// problem by either regenerating this code with an older version of connect or updating the connect
// version compiled into your binary.
const _ = connect.IsAtLeastVersion1_13_0
const (
// StreamName is the fully-qualified name of the Stream service.
StreamName = "sf.substreams.rpc.v2.Stream"
// EndpointInfoName is the fully-qualified name of the EndpointInfo service.
EndpointInfoName = "sf.substreams.rpc.v2.EndpointInfo"
)
// These constants are the fully-qualified names of the RPCs defined in this package. They're
// exposed at runtime as Spec.Procedure and as the final two segments of the HTTP route.
//
// Note that these are different from the fully-qualified method names used by
// google.golang.org/protobuf/reflect/protoreflect. To convert from these constants to
// reflection-formatted method names, remove the leading slash and convert the remaining slash to a
// period.
const (
// StreamBlocksProcedure is the fully-qualified name of the Stream's Blocks RPC.
StreamBlocksProcedure = "/sf.substreams.rpc.v2.Stream/Blocks"
// EndpointInfoInfoProcedure is the fully-qualified name of the EndpointInfo's Info RPC.
EndpointInfoInfoProcedure = "/sf.substreams.rpc.v2.EndpointInfo/Info"
)
// These variables are the protoreflect.Descriptor objects for the RPCs defined in this package.
var (
streamServiceDescriptor = v2.File_sf_substreams_rpc_v2_service_proto.Services().ByName("Stream")
streamBlocksMethodDescriptor = streamServiceDescriptor.Methods().ByName("Blocks")
endpointInfoServiceDescriptor = v2.File_sf_substreams_rpc_v2_service_proto.Services().ByName("EndpointInfo")
endpointInfoInfoMethodDescriptor = endpointInfoServiceDescriptor.Methods().ByName("Info")
)
// StreamClient is a client for the sf.substreams.rpc.v2.Stream service.
type StreamClient interface {
Blocks(context.Context, *connect.Request[v2.Request]) (*connect.ServerStreamForClient[v2.Response], error)
}
// NewStreamClient constructs a client for the sf.substreams.rpc.v2.Stream service. By default, it
// uses the Connect protocol with the binary Protobuf Codec, asks for gzipped responses, and sends
// uncompressed requests. To use the gRPC or gRPC-Web protocols, supply the connect.WithGRPC() or
// connect.WithGRPCWeb() options.
//
// The URL supplied here should be the base URL for the Connect or gRPC server (for example,
// http://api.acme.com or https://acme.com/grpc).
func NewStreamClient(httpClient connect.HTTPClient, baseURL string, opts ...connect.ClientOption) StreamClient {
baseURL = strings.TrimRight(baseURL, "/")
return &streamClient{
blocks: connect.NewClient[v2.Request, v2.Response](
httpClient,
baseURL+StreamBlocksProcedure,
connect.WithSchema(streamBlocksMethodDescriptor),
connect.WithClientOptions(opts...),
),
}
}
// streamClient implements StreamClient.
type streamClient struct {
blocks *connect.Client[v2.Request, v2.Response]
}
// Blocks calls sf.substreams.rpc.v2.Stream.Blocks.
func (c *streamClient) Blocks(ctx context.Context, req *connect.Request[v2.Request]) (*connect.ServerStreamForClient[v2.Response], error) {
return c.blocks.CallServerStream(ctx, req)
}
// StreamHandler is an implementation of the sf.substreams.rpc.v2.Stream service.
type StreamHandler interface {
Blocks(context.Context, *connect.Request[v2.Request], *connect.ServerStream[v2.Response]) error
}
// NewStreamHandler builds an HTTP handler from the service implementation. It returns the path on
// which to mount the handler and the handler itself.
//
// By default, handlers support the Connect, gRPC, and gRPC-Web protocols with the binary Protobuf
// and JSON codecs. They also support gzip compression.
func NewStreamHandler(svc StreamHandler, opts ...connect.HandlerOption) (string, http.Handler) {
streamBlocksHandler := connect.NewServerStreamHandler(
StreamBlocksProcedure,
svc.Blocks,
connect.WithSchema(streamBlocksMethodDescriptor),
connect.WithHandlerOptions(opts...),
)
return "/sf.substreams.rpc.v2.Stream/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.URL.Path {
case StreamBlocksProcedure:
streamBlocksHandler.ServeHTTP(w, r)
default:
http.NotFound(w, r)
}
})
}
// UnimplementedStreamHandler returns CodeUnimplemented from all methods.
type UnimplementedStreamHandler struct{}
func (UnimplementedStreamHandler) Blocks(context.Context, *connect.Request[v2.Request], *connect.ServerStream[v2.Response]) error {
return connect.NewError(connect.CodeUnimplemented, errors.New("sf.substreams.rpc.v2.Stream.Blocks is not implemented"))
}
// EndpointInfoClient is a client for the sf.substreams.rpc.v2.EndpointInfo service.
type EndpointInfoClient interface {
Info(context.Context, *connect.Request[v21.InfoRequest]) (*connect.Response[v21.InfoResponse], error)
}
// NewEndpointInfoClient constructs a client for the sf.substreams.rpc.v2.EndpointInfo service. By
// default, it uses the Connect protocol with the binary Protobuf Codec, asks for gzipped responses,
// and sends uncompressed requests. To use the gRPC or gRPC-Web protocols, supply the
// connect.WithGRPC() or connect.WithGRPCWeb() options.
//
// The URL supplied here should be the base URL for the Connect or gRPC server (for example,
// http://api.acme.com or https://acme.com/grpc).
func NewEndpointInfoClient(httpClient connect.HTTPClient, baseURL string, opts ...connect.ClientOption) EndpointInfoClient {
baseURL = strings.TrimRight(baseURL, "/")
return &endpointInfoClient{
info: connect.NewClient[v21.InfoRequest, v21.InfoResponse](
httpClient,
baseURL+EndpointInfoInfoProcedure,
connect.WithSchema(endpointInfoInfoMethodDescriptor),
connect.WithClientOptions(opts...),
),
}
}
// endpointInfoClient implements EndpointInfoClient.
type endpointInfoClient struct {
info *connect.Client[v21.InfoRequest, v21.InfoResponse]
}
// Info calls sf.substreams.rpc.v2.EndpointInfo.Info.
func (c *endpointInfoClient) Info(ctx context.Context, req *connect.Request[v21.InfoRequest]) (*connect.Response[v21.InfoResponse], error) {
return c.info.CallUnary(ctx, req)
}
// EndpointInfoHandler is an implementation of the sf.substreams.rpc.v2.EndpointInfo service.
type EndpointInfoHandler interface {
Info(context.Context, *connect.Request[v21.InfoRequest]) (*connect.Response[v21.InfoResponse], error)
}
// NewEndpointInfoHandler builds an HTTP handler from the service implementation. It returns the
// path on which to mount the handler and the handler itself.
//
// By default, handlers support the Connect, gRPC, and gRPC-Web protocols with the binary Protobuf
// and JSON codecs. They also support gzip compression.
func NewEndpointInfoHandler(svc EndpointInfoHandler, opts ...connect.HandlerOption) (string, http.Handler) {
endpointInfoInfoHandler := connect.NewUnaryHandler(
EndpointInfoInfoProcedure,
svc.Info,
connect.WithSchema(endpointInfoInfoMethodDescriptor),
connect.WithHandlerOptions(opts...),
)
return "/sf.substreams.rpc.v2.EndpointInfo/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.URL.Path {
case EndpointInfoInfoProcedure:
endpointInfoInfoHandler.ServeHTTP(w, r)
default:
http.NotFound(w, r)
}
})
}
// UnimplementedEndpointInfoHandler returns CodeUnimplemented from all methods.
type UnimplementedEndpointInfoHandler struct{}
func (UnimplementedEndpointInfoHandler) Info(context.Context, *connect.Request[v21.InfoRequest]) (*connect.Response[v21.InfoResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("sf.substreams.rpc.v2.EndpointInfo.Info is not implemented"))
}