smegmesh/pkg/lib/conv.go

package lib

import (
	logging "github.com/tim-beatham/wgmesh/pkg/log"
)

// MapToSlice converts a map to a slice in go
func MapValues[K comparable, V any](m map[K]V) []V {
	return MapValuesWithExclude(m, map[K]struct{}{})
}

func MapValuesWithExclude[K comparable, V any](m map[K]V, exclude map[K]struct{}) []V {
	values := make([]V, len(m)-len(exclude))

	i := 0

	if len(m)-len(exclude) <= 0 {
		return values
	}

	for k, v := range m {
		if _, excluded := exclude[k]; excluded {
			continue
		}

		logging.Log.WriteInfof("Key %s", k)

		values[i] = v
		i++
	}

	return values
}

func MapKeys[K comparable, V any](m map[K]V) []K {
	values := make([]K, len(m))

	i := 0
	for k, _ := range m {
		values[i] = k
		i++
	}

	return values
}