fix: 修复 PostgreSQL 创建数据库未修改所有者

tnb-labs · Sep 26, 2023 · 084f0a0 · 084f0a0
1 parent c08fd0a
commit 084f0a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/app/http/controllers/plugins/postgresql15/postgresql15_controller.go b/app/http/controllers/plugins/postgresql15/postgresql15_controller.go
@@ -326,6 +326,7 @@ func (c *Postgresql15Controller) AddDatabase(ctx http.Context) http.Response {
 
 	tools.Exec(`echo "CREATE DATABASE ` + database + `;" | su - postgres -c "psql"`)
 	tools.Exec(`echo "CREATE USER ` + user + ` WITH PASSWORD '` + password + `';" | su - postgres -c "psql"`)
+	tools.Exec(`echo "ALTER DATABASE ` + database + ` OWNER TO ` + user + `;" | su - postgres -c "psql"`)
 	tools.Exec(`echo "GRANT ALL PRIVILEGES ON DATABASE ` + database + ` TO ` + user + `;" | su - postgres -c "psql"`)
 
 	userConfig := "host    " + database + "    " + user + "    127.0.0.1/32    scram-sha-256"

diff --git a/app/http/controllers/plugins/postgresql16/postgresql16_controller.go b/app/http/controllers/plugins/postgresql16/postgresql16_controller.go
@@ -326,6 +326,7 @@ func (c *Postgresql16Controller) AddDatabase(ctx http.Context) http.Response {
 
 	tools.Exec(`echo "CREATE DATABASE ` + database + `;" | su - postgres -c "psql"`)
 	tools.Exec(`echo "CREATE USER ` + user + ` WITH PASSWORD '` + password + `';" | su - postgres -c "psql"`)
+	tools.Exec(`echo "ALTER DATABASE ` + database + ` OWNER TO ` + user + `;" | su - postgres -c "psql"`)
 	tools.Exec(`echo "GRANT ALL PRIVILEGES ON DATABASE ` + database + ` TO ` + user + `;" | su - postgres -c "psql"`)
 
 	userConfig := "host    " + database + "    " + user + "    127.0.0.1/32    scram-sha-256"