backend: import planner/core package to initialize expression.Rewrite…

…AstExpr (pingcap#526)
YuJuncen · Dec 20, 2020 · ff241c3 · ff241c3
1 parent 56bc32d
commit ff241c3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lightning/backend/sql2kv.go b/lightning/backend/sql2kv.go
@@ -31,6 +31,9 @@ import (
 	"go.uber.org/zap"
 	"go.uber.org/zap/zapcore"
 
+	// Import tidb/planner/core to initialize expression.RewriteAstExpr
+	_ "github.com/pingcap/tidb/planner/core"
+
 	"github.com/pingcap/tidb-lightning/lightning/common"
 	"github.com/pingcap/tidb-lightning/lightning/log"
 	"github.com/pingcap/tidb-lightning/lightning/metric"

diff --git a/lightning/restore/restore.go b/lightning/restore/restore.go
@@ -49,9 +49,6 @@ import (
 	verify "github.com/pingcap/tidb-lightning/lightning/verification"
 	"github.com/pingcap/tidb-lightning/lightning/web"
 	"github.com/pingcap/tidb-lightning/lightning/worker"
-
-	// TODO: remove this after https://github.com/pingcap/tidb/issues/21342 is fixed.
-	_ "github.com/pingcap/tidb/planner/core"
 )
 
 const (