arno/kube-bench
1
0
Fork 0
mirror of https://github.com/aquasecurity/kube-bench.git synced 2025-06-26 01:42:37 +00:00
Code Issues Releases Wiki Activity

chore: move db var to savePgsql function

Browse Source
This commit is contained in:
luisserra 2022-03-28 21:57:06 +01:00
parent dcb167c10d
commit f76b70af81
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cmd/database.go
View File

@ -20,6 +20,7 @@ type PsqlConnInfo struct {
Password string Password string
Schema string Schema string
} }
func getPsqlConnInfo() (PsqlConnInfo, error) { func getPsqlConnInfo() (PsqlConnInfo, error) {
var host string var host string
if value := viper.GetString("PGSQL_HOST"); value != "" { if value := viper.GetString("PGSQL_HOST"); value != "" {
@ -107,6 +108,8 @@ func savePgsql(jsonInfo string) {
exitWithError(err) exitWithError(err)
} }
var db *gorm.DB
if PsqlConnInfo.Schema == "" { if PsqlConnInfo.Schema == "" {
db, err = gorm.Open(postgres.Open(PsqlConnInfo.toString()), &gorm.Config{}) db, err = gorm.Open(postgres.Open(PsqlConnInfo.toString()), &gorm.Config{})
} else { } else {