81 lines
2.6 KiB
Go
81 lines
2.6 KiB
Go
|
package dashboards
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"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 {
|
||
|
|
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)
|
||
|
|
}
|