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

add read lock

Browse source
This commit is contained in:
Samir Patel 2021-05-26 13:22:53 -05:00 • committed by Samir Patel
parent 7d56414557
commit 14533d88a1
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
api.go
View file

@ -899,7 +899,8 @@ func (api *API) PrimaryNode() *topology.Node {
type usageCache struct {
data map[string]NodeUsage
lastUpdated time.Time
mu sync.Mutex
muWrite sync.Mutex
muRead sync.Mutex
refreshInterval time.Duration
}
@ -945,6 +946,8 @@ type MemoryUsage struct {
func (api *API) Usage(ctx context.Context, remote bool) (map[string]NodeUsage, error) {
span, _ := tracing.StartSpanFromContext(ctx, "API.Usage")
defer span.Finish()
api.usageCache.muRead.Lock()
defer api.usageCache.muRead.Unlock()
var t time.Time
if api.usageCache.lastUpdated == t {
@ -972,12 +975,11 @@ func (api *API) requestUsageOfNodes() {
// Calculates disk usage from scratch for each index and stores the results in the usage cache
func (api *API) calculateUsage() {
api.usageCache.mu.Lock()
defer api.usageCache.mu.Unlock()
api.usageCache.muWrite.Lock()
defer api.usageCache.muWrite.Unlock()
if time.Since(api.usageCache.lastUpdated) > api.usageCache.refreshInterval {
fmt.Printf("RefreshRate, expired: time: %v, current time: %v \n", api.usageCache.lastUpdated, time.Now())
api.usageCache.data = make(map[string]NodeUsage)
indexDetails, nodeMetadataBytes, err := api.holder.Txf().IndexUsageDetails()
if err != nil {
@ -1017,6 +1019,9 @@ func (api *API) calculateUsage() {
},
LastUpdated: time.Now(),
}
api.usageCache.muRead.Lock()
defer api.usageCache.muRead.Unlock()
api.usageCache.data = make(map[string]NodeUsage)
api.usageCache.data[api.server.nodeID] = nodeUsage
}
}