depot/packages/networking/ipfs-cluster/monitor/metrics/store.go

package metrics

import (
	"sort"
	"sync"

	"github.com/ipfs-cluster/ipfs-cluster/api"

	peer "github.com/libp2p/go-libp2p/core/peer"
)

// PeerMetrics maps a peer IDs to a metrics window.
type PeerMetrics map[peer.ID]*Window

// Store can be used to store and access metrics.
type Store struct {
	mux    sync.RWMutex
	byName map[string]PeerMetrics
}

// NewStore can be used to create a Store.
func NewStore() *Store {
	return &Store{
		byName: make(map[string]PeerMetrics),
	}
}

// Add inserts a new metric in Metrics.
func (mtrs *Store) Add(m api.Metric) {
	mtrs.mux.Lock()
	defer mtrs.mux.Unlock()

	name := m.Name
	peer := m.Peer
	mbyp, ok := mtrs.byName[name]
	if !ok {
		mbyp = make(PeerMetrics)
		mtrs.byName[name] = mbyp
	}
	window, ok := mbyp[peer]
	if !ok {
		// We always lock the outer map, so we can use unsafe
		// Window.
		window = NewWindow(DefaultWindowCap)
		mbyp[peer] = window
	}

	window.Add(m)
}

// RemovePeer removes all metrics related to a peer from the Store.
func (mtrs *Store) RemovePeer(pid peer.ID) {
	mtrs.mux.Lock()
	for _, metrics := range mtrs.byName {
		delete(metrics, pid)
	}
	mtrs.mux.Unlock()
}

// RemovePeerMetrics removes all metrics of a given name for a given peer ID.
func (mtrs *Store) RemovePeerMetrics(pid peer.ID, name string) {
	mtrs.mux.Lock()
	metrics := mtrs.byName[name]
	delete(metrics, pid)
	mtrs.mux.Unlock()
}

// LatestValid returns all the last known valid metrics of a given type. A metric
// is valid if it has not expired.
func (mtrs *Store) LatestValid(name string) []api.Metric {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	byPeer, ok := mtrs.byName[name]
	if !ok {
		return []api.Metric{}
	}

	metrics := make([]api.Metric, 0, len(byPeer))
	for _, window := range byPeer {
		m, err := window.Latest()
		// TODO(ajl): for accrual, does it matter if a ping has expired?
		if err != nil || m.Discard() {
			continue
		}
		metrics = append(metrics, m)
	}

	sortedMetrics := api.MetricSlice(metrics)
	sort.Stable(sortedMetrics)
	return sortedMetrics
}

// AllMetrics returns the latest metrics for all peers and metrics types.  It
// may return expired metrics.
func (mtrs *Store) AllMetrics() []api.Metric {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	result := make([]api.Metric, 0)

	for _, byPeer := range mtrs.byName {
		for _, window := range byPeer {
			metric, err := window.Latest()
			if err != nil || !metric.Valid {
				continue
			}
			result = append(result, metric)
		}
	}
	return result
}

// PeerMetrics returns the latest metrics for a given peer ID for
// all known metrics types. It may return expired metrics.
func (mtrs *Store) PeerMetrics(pid peer.ID) []api.Metric {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	result := make([]api.Metric, 0)

	for _, byPeer := range mtrs.byName {
		window, ok := byPeer[pid]
		if !ok {
			continue
		}
		metric, err := window.Latest()
		if err != nil || !metric.Valid {
			continue
		}
		result = append(result, metric)
	}
	return result
}

// PeerMetricAll returns all of a particular metrics for a
// particular peer.
func (mtrs *Store) PeerMetricAll(name string, pid peer.ID) []api.Metric {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	byPeer, ok := mtrs.byName[name]
	if !ok {
		return nil
	}

	window, ok := byPeer[pid]
	if !ok {
		return nil
	}
	ms := window.All()
	return ms
}

// PeerLatest returns the latest of a particular metric for a
// particular peer. It may return an expired metric.
func (mtrs *Store) PeerLatest(name string, pid peer.ID) api.Metric {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	byPeer, ok := mtrs.byName[name]
	if !ok {
		return api.Metric{}
	}

	window, ok := byPeer[pid]
	if !ok {
		return api.Metric{}
	}
	m, err := window.Latest()
	if err != nil {
		// ignoring error, as nil metric is indicative enough
		return api.Metric{}
	}
	return m
}

// MetricNames returns all the known metric names
func (mtrs *Store) MetricNames() []string {
	mtrs.mux.RLock()
	defer mtrs.mux.RUnlock()

	list := make([]string, 0, len(mtrs.byName))
	for k := range mtrs.byName {
		list = append(list, k)
	}
	return list
}