LBRYCommunity/sqlboiler
1
0
Fork 0
Code Issues Pull requests1 Projects Releases Packages Wiki Activity

Die if postgres driver config is missing

Browse source
This commit is contained in:
Aaron L 2016-07-31 19:04:48 -07:00
parent ba5034b7ee
commit 40df1b7c86
1 changed files with 7 additions and 2 deletions

9
main.go
View file

@ -2,6 +2,7 @@
package main package main
import ( import (
"errors"
"fmt" "fmt"
"os" "os"
"path/filepath" "path/filepath"
@ -73,7 +74,7 @@ func main() {
return return
} }
fmt.Printf("\n%+v\n", err) fmt.Printf("%+v\n", err)
os.Exit(1) os.Exit(1)
} }
} }
@ -91,8 +92,10 @@ func preRun(cmd *cobra.Command, args []string) error {
return commandFailure("must provide a driver name") return commandFailure("must provide a driver name")
} }
driverName := args[0]
cmdConfig = &Config{ cmdConfig = &Config{
DriverName: args[0], DriverName: driverName,
OutFolder: viper.GetString("output"), OutFolder: viper.GetString("output"),
PkgName: viper.GetString("pkgname"), PkgName: viper.GetString("pkgname"),
} }
@ -137,6 +140,8 @@ func preRun(cmd *cobra.Command, args []string) error {
if err != nil { if err != nil {
return commandFailure(err.Error()) return commandFailure(err.Error())
} }
} else if driverName == "postgres" {
return errors.New("postgres driver requires a postgres section in the config")
} }
cmdState, err = New(cmdConfig) cmdState, err = New(cmdConfig)