status-im/status-go

package rpcstats

import (
    "sync"
)

type RPCUsageStats struct {
    total                  uint
    counterPerMethod       *sync.Map
    counterPerMethodPerTag *sync.Map
}

var stats *RPCUsageStats
var mu sync.Mutex

func getInstance() *RPCUsageStats {
    mu.Lock()
    defer mu.Unlock()

    if stats == nil {
        stats = &RPCUsageStats{}
        stats.counterPerMethod = &sync.Map{}
        stats.counterPerMethodPerTag = &sync.Map{}
    }
    return stats
}

func getStats() (uint, *sync.Map, *sync.Map) {
    stats := getInstance()
    return stats.total, stats.counterPerMethod, stats.counterPerMethodPerTag
}

func resetStats() {
    stats := getInstance()
    stats.total = 0
    stats.counterPerMethod = &sync.Map{}
    stats.counterPerMethodPerTag = &sync.Map{}
}

func CountCall(method string) {
    stats := getInstance()
    stats.total++
    value, _ := stats.counterPerMethod.LoadOrStore(method, uint(0))
    stats.counterPerMethod.Store(method, value.(uint)+1)
}

func CountCallWithTag(method string, tag string) {
    if tag == "" {
        CountCall(method)
        return
    }

    stats := getInstance()
    value, _ := stats.counterPerMethodPerTag.LoadOrStore(tag, &sync.Map{})
    methodMap := value.(*sync.Map)
    value, _ = methodMap.LoadOrStore(method, uint(0))
    methodMap.Store(method, value.(uint)+1)
    stats.total++
}