datasource as cfg: basic implementation

2026-08-13 06:34:55 -05:00 · 2017-11-15 14:30:32 +01:00
parent 014f6d5a75
commit 39b6c04643
21 changed files with 9790 additions and 6 deletions
@@ -0,0 +1,114 @@
+package datasources
+
+import (
+	"io"
+	"io/ioutil"
+	"path/filepath"
+
+	"github.com/grafana/grafana/pkg/bus"
+	"github.com/grafana/grafana/pkg/log"
+
+	"github.com/grafana/grafana/pkg/models"
+	yaml "gopkg.in/yaml.v2"
+)
+
+var (
+	logger log.Logger
+)
+
+// TODO: secure jsonData
+// TODO: auto reload on file changes
+
+func Init(configPath string) (error, io.Closer) {
+	logger = log.New("setting.datasource")
+
+	datasources, err := readDatasources(configPath)
+	if err != nil {
+		return err, ioutil.NopCloser(nil)
+	}
+
+	for _, ds := range datasources {
+		query := &models.GetDataSourceByNameQuery{Name: ds.Name}
+		err := bus.Dispatch(query)
+		if err != nil && err != models.ErrDataSourceNotFound {
+			return err, ioutil.NopCloser(nil)
+		}
+
+		if ds.OrgId == 0 {
+			ds.OrgId = 1
+		}
+
+		if query.Result == nil {
+			logger.Info("inserting ", "name", ds.Name)
+			insertCmd := insertCommand(ds)
+			if err := bus.Dispatch(insertCmd); err != nil {
+				return err, ioutil.NopCloser(nil)
+			}
+		} else {
+			logger.Info("updating", "name", ds.Name)
+			updateCmd := updateCommand(ds, query.Result.Id)
+			if err := bus.Dispatch(updateCmd); err != nil {
+				return err, ioutil.NopCloser(nil)
+			}
+		}
+	}
+
+	return nil, ioutil.NopCloser(nil)
+}
+
+func readDatasources(path string) ([]models.DataSource, error) {
+	filename, _ := filepath.Abs(path)
+	yamlFile, err := ioutil.ReadFile(filename)
+
+	if err != nil {
+		return nil, err
+	}
+
+	var datasources []models.DataSource
+
+	err = yaml.Unmarshal(yamlFile, &datasources)
+	if err != nil {
+		return nil, err
+	}
+
+	return datasources, nil
+}
+
+func insertCommand(ds models.DataSource) *models.AddDataSourceCommand {
+	return &models.AddDataSourceCommand{
+		OrgId:             ds.OrgId,
+		Name:              ds.Name,
+		Type:              ds.Type,
+		Access:            ds.Access,
+		Url:               ds.Url,
+		Password:          ds.Password,
+		User:              ds.User,
+		Database:          ds.Database,
+		BasicAuth:         ds.BasicAuth,
+		BasicAuthUser:     ds.BasicAuthUser,
+		BasicAuthPassword: ds.BasicAuthPassword,
+		WithCredentials:   ds.WithCredentials,
+		IsDefault:         ds.IsDefault,
+		JsonData:          ds.JsonData,
+	}
+}
+
+func updateCommand(ds models.DataSource, id int64) *models.UpdateDataSourceCommand {
+	return &models.UpdateDataSourceCommand{
+		Id:                id,
+		OrgId:             ds.OrgId,
+		Name:              ds.Name,
+		Type:              ds.Type,
+		Access:            ds.Access,
+		Url:               ds.Url,
+		Password:          ds.Password,
+		User:              ds.User,
+		Database:          ds.Database,
+		BasicAuth:         ds.BasicAuth,
+		BasicAuthUser:     ds.BasicAuthUser,
+		BasicAuthPassword: ds.BasicAuthPassword,
+		WithCredentials:   ds.WithCredentials,
+		IsDefault:         ds.IsDefault,
+		JsonData:          ds.JsonData,
+	}
+}