aboutsummaryrefslogtreecommitdiff
path: root/utils/collections/orderedmap.go
blob: b9f17b77bcb1c5eec2f538ba6b9fe47ea1affe1b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package collections

type OrderedMap[K comparable, V any] struct {
	keys   []K
	values map[K]V
}

func OrderedMapOf[K comparable, V any]() OrderedMap[K, V] {
	return OrderedMap[K, V]{
		keys:   make([]K, 0),
		values: make(map[K]V),
	}
}

func (orderedMap *OrderedMap[K, V]) Set(key K, value V) {
	if _, exists := orderedMap.values[key]; !exists {
		orderedMap.keys = append(orderedMap.keys, key)
	}
	orderedMap.values[key] = value
}

func (orderedMap *OrderedMap[K, V]) Get(key K) (V, bool) {
	value, exists := orderedMap.values[key]
	return value, exists
}

func (orderedMap *OrderedMap[K, V]) All() []V {
	result := make([]V, 0, len(orderedMap.keys))
	for _, key := range orderedMap.keys {
		result = append(result, orderedMap.values[key])
	}
	return result
}

func (orderedMap *OrderedMap[K, V]) Len() int {
	return len(orderedMap.keys)
}