From 51b88061846cc761e976035240e4a9990fb8fb43 Mon Sep 17 00:00:00 2001 From: yusing Date: Fri, 13 Dec 2024 12:54:54 +0800 Subject: [PATCH] properly close docker client --- internal/route/http.go | 1 + internal/route/stream.go | 1 + 2 files changed, 2 insertions(+) diff --git a/internal/route/http.go b/internal/route/http.go index c6ffce5..e74d78f 100755 --- a/internal/route/http.go +++ b/internal/route/http.go @@ -98,6 +98,7 @@ func (r *HTTPRoute) Start(providerSubtask task.Task) E.Error { if err == nil { fallback := monitor.NewHTTPHealthChecker(r.rp.TargetURL, r.HealthCheck) r.HealthMon = monitor.NewDockerHealthMonitor(client, r.Idlewatcher.ContainerID, r.HealthCheck, fallback) + r.task.OnCancel("close docker client", client.Close) } } if r.HealthMon == nil { diff --git a/internal/route/stream.go b/internal/route/stream.go index 4031c3e..9d94cad 100755 --- a/internal/route/stream.go +++ b/internal/route/stream.go @@ -73,6 +73,7 @@ func (r *StreamRoute) Start(providerSubtask task.Task) E.Error { if err == nil { fallback := monitor.NewRawHealthChecker(r.TargetURL(), r.HealthCheck) r.HealthMon = monitor.NewDockerHealthMonitor(client, r.Idlewatcher.ContainerID, r.HealthCheck, fallback) + r.task.OnCancel("close docker client", client.Close) } } if r.HealthMon == nil {