Merge branch '21535-multi-wf-delete'

[arvados.git] / services / ws / service.go

diff --git a/services/ws/service.go b/services/ws/service.go
index c38dcf59e68aac5a975f3362e7b258310341aef5..9a4a239ea195952a01a3726d2e53c52c892f0e93 100644 (file)
--- a/services/ws/service.go
+++ b/services/ws/service.go
@@ -7,6 +7,7 @@ package ws
 import (
        "context"
        "fmt"
+       "time"
 
        "git.arvados.org/arvados.git/lib/cmd"
        "git.arvados.org/arvados.git/lib/service"
@@ -24,11 +25,13 @@ func newHandler(ctx context.Context, cluster *arvados.Cluster, token string, reg
        if err != nil {
                return service.ErrorHandler(ctx, cluster, fmt.Errorf("error initializing client from cluster config: %s", err))
        }
+       client.Timeout = time.Minute
        eventSource := &pgEventSource{
                DataSource:   cluster.PostgreSQL.Connection.String(),
                MaxOpenConns: cluster.PostgreSQL.ConnectionPool,
                QueueSize:    cluster.API.WebsocketServerEventQueue,
                Logger:       ctxlog.FromContext(ctx),
+               Reg:          reg,
        }
        done := make(chan struct{})
        go func() {
@@ -44,8 +47,9 @@ func newHandler(ctx context.Context, cluster *arvados.Cluster, token string, reg
                cluster:        cluster,
                client:         client,
                eventSource:    eventSource,
-               newPermChecker: func() permChecker { return newPermChecker(*client) },
+               newPermChecker: func() permChecker { return newPermChecker(client) },
                done:           done,
+               reg:            reg,
        }
        return rtr
 }