signoz/pkg/query-service/app/dashboards/provision.go

package dashboards

import (
	"context"
	"encoding/json"
	"os"
	"path/filepath"
	"strings"

	"go.uber.org/zap"

	"go.signoz.io/signoz/pkg/query-service/constants"
	"go.signoz.io/signoz/pkg/query-service/interfaces"
	"go.signoz.io/signoz/pkg/query-service/model"
)

func readCurrentDir(dir string, fm interfaces.FeatureLookup) error {
	file, err := os.Open(dir)
	if err != nil {
		zap.L().Warn("failed opening directory", zap.Error(err))
		return nil
	}
	defer file.Close()

	list, _ := file.Readdirnames(0) // 0 to read all files and folders
	for _, filename := range list {
		if strings.ToLower(filepath.Ext(filename)) != ".json" {
			zap.L().Debug("Skipping non-json file", zap.String("filename", filename))
			continue
		}
		zap.L().Info("Provisioning dashboard: ", zap.String("filename", filename))

		// using filepath.Join for platform specific path creation
		// which is equivalent to "dir+/+filename" (on unix based systems) but cleaner
		plan, err := os.ReadFile(filepath.Join(dir, filename))
		if err != nil {
			zap.L().Error("Creating Dashboards: Error in reading json fron file", zap.String("filename", filename), zap.Error(err))
			continue
		}
		var data map[string]interface{}
		err = json.Unmarshal(plan, &data)
		if err != nil {
			zap.L().Error("Creating Dashboards: Error in unmarshalling json from file", zap.String("filename", filename), zap.Error(err))
			continue
		}
		err = IsPostDataSane(&data)
		if err != nil {
			zap.L().Info("Creating Dashboards: Error in file", zap.String("filename", filename), zap.Error(err))
			continue
		}

		id := data["uuid"]
		if id == nil {
			_, apiErr := CreateDashboard(context.Background(), data, fm)
			if apiErr != nil {
				zap.L().Error("Creating Dashboards: Error in file", zap.String("filename", filename), zap.Error(apiErr.Err))
			}
			continue
		}

		apiErr := upsertDashboard(id.(string), data, filename, fm)
		if apiErr != nil {
			zap.L().Error("Creating Dashboards: Error upserting dashboard", zap.String("filename", filename), zap.Error(apiErr.Err))
		}
	}
	return nil
}

func upsertDashboard(uuid string, data map[string]interface{}, filename string, fm interfaces.FeatureLookup) *model.ApiError {
	_, apiErr := GetDashboard(context.Background(), uuid)
	if apiErr == nil {
		zap.S().Infof("Creating Dashboards: Already exists: %s\t%s", filename, "Dashboard already present in database, Updating dashboard")
		_, apiErr := UpdateDashboard(context.Background(), uuid, data, fm)
		return apiErr
	}

	zap.S().Infof("Creating Dashboards: UUID not found: %s\t%s", filename, "Dashboard not present in database, Creating dashboard")
	_, apiErr = CreateDashboard(context.Background(), data, fm)
	return apiErr
}

func LoadDashboardFiles(fm interfaces.FeatureLookup) error {
	dashboardsPath := constants.GetOrDefaultEnv("DASHBOARDS_PATH", "./config/dashboards")
	return readCurrentDir(dashboardsPath, fm)
}