conn: use GetDomain to avoid some TiDB breaking changes

Signed-off-by: Neil Shen <[email protected]>
pingcap · Mar 11, 2020 · 63c41a4 · 63c41a4
1 parent 237fe5b
commit 63c41a4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/pkg/conn/conn.go b/pkg/conn/conn.go
@@ -159,7 +159,7 @@ func NewMgr(
 		return nil, errors.Errorf("tikv cluster not health %+v", stores)
 	}
 
-	dom, err := g.BootstrapSession(storage)
+	dom, err := g.GetDomain(storage)
 	if err != nil {
 		return nil, errors.Trace(err)
 	}

diff --git a/pkg/glue/glue.go b/pkg/glue/glue.go
@@ -12,7 +12,7 @@ import (
 
 // Glue is an abstraction of TiDB function calls used in BR.
 type Glue interface {
-	BootstrapSession(store kv.Storage) (*domain.Domain, error)
+	GetDomain(store kv.Storage) (*domain.Domain, error)
 	CreateSession(store kv.Storage) (Session, error)
 	Open(path string, option pd.SecurityOption) (kv.Storage, error)
 }

diff --git a/pkg/gluetidb/glue.go b/pkg/gluetidb/glue.go
@@ -24,9 +24,9 @@ type tidbSession struct {
 	se session.Session
 }
 
-// BootstrapSession implements glue.Glue
-func (Glue) BootstrapSession(store kv.Storage) (*domain.Domain, error) {
-	return session.BootstrapSession(store)
+// GetDomain implements glue.Glue
+func (Glue) GetDomain(store kv.Storage) (*domain.Domain, error) {
+	return session.GetDomain(store)
 }
 
 // CreateSession implements glue.Glue

diff --git a/pkg/mock/mock_cluster.go b/pkg/mock/mock_cluster.go
@@ -68,7 +68,7 @@ func NewCluster() (*Cluster, error) {
 	}
 	session.SetSchemaLease(0)
 	session.DisableStats4Test()
-	dom, err := session.BootstrapSession(storage)
+	dom, err := session.GetDomain(storage)
 	if err != nil {
 		return nil, err
 	}