refine: distro type (#968)

pingcap · Jul 23, 2021 · 58aa412 · 58aa412
1 parent 384cc68
commit 58aa412
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/utils/distro/default_distro_info.go b/pkg/utils/distro/default_distro_info.go
@@ -2,7 +2,7 @@
 
 package distro
 
-var Resource = map[string]interface{}{
+var Resource = map[string]string{
 	"tidb":    "TiDB",
 	"tikv":    "TiKV",
 	"tiflash": "TiFlash",

diff --git a/pkg/utils/distro/distro.go b/pkg/utils/distro/distro.go
@@ -18,7 +18,7 @@ import (
 	"sync/atomic"
 )
 
-type introData map[string]interface{}
+type introData map[string]string
 
 var data atomic.Value
 
@@ -28,8 +28,8 @@ func Replace(distro introData) {
 
 func Data(k string) string {
 	d := data.Load().(introData)
-	if d[k] == nil {
+	if d[k] == "" {
 		return k
 	}
-	return d[k].(string)
+	return d[k]
 }