@@ -7,6 +7,7 @@ package main
77import (
88 "fmt"
99 "github.com/AlecAivazis/survey/v2"
10+ "github.com/AlecAivazis/survey/v2/core"
1011 "github.com/chenhg5/go-admin/modules/config"
1112 "github.com/chenhg5/go-admin/modules/db"
1213 "github.com/chenhg5/go-admin/modules/system"
@@ -113,15 +114,15 @@ func generating() {
113114
114115 err := survey .Ask (qs , & result )
115116 checkError (err )
116- driver := result ["driver" ].(string )
117+ driver := result ["driver" ].(core. OptionAnswer )
117118
118119 var (
119120 cfg map [string ]config.Database
120121 name string
121- conn = db .GetConnectionByDriver (driver )
122+ conn = db .GetConnectionByDriver (driver . Value )
122123 )
123124
124- if driver != "sqlite" {
125+ if driver . Value != "sqlite" {
125126 host := promptWithDefault ("sql address" , "127.0.0.1" )
126127 port := promptWithDefault ("sql port" , "3306" )
127128 user := promptWithDefault ("sql username" , "root" )
@@ -142,7 +143,7 @@ func generating() {
142143 Name : name ,
143144 MaxIdleCon : 50 ,
144145 MaxOpenCon : 150 ,
145- Driver : driver ,
146+ Driver : driver . Value ,
146147 File : "" ,
147148 },
148149 }
@@ -157,7 +158,7 @@ func generating() {
157158 }
158159 cfg = map [string ]config.Database {
159160 "default" : {
160- Driver : driver ,
161+ Driver : driver . Value ,
161162 File : file ,
162163 },
163164 }
@@ -169,7 +170,7 @@ func generating() {
169170 // step 2. show tables
170171 tableModels , _ := db .WithDriver (conn .GetName ()).ShowTables ()
171172
172- tables := getTablesFromSqlResult (tableModels , driver , name )
173+ tables := getTablesFromSqlResult (tableModels , driver . Value , name )
173174 if len (tables ) == 0 {
174175 exitWithError (`no tables, you should build a table of your own business first.
175176
@@ -191,7 +192,7 @@ see: http://www.go-admin.cn/en/docs/#/plugins/admin`)
191192
192193 fieldField := "Field"
193194 typeField := "Type"
194- if driver == "postgresql" {
195+ if driver . Value == "postgresql" {
195196 fieldField = "column_name"
196197 typeField = "udt_name"
197198 }
@@ -200,7 +201,7 @@ see: http://www.go-admin.cn/en/docs/#/plugins/admin`)
200201 for i := 0 ; i < len (chooseTables ); i ++ {
201202 _ = bar .Add (1 )
202203 time .Sleep (10 * time .Millisecond )
203- generateFile (chooseTables [i ], conn , fieldField , typeField , packageName , driver , outputPath )
204+ generateFile (chooseTables [i ], conn , fieldField , typeField , packageName , driver . Value , outputPath )
204205 }
205206 generateTables (outputPath , chooseTables , packageName )
206207
0 commit comments