Merge branch '19413-warning-update' refs #19413

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

diff --git a/services/ws/service.go b/services/ws/service.go
index fb313bb799be2ccfcd68d323f5828ac12c79ba8f..761e22e16c2cd4fd5025341e9ad284eb74b6f8c7 100644 (file)
--- a/services/ws/service.go
+++ b/services/ws/service.go
@@ -7,7 +7,6 @@ package ws
 import (
        "context"
        "fmt"
-       "os"
 
        "git.arvados.org/arvados.git/lib/cmd"
        "git.arvados.org/arvados.git/lib/service"
@@ -30,13 +29,13 @@ func newHandler(ctx context.Context, cluster *arvados.Cluster, token string, reg
                MaxOpenConns: cluster.PostgreSQL.ConnectionPool,
                QueueSize:    cluster.API.WebsocketServerEventQueue,
                Logger:       ctxlog.FromContext(ctx),
+               Reg:          reg,
        }
+       done := make(chan struct{})
        go func() {
                eventSource.Run()
                ctxlog.FromContext(ctx).Error("event source stopped")
-               if !testMode {
-                       os.Exit(1)
-               }
+               close(done)
        }()
        eventSource.WaitReady()
        if err := eventSource.DBHealth(); err != nil {
@@ -47,6 +46,8 @@ func newHandler(ctx context.Context, cluster *arvados.Cluster, token string, reg
                client:         client,
                eventSource:    eventSource,
                newPermChecker: func() permChecker { return newPermChecker(*client) },
+               done:           done,
+               reg:            reg,
        }
        return rtr
 }