diff options
Diffstat (limited to 'internal/gitaly/gitaly.go')
-rw-r--r-- | internal/gitaly/gitaly.go | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/internal/gitaly/gitaly.go b/internal/gitaly/gitaly.go index 9f73661..f11c72a 100644 --- a/internal/gitaly/gitaly.go +++ b/internal/gitaly/gitaly.go @@ -42,7 +42,11 @@ func (c *Client) InitSidechannelRegistry(ctx context.Context) { c.SidechannelRegistry = gitalyclient.NewSidechannelRegistry(log.ContextLogger(ctx)) } -func (c *Client) GetConnection(ctx context.Context, cmd Command) (*grpc.ClientConn, error) { +func (c *Client) GetConnection(ctx context.Context, reuseConnections bool, cmd Command) (*grpc.ClientConn, error) { + if !reuseConnections { + return c.newConnection(ctx, cmd) + } + c.cache.RLock() conn := c.cache.connections[cmd] c.cache.RUnlock() |