74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
|
// 热重载驱动：fsnotify 文件监听 + 防抖（热更新.md §4.5）。
|
|||
|
|
package mods
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"path/filepath"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/fsnotify/fsnotify"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Watcher 脚本文件监听器（开发模式）。
|
|||
|
|
type Watcher struct {
|
|||
|
|
w *fsnotify.Watcher
|
|||
|
|
mu sync.Mutex
|
|||
|
|
reload map[string]func() error // 脚本路径 → 重载函数
|
|||
|
|
timers map[string]*time.Timer // 防抖计时器（500ms）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewWatcher 创建监听器。
|
|||
|
|
func NewWatcher() (*Watcher, error) {
|
|||
|
|
w, err := fsnotify.NewWatcher()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("mods.NewWatcher: %w", err)
|
|||
|
|
}
|
|||
|
|
return &Watcher{w: w, reload: make(map[string]func() error), timers: make(map[string]*time.Timer)}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WatchFile 监听一个脚本文件：变化后防抖触发 reload（热更新.md §4.5 防抖 500ms）。
|
|||
|
|
func (w *Watcher) WatchFile(path string, reload func() error) error {
|
|||
|
|
if err := w.w.Add(filepath.Dir(path)); err != nil {
|
|||
|
|
return fmt.Errorf("mods.WatchFile %s: %w", path, err)
|
|||
|
|
}
|
|||
|
|
w.mu.Lock()
|
|||
|
|
w.reload[path] = reload
|
|||
|
|
w.mu.Unlock()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Run 事件循环（goroutine）：文件变化 → 防抖 → 重载。
|
|||
|
|
func (w *Watcher) Run() {
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case ev, ok := <-w.w.Events:
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if ev.Op&(fsnotify.Write|fsnotify.Create) == 0 {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
w.mu.Lock()
|
|||
|
|
fn, ok := w.reload[ev.Name]
|
|||
|
|
if !ok {
|
|||
|
|
w.mu.Unlock()
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if t, exists := w.timers[ev.Name]; exists {
|
|||
|
|
t.Stop()
|
|||
|
|
}
|
|||
|
|
reloadFn := fn
|
|||
|
|
w.timers[ev.Name] = time.AfterFunc(500*time.Millisecond, func() { _ = reloadFn() })
|
|||
|
|
w.mu.Unlock()
|
|||
|
|
case _, ok := <-w.w.Errors:
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Close 停止监听。
|
|||
|
|
func (w *Watcher) Close() error { return w.w.Close() }
|