731f40da34
- Fix Debouncer race condition with sequence numbers to prevent double-fire - Add parent directory watch to catch file creates/renames - Add .git/HEAD watch for branch change detection - Implement retry/backoff (50-400ms) for re-establishing JSONL watch - Handle Create/Chmod events in addition to Write - Add .git/HEAD polling in polling mode - All 18 debouncer and watcher tests pass Amp-Thread-ID: https://ampcode.com/threads/T-4029d643-b4b4-4d3b-bd85-74461f78cd7f Co-authored-by: Amp <amp@ampcode.com>
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// Debouncer batches rapid events into a single action after a quiet period.
|
|
// Thread-safe for concurrent triggers.
|
|
type Debouncer struct {
|
|
mu sync.Mutex
|
|
timer *time.Timer
|
|
duration time.Duration
|
|
action func()
|
|
seq uint64 // Sequence number to prevent stale timer fires
|
|
}
|
|
|
|
// NewDebouncer creates a new debouncer with the given duration and action.
|
|
// The action will be called once after the duration has passed since the last trigger.
|
|
func NewDebouncer(duration time.Duration, action func()) *Debouncer {
|
|
return &Debouncer{
|
|
duration: duration,
|
|
action: action,
|
|
}
|
|
}
|
|
|
|
// Trigger schedules the action to run after the debounce duration.
|
|
// If called multiple times, the timer is reset each time, ensuring
|
|
// the action only fires once after the last trigger.
|
|
func (d *Debouncer) Trigger() {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
|
|
if d.timer != nil {
|
|
d.timer.Stop()
|
|
}
|
|
|
|
// Increment sequence number to invalidate any pending timers
|
|
d.seq++
|
|
currentSeq := d.seq
|
|
|
|
d.timer = time.AfterFunc(d.duration, func() {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
|
|
// Only fire if this is still the latest trigger
|
|
if d.seq == currentSeq {
|
|
d.timer = nil
|
|
d.mu.Unlock() // Unlock before calling action to avoid holding lock during callback
|
|
d.action()
|
|
d.mu.Lock() // Re-lock for defer
|
|
}
|
|
})
|
|
}
|
|
|
|
// Cancel stops any pending debounced action.
|
|
// Safe to call even if no action is pending.
|
|
func (d *Debouncer) Cancel() {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
|
|
if d.timer != nil {
|
|
d.timer.Stop()
|
|
d.timer = nil
|
|
}
|
|
}
|