Generator cleanup.

This commit is contained in:
go-jet 2019-06-23 18:55:57 +02:00
parent ee5d526d26
commit a44bd85d32
22 changed files with 97 additions and 92 deletions

View file

@ -1,63 +0,0 @@
package main
import (
"flag"
"fmt"
"github.com/go-jet/jet/generator"
"os"
)
var (
host string
port string
user string
password string
sslmode string
params string
dbName string
schemaName string
destDir string
)
func init() {
flag.StringVar(&host, "host", "", "Database host path (Example: localhost)")
flag.StringVar(&port, "port", "", "Database port")
flag.StringVar(&user, "user", "", "Database user")
flag.StringVar(&password, "password", "", "The users password")
flag.StringVar(&sslmode, "sslmode", "disable", "Whether or not to use SSL")
flag.StringVar(&params, "params", "", "Additional connection string parameters.")
flag.StringVar(&dbName, "dbname", "", "name of the database")
flag.StringVar(&schemaName, "schema", "public", "Database schema name.")
flag.StringVar(&destDir, "path", "", "Destination dir for generated files.")
flag.Parse()
}
func main() {
genData := generator.GeneratorData{
Host: host,
Port: port,
User: user,
Password: password,
SslMode: sslmode,
Params: params,
DBName: dbName,
SchemaName: schemaName,
}
fmt.Println(destDir, genData)
err := generator.Generate(destDir, genData)
if err != nil {
fmt.Println(err.Error())
os.Exit(-1)
}
fmt.Println("SUCCESS")
}