41 lines
735 B
Go
41 lines
735 B
Go
|
package output
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
||
|
|
)
|
||
|
|
|
||
|
|
var quit chan struct{}
|
||
|
|
var wg = sync.WaitGroup{}
|
||
|
|
|
||
|
|
// Start workers of database
|
||
|
|
// WARNING: Do not override this function
|
||
|
|
// you should use New()
|
||
|
|
func Start(output Output, nodes *runtime.Nodes, config *runtime.Config) {
|
||
|
|
quit = make(chan struct{})
|
||
|
|
wg.Add(1)
|
||
|
|
go saveWorker(output, nodes, config.Nodes.SaveInterval.Duration)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Close() {
|
||
|
|
close(quit)
|
||
|
|
wg.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
// save periodically to output
|
||
|
|
func saveWorker(output Output, nodes *runtime.Nodes, saveInterval time.Duration) {
|
||
|
|
ticker := time.NewTicker(saveInterval)
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-ticker.C:
|
||
|
|
output.Save(nodes)
|
||
|
|
case <-quit:
|
||
|
|
wg.Done()
|
||
|
|
ticker.Stop()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|