scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-09 22:51:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

stops calculation on server close

Browse source
This commit is contained in:
Samir Patel 2021-06-01 14:34:51 -05:00
parent edf71c1bf7
commit 6a8bf3acd6
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
api.go
View file

@ -990,6 +990,7 @@ func (api *API) calculateUsage() {
api.usageCache.muCalculate.Lock()
defer api.usageCache.muCalculate.Unlock()
api.server.wg.Add(1)
defer api.server.wg.Done()
lastUpdated := api.usageCache.lastUpdated
if time.Since(lastUpdated) > api.usageCache.refreshInterval {
@ -1038,23 +1039,34 @@ func (api *API) calculateUsage() {
api.usageCache.lastUpdated = lastUpdated
api.usageCache.muAssign.Unlock()
}
api.server.wg.Done()
}
// Periodically calculates disk usage
func (api *API) RefreshUsageCache(refresh time.Duration, trigger chan bool) {
defer close(trigger)
api.usageCache = &usageCache{
data: make(map[string]NodeUsage),
refreshInterval: refresh,
resetTrigger: trigger,
}
for {
api.calculateUsage()
select {
case <-trigger:
fmt.Println("Refresh Thread Resetting")
api.server.logger.Infof("Refresh Thread Resetting")
continue
case <-time.After(api.usageCache.refreshInterval):
continue
case <-api.server.closing:
fmt.Println("Refresh Thread Closing")
api.server.logger.Infof("Refresh Thread Closing")
return
default:
fmt.Println("Refresh Thread Calculating")
api.server.logger.Infof("Refresh Thread Calculating")
api.calculateUsage()
time.Sleep(api.usageCache.refreshInterval)
// case <-time.After(api.usageCache.refreshInterval):
// continue
}
}
}