arikawa/internal/moreatomic/syncmap.go

package moreatomic

import (
	"sync/atomic"

	"github.com/diamondburned/arikawa/v2/internal/moreatomic/syncmod"
)

// Map is a thread-safe map that is a wrapper around sync.Map with slight API
// additions.
type Map struct {
	val  atomic.Value
	ctor func() interface{}
}

func NewMap(ctor func() interface{}) *Map {
	sm := &Map{ctor: ctor}
	sm.Reset()
	return sm
}

// Reset swaps the internal map out with a fresh one, dropping the old map. This
// method never errors.
func (sm *Map) Reset() error {
	sm.val.Store(&syncmod.Map{New: sm.ctor})
	return nil
}

// LoadOrStore loads an existing value or stores a new value created from the
// given constructor then return that value.
func (sm *Map) LoadOrStore(k interface{}) (lv interface{}, loaded bool) {
	return sm.val.Load().(*syncmod.Map).LoadOrStore(k)
}

// Load loads an existing value; it returns ok set to false if there is no
// value with that key.
func (sm *Map) Load(k interface{}) (lv interface{}, ok bool) {
	return sm.val.Load().(*syncmod.Map).Load(k)
}