package utils

import (
	"sync"
)

// Map exported
type Map struct {
	sync.Map
	mu sync.RWMutex
}

// Incr map
func (m *Map) Incr(key string, amount int) {
	m.mu.Lock()
	if count, ok := m.LoadOrStore(key, amount); ok {
		m.Store(key, count.(int)+amount)
	}
	m.mu.Unlock()
}

// Counter exported
type Counter struct {
	sync.RWMutex
	Data map[string]int
}

// Get exported
func (c *Counter) Get(key string) int {
	c.RLock()
	count := c.Data[key]
	c.RUnlock()
	return count
}

// Set exported
func (c *Counter) Set(key string, value int) {
	c.Lock()
	c.Data[key] = value
	c.Unlock()
}

// Incr exported
func (c *Counter) Incr(key string, amount int) int {
	c.Lock()
	c.Data[key] += amount
	count := c.Data[key]
	c.Unlock()
	return count
}

// Delete exported
func (c *Counter) Delete(key string) {
	c.Lock()
	delete(c.Data, key)
	c.Unlock()
}