e241b33f26
Some checks failed
codecov / upload (push) Failing after 13m59s
CodeQL / Analyze (push) Failing after 1s
Testing / vet-proto (push) Failing after 1s
Testing / tests (arm, latest) (push) Waiting to run
Testing / extras (latest) (push) Failing after 1s
Testing / static checks (latest-1) (push) Failing after 1s
Testing / tests (-race, latest) (push) Failing after 1s
Testing / tests (i386, latest) (push) Failing after 1s
Testing / tests (latest) (push) Failing after 1s
Testing / tests (latest-1) (push) Failing after 1s
Lock Threads / lock (push) Failing after 1s
govulncheck / govulncheck (.) (push) Failing after 1s
govulncheck / govulncheck (./cmd/protoc-gen-go-grpc) (push) Failing after 1s
govulncheck / govulncheck (./gcp/observability) (push) Failing after 1s
govulncheck / govulncheck (./security/advancedtls) (push) Failing after 1s
govulncheck / govulncheck (./stats/opencensus) (push) Failing after 1s
Stale bot / stale (push) Failing after 1s
84 lines
2.8 KiB
Go
84 lines
2.8 KiB
Go
/*
|
|
*
|
|
* Copyright 2018 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 is for testing only. Runs a fake grpclb balancer server.
|
|
// The name of the service to load balance for and the addresses
|
|
// of that service are provided by command line flags.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"strings"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials"
|
|
"google.golang.org/grpc/credentials/alts"
|
|
"google.golang.org/grpc/grpclog"
|
|
"google.golang.org/grpc/internal/testutils/fakegrpclb"
|
|
"google.golang.org/grpc/testdata"
|
|
)
|
|
|
|
var (
|
|
port = flag.Int("port", 10000, "Port to listen on.")
|
|
backendAddrs = flag.String("backend_addrs", "", "Comma separated list of backend IP/port addresses.")
|
|
useALTS = flag.Bool("use_alts", false, "Listen on ALTS credentials.")
|
|
useTLS = flag.Bool("use_tls", false, "Listen on TLS credentials, using a test certificate.")
|
|
shortStream = flag.Bool("short_stream", false, "End the balancer stream immediately after sending the first server list.")
|
|
serviceName = flag.String("service_name", "UNSET", "Name of the service being load balanced for.")
|
|
|
|
logger = grpclog.Component("interop")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
var opts []grpc.ServerOption
|
|
if *useTLS {
|
|
certFile := testdata.Path("server1.pem")
|
|
keyFile := testdata.Path("server1.key")
|
|
creds, err := credentials.NewServerTLSFromFile(certFile, keyFile)
|
|
if err != nil {
|
|
logger.Fatalf("Failed to generate credentials: %v", err)
|
|
}
|
|
opts = append(opts, grpc.Creds(creds))
|
|
} else if *useALTS {
|
|
altsOpts := alts.DefaultServerOptions()
|
|
altsTC := alts.NewServerCreds(altsOpts)
|
|
opts = append(opts, grpc.Creds(altsTC))
|
|
}
|
|
|
|
rawBackendAddrs := strings.Split(*backendAddrs, ",")
|
|
server, err := fakegrpclb.NewServer(fakegrpclb.ServerParams{
|
|
ListenPort: *port,
|
|
ServerOptions: opts,
|
|
LoadBalancedServiceName: *serviceName,
|
|
LoadBalancedServicePort: 443, // TODO: make this configurable?
|
|
BackendAddresses: rawBackendAddrs,
|
|
ShortStream: *shortStream,
|
|
})
|
|
if err != nil {
|
|
logger.Fatalf("Failed to create balancer server: %v", err)
|
|
}
|
|
|
|
// Serve() starts serving and blocks until Stop() is called. We don't need to
|
|
// call Stop() here since we want the server to run until we are killed.
|
|
if err := server.Serve(); err != nil {
|
|
logger.Fatalf("Failed to start balancer server: %v", err)
|
|
}
|
|
}
|