72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
/*
|
|
*
|
|
* Copyright 2017 gRPC authors.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
// This file contains the generated server side code.
|
|
// It's only used for grpclb testing.
|
|
|
|
package grpclb
|
|
|
|
import (
|
|
"google.golang.org/grpc"
|
|
lbpb "google.golang.org/grpc/grpclb/grpc_lb_v1"
|
|
)
|
|
|
|
// Server API for LoadBalancer service
|
|
|
|
type loadBalancerServer interface {
|
|
// Bidirectional rpc to get a list of servers.
|
|
BalanceLoad(*loadBalancerBalanceLoadServer) error
|
|
}
|
|
|
|
func registerLoadBalancerServer(s *grpc.Server, srv loadBalancerServer) {
|
|
s.RegisterService(
|
|
&grpc.ServiceDesc{
|
|
ServiceName: "grpc.lb.v1.LoadBalancer",
|
|
HandlerType: (*loadBalancerServer)(nil),
|
|
Methods: []grpc.MethodDesc{},
|
|
Streams: []grpc.StreamDesc{
|
|
{
|
|
StreamName: "BalanceLoad",
|
|
Handler: balanceLoadHandler,
|
|
ServerStreams: true,
|
|
ClientStreams: true,
|
|
},
|
|
},
|
|
Metadata: "grpclb.proto",
|
|
}, srv)
|
|
}
|
|
|
|
func balanceLoadHandler(srv interface{}, stream grpc.ServerStream) error {
|
|
return srv.(loadBalancerServer).BalanceLoad(&loadBalancerBalanceLoadServer{stream})
|
|
}
|
|
|
|
type loadBalancerBalanceLoadServer struct {
|
|
grpc.ServerStream
|
|
}
|
|
|
|
func (x *loadBalancerBalanceLoadServer) Send(m *lbpb.LoadBalanceResponse) error {
|
|
return x.ServerStream.SendMsg(m)
|
|
}
|
|
|
|
func (x *loadBalancerBalanceLoadServer) Recv() (*lbpb.LoadBalanceRequest, error) {
|
|
m := new(lbpb.LoadBalanceRequest)
|
|
if err := x.ServerStream.RecvMsg(m); err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|