iframe-proxy | Sunbelt Computer Software

76 lines (64 loc) · 1.93 KB

package coderdtest
	"github.com/coder/coder/v2/coderd/database"
	"github.com/coder/coder/v2/coderd/usage"
	"github.com/coder/coder/v2/coderd/usage/usagetypes"
var _ usage.Inserter = (*UsageInserter)(nil)
type UsageInserter struct {
	sync.Mutex
	discreteEvents  []usagetypes.DiscreteEvent
	heartbeatEvents []usagetypes.HeartbeatEvent
	seenHeartbeats  map[string]struct{}
func NewUsageInserter() *UsageInserter {
	return &UsageInserter{
		discreteEvents:  []usagetypes.DiscreteEvent{},
		seenHeartbeats:  map[string]struct{}{},
		heartbeatEvents: []usagetypes.HeartbeatEvent{},
func (u *UsageInserter) InsertDiscreteUsageEvent(_ context.Context, _ database.Store, event usagetypes.DiscreteEvent) error {
	defer u.Unlock()
	u.discreteEvents = append(u.discreteEvents, event)
	return nil
func (u *UsageInserter) InsertHeartbeatUsageEvent(_ context.Context, _ database.Store, id string, event usagetypes.HeartbeatEvent) error {
	defer u.Unlock()
	if _, seen := u.seenHeartbeats[id]; seen {
		return nil
	u.seenHeartbeats[id] = struct{}{}
	u.heartbeatEvents = append(u.heartbeatEvents, event)
	return nil
func (u *UsageInserter) GetHeartbeatEvents() []usagetypes.HeartbeatEvent {
	defer u.Unlock()
	eventsCopy := make([]usagetypes.HeartbeatEvent, len(u.heartbeatEvents))
	copy(eventsCopy, u.heartbeatEvents)
	return eventsCopy
func (u *UsageInserter) GetDiscreteEvents() []usagetypes.DiscreteEvent {
	defer u.Unlock()
	eventsCopy := make([]usagetypes.DiscreteEvent, len(u.discreteEvents))
	copy(eventsCopy, u.discreteEvents)
	return eventsCopy
func (u *UsageInserter) TotalEventCount() int {
	defer u.Unlock()
	return len(u.discreteEvents) + len(u.heartbeatEvents)
func (u *UsageInserter) Reset() {
	defer u.Unlock()
	u.seenHeartbeats = map[string]struct{}{}
	u.discreteEvents = []usagetypes.DiscreteEvent{}
	u.heartbeatEvents = []usagetypes.HeartbeatEvent{}

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

usage.go

Latest commit

History

usage.go

File metadata and controls