修正参数错误

Author: xfali

cmd/gobatis-cmd/main.go

@@ -42,12 +42,6 @@ func main() {
     register := flag.Bool("register", false, "add register code")
     flag.Parse()
 
-    dbDriver := db.GetDriver(*driver)
-    if dbDriver == nil {
-        log.Print("not support driver: ", *driver)
-        os.Exit(-1)
-    }
-
     conf := config.FileConfig{}
     if *confFile != "" {
         err := loadFromFile(&conf, *confFile)
@@ -75,6 +69,12 @@ func main() {
         conf.Password = *pw
     }
 
+    dbDriver := db.GetDriver(conf.Driver)
+    if dbDriver == nil {
+        log.Print("not support driver: ", conf.Driver)
+        os.Exit(-1)
+    }
+
     err := dbDriver.Open(conf.Driver, db.GenDBInfo(conf.Driver, conf.DBName, conf.User, conf.Password, conf.Host, conf.Port))
     if err != nil {
         log.Print(err)

pkg/db/db.go

@@ -29,7 +29,7 @@ type PostgresDB struct {
 var buildinDrivers = map[string]common.DBDriver{
 	"mysql":    &MysqlDB{},
 	"postgres": &PostgresDB{},
-	"sqlite":   &SqliteDB{},
+	"sqlite3":  &SqliteDB{},
 }
 
 func GetDriver(name string) common.DBDriver {