Chore: SQL Store Split of datasource (#54262)

* refectory datasource * fix linter
2022-08-26 11:03:38 -04:00
parent ec3697584a
commit 6227528ea0
8 changed files with 920 additions and 55 deletions
@@ -0,0 +1,114 @@
+package service
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/grafana/grafana/pkg/infra/localcache"
+	"github.com/grafana/grafana/pkg/infra/log"
+	"github.com/grafana/grafana/pkg/services/datasources"
+	"github.com/grafana/grafana/pkg/services/sqlstore"
+	"github.com/grafana/grafana/pkg/services/user"
+)
+
+const (
+	DefaultCacheTTL = 5 * time.Second
+)
+
+func ProvideCacheService(cacheService *localcache.CacheService, sqlStore *sqlstore.SQLStore) *CacheServiceImpl {
+	return &CacheServiceImpl{
+		logger:       log.New("datasources"),
+		cacheTTL:     DefaultCacheTTL,
+		CacheService: cacheService,
+		SQLStore:     sqlStore,
+	}
+}
+
+type CacheServiceImpl struct {
+	logger       log.Logger
+	cacheTTL     time.Duration
+	CacheService *localcache.CacheService
+	SQLStore     *sqlstore.SQLStore
+}
+
+func (dc *CacheServiceImpl) GetDatasource(
+	ctx context.Context,
+	datasourceID int64,
+	user *user.SignedInUser,
+	skipCache bool,
+) (*datasources.DataSource, error) {
+	cacheKey := idKey(datasourceID)
+
+	if !skipCache {
+		if cached, found := dc.CacheService.Get(cacheKey); found {
+			ds := cached.(*datasources.DataSource)
+			if ds.OrgId == user.OrgID {
+				return ds, nil
+			}
+		}
+	}
+
+	dc.logger.Debug("Querying for data source via SQL store", "id", datasourceID, "orgId", user.OrgID)
+
+	query := &datasources.GetDataSourceQuery{Id: datasourceID, OrgId: user.OrgID}
+	ss := SqlStore{db: dc.SQLStore}
+	err := ss.GetDataSource(ctx, query)
+	if err != nil {
+		return nil, err
+	}
+
+	ds := query.Result
+
+	if ds.Uid != "" {
+		dc.CacheService.Set(uidKey(ds.OrgId, ds.Uid), ds, time.Second*5)
+	}
+	dc.CacheService.Set(cacheKey, ds, dc.cacheTTL)
+	return ds, nil
+}
+
+func (dc *CacheServiceImpl) GetDatasourceByUID(
+	ctx context.Context,
+	datasourceUID string,
+	user *user.SignedInUser,
+	skipCache bool,
+) (*datasources.DataSource, error) {
+	if datasourceUID == "" {
+		return nil, fmt.Errorf("can not get data source by uid, uid is empty")
+	}
+	if user.OrgID == 0 {
+		return nil, fmt.Errorf("can not get data source by uid, orgId is missing")
+	}
+	uidCacheKey := uidKey(user.OrgID, datasourceUID)
+
+	if !skipCache {
+		if cached, found := dc.CacheService.Get(uidCacheKey); found {
+			ds := cached.(*datasources.DataSource)
+			if ds.OrgId == user.OrgID {
+				return ds, nil
+			}
+		}
+	}
+
+	dc.logger.Debug("Querying for data source via SQL store", "uid", datasourceUID, "orgId", user.OrgID)
+	query := &datasources.GetDataSourceQuery{Uid: datasourceUID, OrgId: user.OrgID}
+	ss := SqlStore{db: dc.SQLStore}
+	err := ss.GetDataSource(ctx, query)
+	if err != nil {
+		return nil, err
+	}
+
+	ds := query.Result
+
+	dc.CacheService.Set(uidCacheKey, ds, dc.cacheTTL)
+	dc.CacheService.Set(idKey(ds.Id), ds, dc.cacheTTL)
+	return ds, nil
+}
+
+func idKey(id int64) string {
+	return fmt.Sprintf("ds-%d", id)
+}
+
+func uidKey(orgID int64, uid string) string {
+	return fmt.Sprintf("ds-orgid-uid-%d-%s", orgID, uid)
+}