forked from pingcap/dm
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
cherry pick pingcap#1305 to release-2.0 (pingcap#1311)
- Loading branch information
Showing
9 changed files
with
6,420 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| FROM busybox | ||
|
|
||
| ADD init.sh /usr/bin/init.sh | ||
| RUN chmod +x /usr/bin/init.sh | ||
|
|
||
| COPY dashboards/*.json /tmp/ | ||
| COPY rules/*.rules.yml /tmp/ | ||
| COPY datasources/*.yaml /tmp/ | ||
|
|
||
| ENTRYPOINT ["/usr/bin/init.sh"] | ||
| CMD ["TIDB-Cluster", "/grafana-dashboard-definitions/tidb/", "false", "/etc/prometheus"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,104 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io/ioutil" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
|
|
||
| "github.com/pingcap/errors" | ||
| "github.com/tidwall/gjson" | ||
| "github.com/tidwall/sjson" | ||
| ) | ||
|
|
||
| const ( | ||
| datasourceName = "dm-cluster" | ||
| outputDir = "./dashboards" | ||
| name = "dm.json" | ||
| ) | ||
|
|
||
| var ( | ||
| dashboards = map[string]string{ | ||
| "dm.json": "Test-Cluster-DM", | ||
| } | ||
| ) | ||
|
|
||
| func readDashboard(dir string, name string) (string, error) { | ||
| file := filepath.Join(dir, name) | ||
| data, err := ioutil.ReadFile(file) | ||
| if err != nil { | ||
| return "", errors.Errorf("File %s do not exists", file) | ||
| } | ||
| return string(data), nil | ||
| } | ||
|
|
||
| func writeDashboard(dir string, name string, body string) error { | ||
| title, exist := dashboards[name] | ||
| if !exist { | ||
| return errors.Errorf("%s dashboard is not found in operator", name) | ||
| } | ||
|
|
||
| writeFile(dir, name, filterDashboard(body, name, title)) | ||
| return nil | ||
| } | ||
|
|
||
| func writeFile(baseDir string, fileName string, body string) { | ||
| if body == "" { | ||
| return | ||
| } | ||
|
|
||
| fn := filepath.Join(baseDir, fileName) | ||
| f, err := os.Create(fn) | ||
| checkErr(err, "create file failed, f="+fn) | ||
| defer f.Close() | ||
|
|
||
| if _, err := f.WriteString(body); err != nil { | ||
| checkErr(err, "write file failed, f="+fn) | ||
| } | ||
| } | ||
|
|
||
| func checkErr(err error, msg string) { | ||
| if err != nil { | ||
| panic(errors.Wrap(err, msg)) | ||
| } | ||
| } | ||
|
|
||
| func filterDashboard(str string, dashboard string, title string) string { | ||
| // replace grafana item | ||
| var err error | ||
| r := gjson.Get(str, "__requires.0.type") | ||
| if r.Exists() && r.Str == "grafana" { | ||
| str, err = sjson.Set(str, "__requires.0.version", "") | ||
| checkErr(err, "update links filed failed") | ||
| } | ||
| // replace links item | ||
| if gjson.Get(str, "links").Exists() { | ||
| str, err = sjson.Set(str, "links", []struct{}{}) | ||
| checkErr(err, "update links failed") | ||
| } | ||
|
|
||
| // replace datasource name | ||
| if gjson.Get(str, "__inputs").Exists() && gjson.Get(str, "__inputs.0.name").Exists() { | ||
| datasource := gjson.Get(str, "__inputs.0.name").Str | ||
| str = strings.ReplaceAll(str, fmt.Sprintf("${%s}", datasource), datasourceName) | ||
| } | ||
|
|
||
| // delete input defination | ||
| if gjson.Get(str, "__inputs").Exists() { | ||
| str, err = sjson.Delete(str, "__inputs") | ||
| checkErr(err, "delete path failed") | ||
| } | ||
|
|
||
| // unify the title name | ||
| str, err = sjson.Set(str, "title", title) | ||
| checkErr(err, "replace title failed") | ||
|
|
||
| return str | ||
| } | ||
|
|
||
| func main() { | ||
| str, err := readDashboard(outputDir, name) | ||
| checkErr(err, "read dashboard file failed") | ||
| checkErr(writeDashboard(outputDir, name, str), "write dashboard file failed") | ||
| } |
Oops, something went wrong.