Export: include alerts, thumbnails, usage stats, and short urls (#51938)

2022-07-11 16:25:40 -07:00
parent 0e066dd5f8
commit 3003a48dc6
15 changed files with 454 additions and 211 deletions
@@ -0,0 +1,85 @@
+package export
+
+import (
+	"path"
+	"strconv"
+	"strings"
+
+	"github.com/grafana/grafana-plugin-sdk-go/data/sqlutil"
+	"github.com/grafana/grafana/pkg/services/sqlstore"
+)
+
+func exportUsage(helper *commitHelper, job *gitExportJob) error {
+	return job.sql.WithDbSession(helper.ctx, func(sess *sqlstore.DBSession) error {
+		commit := commitOptions{
+			comment: "usage stats",
+		}
+
+		type statsTables struct {
+			table      string
+			sql        string
+			converters []sqlutil.Converter
+		}
+
+		dump := []statsTables{
+			{
+				table: "data_source_usage_by_day",
+				sql: `SELECT day,uid,queries,errors,load_duration_ms 
+					FROM data_source_usage_by_day 
+					JOIN data_source ON data_source.id = data_source_usage_by_day.data_source_id
+					WHERE org_id =` + strconv.FormatInt(helper.orgID, 10),
+				converters: []sqlutil.Converter{{Dynamic: true}},
+			},
+			{
+				table: "dashboard_usage_by_day",
+				sql: `SELECT uid,day,views,queries,errors,load_duration 
+					FROM dashboard_usage_by_day
+					JOIN dashboard ON dashboard_usage_by_day.dashboard_id = dashboard.id
+					WHERE org_id =` + strconv.FormatInt(helper.orgID, 10),
+				converters: []sqlutil.Converter{{Dynamic: true}},
+			},
+			{
+				table: "dashboard_usage_sums",
+				sql: `SELECT uid,
+					views_last_1_days,
+					views_last_7_days,
+					views_last_30_days,
+					views_total,
+					queries_last_1_days,
+					queries_last_7_days,
+					queries_last_30_days,
+					queries_total,
+					errors_last_1_days,
+					errors_last_7_days,
+					errors_last_30_days,
+					errors_total
+					FROM dashboard_usage_sums
+					JOIN dashboard ON dashboard_usage_sums.dashboard_id = dashboard.id
+					WHERE org_id =` + strconv.FormatInt(helper.orgID, 10),
+				converters: []sqlutil.Converter{{Dynamic: true}},
+			},
+		}
+
+		for _, usage := range dump {
+			rows, err := sess.DB().QueryContext(helper.ctx, usage.sql)
+			if err != nil {
+				if strings.HasPrefix(err.Error(), "no such table") {
+					continue
+				}
+				return err
+			}
+
+			frame, err := sqlutil.FrameFromRows(rows.Rows, -1, usage.converters...)
+			if err != nil {
+				return err
+			}
+			frame.Name = usage.table
+			commit.body = append(commit.body, commitBody{
+				fpath: path.Join(helper.orgDir, "usage", usage.table+".json"),
+				frame: frame,
+			})
+		}
+
+		return helper.add(commit)
+	})
+}