mirror of
https://gitee.com/johng/gf.git
synced 2024-11-29 18:57:44 +08:00
44 lines
1.2 KiB
Go
44 lines
1.2 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 file
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gogf/gf/v2/container/gtype"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/net/gsvc"
|
|
)
|
|
|
|
// Watcher for file changes watch.
|
|
type Watcher struct {
|
|
prefix string // Watched prefix key, not file name prefix.
|
|
discovery gsvc.Discovery // Service discovery.
|
|
ch chan gsvc.Service // Changes that caused by inotify.
|
|
closed *gtype.Bool // Whether the channel has been closed
|
|
}
|
|
|
|
// Proceed proceeds watch in blocking way.
|
|
// It returns all complete services that watched by `key` if any change.
|
|
func (w *Watcher) Proceed() (services []gsvc.Service, err error) {
|
|
if w.closed.Val() {
|
|
return nil, gerror.New("discovery service was closed")
|
|
}
|
|
<-w.ch
|
|
return w.discovery.Search(context.Background(), gsvc.SearchInput{
|
|
Prefix: w.prefix,
|
|
})
|
|
}
|
|
|
|
// Close closes the watcher.
|
|
func (w *Watcher) Close() error {
|
|
if w.closed.Cas(false, true) {
|
|
close(w.ch)
|
|
}
|
|
return nil
|
|
}
|