cost-model/pkg/metrics/metricsconfig.go

package metrics

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"sync"

	"github.com/kubecost/cost-model/pkg/env"
	"github.com/kubecost/cost-model/pkg/util/watcher"
)

var metricsConfigLock = new(sync.Mutex)

type MetricsConfig struct {
	DisabledMetrics []string `json:"disabledMetrics"`
}

// Gets map of disabled metrics to empty structs
func (mc MetricsConfig) GetDisabledMetricsMap() map[string]struct{} {
	disabledMetricsMap := make(map[string]struct{})

	for i := range mc.DisabledMetrics {
		disabledMetricsMap[mc.DisabledMetrics[i]] = struct{}{}
	}

	return disabledMetricsMap
}

// Unmarshals metrics.json to a MetricsConfig struct
func GetMetricsConfig() (*MetricsConfig, error) {
	metricsConfigLock.Lock()
	defer metricsConfigLock.Unlock()
	mc := &MetricsConfig{}
	body, err := ioutil.ReadFile("/var/configs/metrics.json")
	if os.IsNotExist(err) {

		return mc, nil
	} else if err != nil {
		return mc, fmt.Errorf("error reading metrics config file: %s", err)
	}

	err = json.Unmarshal(body, mc)
	if err != nil {
		return mc, fmt.Errorf("error decoding metrics config: %s", err)
	}

	return mc, nil
}

// Writes MetricsConfig struct to json file
func UpdateMetricsConfig(mc *MetricsConfig) (*MetricsConfig, error) {
	metricsConfigLock.Lock()
	defer metricsConfigLock.Unlock()

	mcb, err := json.Marshal(mc)
	if err != nil {
		return nil, fmt.Errorf("error encoding metrics config struct: %s", err)
	}

	err = ioutil.WriteFile("/var/configs/metrics.json", mcb, 0644)
	if err != nil {
		return nil, fmt.Errorf("error writing to metrics config file: %s", err)
	}

	return mc, nil
}

// Updates metric config file from configmap
func UpdateMetricsConfigFromConfigmap(data map[string]string) error {

	mc := &MetricsConfig{}
	key := "metrics.json"

	cdata, ok := data[key]
	if !ok {
		return fmt.Errorf("error finding metrics config data")
	}

	err := json.Unmarshal([]byte(cdata), &mc)
	if err != nil {
		return fmt.Errorf("failed to unmarshal metrics configs: %s", err)
	}

	_, err = UpdateMetricsConfig(mc)
	if err != nil {
		return err
	}

	return nil

}

// Returns ConfigMapWatcher for metrics configuration configmap
func GetMetricsConfigWatcher() *watcher.ConfigMapWatcher {
	return &watcher.ConfigMapWatcher{
		ConfigMapName: env.GetMetricsConfigmapName(),
		WatchFunc: func(name string, data map[string]string) error {
			err := UpdateMetricsConfigFromConfigmap(data)
			return err
		},
	}
}