snapshot of https://github.com/grpc/grpc-go.git (master, 2026-09-15; blobs > 131072B omitted)

interop/fake_grpclb/fake_grpclb.go

@@ -0,0 +1,83 @@
+/*
+ *
+ * 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)
+	}
+}