starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-27 19:41:11 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

fix(idlewatcher): wake time outs before actual timeout

Browse Source
This commit is contained in:
yusing
2025-06-02 23:26:47 +08:00
parent 9087c4f195
commit e737737415
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/idlewatcher/handle_http.go
View File

@@ -90,16 +90,14 @@ func (w *Watcher) wakeFromHTTP(rw http.ResponseWriter, r *http.Request) (shouldN
return false
}
ctx, cancel := context.WithTimeoutCause(r.Context(), w.cfg.WakeTimeout, errors.New("wake timeout"))
defer cancel()
ctx := r.Context()
if w.cancelled(ctx) {
gphttp.ServerError(rw, r, context.Cause(ctx), http.StatusServiceUnavailable)
return false
}
w.l.Trace().Msg("signal received")
err := w.wakeIfStopped()
err := w.Wake(ctx)
if err != nil {
gphttp.ServerError(rw, r, err)
return false

7
internal/idlewatcher/watcher.go
View File

@@ -186,18 +186,17 @@ func (w *Watcher) Key() string {
return w.cfg.Key()
}
func (w *Watcher) Wake() error {
return w.wakeIfStopped()
func (w *Watcher) Wake(ctx context.Context) error {
}
func (w *Watcher) wakeIfStopped() error {
func (w *Watcher) wakeIfStopped(ctx context.Context) error {
state := w.state.Load()
if state.status == idlewatcher.ContainerStatusRunning {
w.l.Debug().Msg("container is already running")
return nil
}
ctx, cancel := context.WithTimeout(w.task.Context(), w.cfg.WakeTimeout)
ctx, cancel := context.WithTimeout(ctx, w.cfg.WakeTimeout)
defer cancel()
switch state.status {
case idlewatcher.ContainerStatusStopped: