OliveTin/service/internal/stringvariables/map.go

/**
 * The ephemeralvariablemap is used "only" for variable substitution in config
 * titles, shell arguments, etc, in the foorm of {{ key }}, like Jinja2.
 *
 * OliveTin itself really only ever "writes" to this map, mostly by loading
 * EntityFiles, and the only form of "reading" is for the variable substitution
 * in configs.
 */

package stringvariables

import (
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	"strings"
	"sync"
)

var (
	contents map[string]string

	metricSvCount = promauto.NewGauge(prometheus.GaugeOpts{
		Name: "olivetin_sv_count",
		Help: "The number entries in the sv map",
	})

	rwmutex = sync.RWMutex{}
)

func init() {
	rwmutex.Lock()

	contents = make(map[string]string)

	rwmutex.Unlock()
}

func Get(key string) string {
	rwmutex.RLock()

	v, ok := contents[key]

	rwmutex.RUnlock()

	if !ok {
		return ""
	} else {
		return v
	}
}

func GetAll() map[string]string {
	return contents
}

func Set(key string, value string) {
	rwmutex.Lock()

	contents[key] = value

	metricSvCount.Set(float64(len(contents)))

	rwmutex.Unlock()
}

func RemoveKeysThatStartWith(search string) {
	rwmutex.Lock()

	for k, _ := range contents {
		if strings.HasPrefix(k, search) {
			delete(contents, k)
		}
	}

	rwmutex.Unlock()
}