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

add locks

Browse source
This commit is contained in:
Samir Patel 2021-05-28 18:36:21 -05:00 • committed by Samir Patel
parent 5f46635094
commit a5ae6c15fc
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
api.go
View file

@ -234,6 +234,7 @@ func (api *API) CreateIndex(ctx context.Context, indexName string, options Index
return nil, errors.Wrap(err, "creating index")
}
api.ResetUsageCache()
api.holder.Stats.Count(MetricCreateIndex, 1, 1.0)
return index, nil
}
@ -978,9 +979,9 @@ func (api *API) requestUsageOfNodes() {
api.server.logger.Infof("couldn't collect disk usage from %s: %s", node.URI, err)
}
api.usageCache.muWrite.Lock()
api.usageCache.muRead.Lock()
api.usageCache.data[node.ID] = nodeUsage[node.ID]
api.usageCache.muWrite.Unlock()
api.usageCache.muRead.Unlock()
}
}
@ -988,7 +989,7 @@ func (api *API) requestUsageOfNodes() {
func (api *API) calculateUsage() {
api.usageCache.muWrite.Lock()
defer api.usageCache.muWrite.Unlock()
api.server.wg.Add(1)
// api.server.wg.Add(1)
lastUpdated := api.usageCache.lastUpdated
if time.Since(lastUpdated) > api.usageCache.refreshInterval {
@ -1032,12 +1033,13 @@ func (api *API) calculateUsage() {
},
LastUpdated: lastUpdated,
}
fmt.Printf("node Usage: %+v\n", nodeUsage)
api.usageCache.muRead.Lock()
api.usageCache.data = make(map[string]NodeUsage)
api.usageCache.data[api.server.nodeID] = nodeUsage
api.usageCache.lastUpdated = lastUpdated
api.usageCache.muRead.Unlock()
}
api.server.wg.Done()
// api.server.wg.Done()
}
// Periodically calculates disk usage
@ -1062,7 +1064,9 @@ func (api *API) RefreshUsageCache(refresh time.Duration, trigger chan bool) {
func (api *API) ResetUsageCache() error {
fmt.Println("Reset Cache")
if api.usageCache != nil {
api.usageCache.muRead.Lock()
api.usageCache.lastUpdated = time.Time{}
api.usageCache.muRead.Unlock()
} else {
return errors.New("invalidating cache: cache not initialized")
}