client: add sidechannel support

Implements gitlab-com/gl-infra/scalability#1303.

This change adds publicly exported code that allows Gitaly clients to accept sidechannel connections coming back from a Gitaly server they have connected to. This is done via a new dial function DialSidechannel.

We need this code to be public so that Workhorse can import it. Without this code, Workhorse won't be able to use sidechannel RPC's.

client/dial.go

@@ -3,8 +3,12 @@ package client
 import (
 	"context"
 
+	"gitlab.com/gitlab-org/gitaly/v14/internal/backchannel"
 	"gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/client"
+	"gitlab.com/gitlab-org/gitaly/v14/internal/listenmux"
+	"gitlab.com/gitlab-org/gitaly/v14/internal/sidechannel"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
 )
 
@@ -30,6 +34,22 @@ func Dial(rawAddress string, connOpts []grpc.DialOption) (*grpc.ClientConn, erro
 	return DialContext(context.Background(), rawAddress, connOpts)
 }
 
+// DialSidechannel configures the dialer to establish a Gitaly
+// backchannel connection instead of a regular gRPC connection. It also
+// injects sr as a sidechannel registry, so that Gitaly can establish
+// sidechannels back to the client.
+func DialSidechannel(ctx context.Context, rawAddress string, sr *SidechannelRegistry, connOpts []grpc.DialOption) (*grpc.ClientConn, error) {
+	clientHandshaker := backchannel.NewClientHandshaker(
+		sr.logger,
+		func() backchannel.Server {
+			lm := listenmux.New(insecure.NewCredentials())
+			lm.Register(sidechannel.NewServerHandshaker(sr.registry))
+			return grpc.NewServer(grpc.Creds(lm))
+		},
+	)
+	return client.Dial(ctx, rawAddress, connOpts, clientHandshaker)
+}
+
 // FailOnNonTempDialError helps to identify if remote listener is ready to accept new connections.
 func FailOnNonTempDialError() []grpc.DialOption {
 	return []grpc.DialOption{