func init() { vtgate.RegisterVTGates = append(vtgate.RegisterVTGates, func(vtGate vtgateservice.VTGateService) { if servenv.GRPCCheckServiceMap("vtgateservice") { vtgateservicepb.RegisterVitessServer(servenv.GRPCServer, &VTGate{vtGate}) } }) }
// RegisterForTest registers the gRPC implementation on the gRPC // server. Useful for unit tests only, for real use, the init() // function does the registration. func RegisterForTest(s *grpc.Server, service vtgateservice.VTGateService) { vtgateservicepb.RegisterVitessServer(s, &VTGate{service}) }