gods/src/mapx/main.go

package mapx

// The type implements map type where
// you can get, set and delete by value
// since it store everything as ONLY entity
// both for keys and values way.
// Use only when you do not care about the order.
type UniqMap[K, V comparable] struct {
	store map[K] V
	rstore map[V] K
}

// Returns new empty UniqMap.
func NewUniq[K, V comparable]() *UniqMap[K, V] {
	return &UniqMap[K, V]{
		make(map[K] V),
		make(map[V] K),
	}
}

// Sets new value v for the k key.
func (m *UniqMap[K, V]) Set(k K, v V) {
	m.store[k] = v
	m.rstore[v] = k
}

// Get value by the k key.
func (m *UniqMap[K, V]) Get(k K) (V, bool) {
	v, ok := m.store[k]
	return v, ok
}

func (m *UniqMap[K, V]) GetByValue(v V) (K, bool) {
	k, ok := m.rstore[v]
	return k, ok
}

func Keys[K comparable, V any](m map[K] V) []K {
	r := make([]K, 0, len(m))
	for k := range m {
		r = append(r, k)
	}

	return r
}

func Values[K comparable, V any](m map[K] V) []V {
	r := make([]V, 0, len(m))
	for _, v := range m {
		r = append(r, v)
	}

	return r
}

func Reverse[K, V comparable](m map[K] V) map[V] K {
	r := make(map[V] K)
	for k, v := range m {
		r[v] = k
	}

	return r
}