mirror of
https://gitee.com/johng/gf
synced 2026-06-06 16:21:40 +08:00
ac3efe5a00
为`gcfg`添加配置文件变更自定义回调，实现了`WatcherAdapter`接口，以下是`AdapterFile`的用法
test.yaml
```
b: "b"
```
```
package main
import (
"fmt"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gcfg"
"github.com/gogf/gf/v2/os/gctx"
)
func main() {
ctx := gctx.New()
file, _ := gcfg.NewAdapterFile("test.yaml")
file.Data(ctx)
file.AddWatcher("test", func() {
value := file.MustGet(ctx, "b")
fmt.Println(value.String())
})
server := g.Server()
server.Run()
}
```
使用`g`和默认配置文件
```
file := g.Cfg().GetAdapter().(*gcfg.AdapterFile)
file.AddWatcher("test", func() {
})
file := g.Cfg().GetAdapter().(*gcfg.AdapterFile)
file.RemoveWatcher("test")
```
注意：由于`gf`的`AdapterFile`使用的监听到文件变化删除缓存下一次重新初始化的懒加载方案，所有除了默认加载的`config.xxx`文件外，自定义的配置文件像`test.yaml`之类的都需要在`AddWatcher`前主动读取一次数据进行初始化监听(
`g.Cfg("test").Data(ctx)`)
---------
Co-authored-by: hailaz <739476267@qq.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Hunk Zhu <hunk@joy999.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gcfg
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gogf/gf/v2/container/gmap"
|
|
"github.com/gogf/gf/v2/internal/intlog"
|
|
)
|
|
|
|
// WatcherRegistry is a helper type for managing configuration watchers.
|
|
// It provides a unified implementation of watcher management to avoid code duplication
|
|
// across different adapter implementations.
|
|
type WatcherRegistry struct {
|
|
watchers *gmap.StrAnyMap // Watchers map storing watcher callbacks.
|
|
}
|
|
|
|
// NewWatcherRegistry creates and returns a new WatcherRegistry instance.
|
|
func NewWatcherRegistry() *WatcherRegistry {
|
|
return &WatcherRegistry{
|
|
watchers: gmap.NewStrAnyMap(true),
|
|
}
|
|
}
|
|
|
|
// Add adds a watcher with the specified name and callback function.
|
|
func (r *WatcherRegistry) Add(name string, fn func(ctx context.Context)) {
|
|
r.watchers.Set(name, fn)
|
|
}
|
|
|
|
// Remove removes the watcher with the specified name.
|
|
func (r *WatcherRegistry) Remove(name string) {
|
|
r.watchers.Remove(name)
|
|
}
|
|
|
|
// GetNames returns all watcher names.
|
|
func (r *WatcherRegistry) GetNames() []string {
|
|
return r.watchers.Keys()
|
|
}
|
|
|
|
// Notify notifies all registered watchers by calling their callback functions.
|
|
// Each callback is executed in a separate goroutine with panic recovery to prevent
|
|
// one watcher's panic from affecting others.
|
|
func (r *WatcherRegistry) Notify(ctx context.Context) {
|
|
r.watchers.Iterator(func(k string, v any) bool {
|
|
if fn, ok := v.(func(ctx context.Context)); ok {
|
|
go func(k string, fn func(ctx context.Context), ctx context.Context) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
intlog.Errorf(ctx, "watcher %s panic: %v", k, r)
|
|
}
|
|
}()
|
|
fn(ctx)
|
|
}(k, fn, ctx)
|
|
}
|
|
return true
|
|
})
|
|
}
|