RHINENG-22333: allow app to connect as admin

Author: MichaelMraka

base/database/setup.go

@@ -19,8 +19,8 @@ var (
 	globalPgConfig     *PostgreSQLConfig
 )
 
-func InitDB() {
-	pgConfig := loadEnvPostgreSQLConfig(false)
+func InitDB(useAdmin bool) {
+	pgConfig := loadEnvPostgreSQLConfig(useAdmin, false)
 	if DB != nil && pgConfig == globalPgConfig {
 		// reuse connection
 		check(DB)
@@ -30,18 +30,22 @@ func InitDB() {
 	DB = openPostgreSQL(pgConfig)
 	check(DB)
 	if utils.CoreCfg.DBReadReplicaEnabled {
-		pgConfig := loadEnvPostgreSQLConfig(ReadReplicaConfigured())
+		pgConfig := loadEnvPostgreSQLConfig(useAdmin, ReadReplicaConfigured())
 		DBReadReplica = openPostgreSQL(pgConfig)
 		check(DBReadReplica)
 	}
 }
 
 // Configure Configure database, PostgreSQL or SQLite connection
-func Configure() {
-	InitDB()
+func ConfigureAdmin(useAdmin bool) {
+	InitDB(useAdmin)
 	loadAdditionalParamsFromDB()
 }
 
+func Configure() {
+	ConfigureAdmin(false)
+}
+
 // PostgreSQLConfig PostgreSQL database config
 type PostgreSQLConfig struct {
 	Host        string
@@ -104,19 +108,25 @@ func check(db *gorm.DB) {
 }
 
 // load database config from environment vars using inserted prefix
-func loadEnvPostgreSQLConfig(useReadReplica bool) *PostgreSQLConfig {
+func loadEnvPostgreSQLConfig(useAdmin bool, useReadReplica bool) *PostgreSQLConfig {
+	user := utils.CoreCfg.DBUser
+	passwd := utils.CoreCfg.DBPassword
+	if useAdmin {
+		user = utils.CoreCfg.DBAdminUser
+		passwd = utils.CoreCfg.DBAdminPassword
+	}
 	host := utils.CoreCfg.DBHost
 	port := utils.CoreCfg.DBPort
 	if useReadReplica {
 		host = utils.CoreCfg.DBReadReplicaHost
 		port = utils.CoreCfg.DBReadReplicaPort
 	}
 	config := PostgreSQLConfig{
-		User:                   utils.CoreCfg.DBUser,
+		User:                   user,
 		Host:                   host,
 		Port:                   port,
 		Database:               utils.CoreCfg.DBName,
-		Passwd:                 utils.CoreCfg.DBPassword,
+		Passwd:                 passwd,
 		SSLMode:                utils.CoreCfg.DBSslMode,
 		SSLRootCert:            utils.CoreCfg.DBSslRootCert,
 		Debug:                  utils.CoreCfg.DBDebug,