pkg/cluster/spec: add supported configuration for grafana (#1566)

pingcap · Sep 27, 2021 · a9a260a · a9a260a
1 parent 7ad8f97
commit a9a260a
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/embed/templates/config/grafana.ini.tpl b/embed/templates/config/grafana.ini.tpl
@@ -169,7 +169,11 @@ admin_password = {{.Password}}
 ;login_hint = email or username
 
 # Default UI theme ("dark" or "light")
+{{- if .DefaultTheme}}
+default_theme = {{.DefaultTheme}}
+{{- else}}
 ;default_theme = dark
+{{- end}}
 
 ############### Set Cookie Name for Multiple Instances #######################
 [auth]
@@ -182,10 +186,18 @@ enabled = true
 {{- end}}
 
 # specify organization name that should be used for unauthenticated users
+{{- if .OrgName}}
+org_name = {{.OrgName}}
+{{- else}}
 ;org_name = Main Org.
+{{- end}}
 
 # specify role for unauthenticated users
+{{- if .OrgRole}}
+org_role = {{.OrgRole}}
+{{- else}}
 ;org_role = Viewer
+{{- end}}
 
 #################################### Basic Auth ##########################
 [auth.basic]

diff --git a/pkg/cluster/spec/grafana.go b/pkg/cluster/spec/grafana.go
@@ -47,6 +47,9 @@ type GrafanaSpec struct {
 	AnonymousEnable bool                 `yaml:"anonymous_enable" default:"false" validate:"anonymous_enable:editable"`
 	RootURL         string               `yaml:"root_url" validate:"root_url:editable"`
 	Domain          string               `yaml:"domain" validate:"domain:editable"`
+	DefaultTheme    string               `yaml:"default_theme,omitempty" validate:"default_theme:editable"`
+	OrgName         string               `yaml:"org_name,omitempty" validate:"org_name:editable"`
+	OrgRole         string               `yaml:"org_role,omitempty" validate:"org_role:editable"`
 }
 
 // Role returns the component role of the instance
@@ -166,6 +169,9 @@ func (i *GrafanaInstance) InitConfig(
 		WithAnonymousenable(spec.AnonymousEnable).
 		WithRootURL(spec.RootURL).
 		WithDomain(spec.Domain).
+		WithDefaultTheme(spec.DefaultTheme).
+		WithOrgName(spec.OrgName).
+		WithOrgRole(spec.OrgRole).
 		ConfigToFile(fp); err != nil {
 		return err
 	}

diff --git a/pkg/cluster/template/config/grafana.go b/pkg/cluster/template/config/grafana.go
@@ -32,6 +32,9 @@ type GrafanaConfig struct {
 	AnonymousEnable bool   // anonymous enable
 	RootURL         string // root_url
 	Domain          string // domain
+	DefaultTheme    string // default_theme
+	OrgName         string // org_name
+	OrgRole         string // org_role
 }
 
 // NewGrafanaConfig returns a GrafanaConfig
@@ -79,6 +82,24 @@ func (c *GrafanaConfig) WithDomain(domain string) *GrafanaConfig {
 	return c
 }
 
+// WithDefaultTheme sets defaultTheme of default theme
+func (c *GrafanaConfig) WithDefaultTheme(defaultTheme string) *GrafanaConfig {
+	c.DefaultTheme = defaultTheme
+	return c
+}
+
+// WithOrgName sets orgName of org name
+func (c *GrafanaConfig) WithOrgName(orgName string) *GrafanaConfig {
+	c.OrgName = orgName
+	return c
+}
+
+// WithOrgRole sets orgName of org role
+func (c *GrafanaConfig) WithOrgRole(orgRole string) *GrafanaConfig {
+	c.OrgRole = orgRole
+	return c
+}
+
 // Config generate the config file data.
 func (c *GrafanaConfig) Config() ([]byte, error) {
 	fp := path.Join("templates", "config", "grafana.ini.tpl")