support sqlite

支持sqlite

data/dlg/common.go

@@ -1,14 +1,17 @@
 package dlg
 
 import (
+	"fmt"
 	"os/exec"
 
 	"github.com/jroimartin/gocui"
 	"github.com/xxjwxc/gormt/data/view/model"
 	"github.com/xxjwxc/gormt/data/view/model/genmysql"
+	"github.com/xxjwxc/gormt/data/view/model/gensqlite"
 
 	"github.com/xxjwxc/gormt/data/config"
 
+	"github.com/xxjwxc/public/mylog"
 	"github.com/xxjwxc/public/tools"
 )
 
@@ -72,7 +75,18 @@ func getBool(bstr string) bool {
 }
 
 func generate(g *gocui.Gui, v *gocui.View) {
-	modeldb := genmysql.GetMysqlModel()
+	var modeldb model.IModel
+	switch config.GetDbInfo().Type {
+	case 0:
+		modeldb = genmysql.GetModel()
+	case 1:
+		modeldb = gensqlite.GetModel()
+	}
+	if modeldb == nil {
+		mylog.Error(fmt.Errorf("modeldb not fund : please check db_info.type (0:mysql , 1:sqlite , 2:mssql) "))
+		return
+	}
+
 	pkg := modeldb.GenModel()
 	// just for test
 	// out, _ := json.Marshal(pkg)