mirror of
1
Fork 0

Fix incorrect PostgreSQL connection string for Unix sockets (#28865)

Fix #28864
This commit is contained in:
sdvcrx 2024-01-21 00:04:47 +08:00 committed by GitHub
parent 6c771a311b
commit 80d48621cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 2 deletions

View File

@ -169,8 +169,8 @@ func getPostgreSQLConnectionString(dbHost, dbUser, dbPasswd, dbName, dbsslMode s
RawQuery: dbParam, RawQuery: dbParam,
} }
query := connURL.Query() query := connURL.Query()
if strings.HasPrefix(dbHost, "/") { // looks like a unix socket if strings.HasPrefix(host, "/") { // looks like a unix socket
query.Add("host", dbHost) query.Add("host", host)
connURL.Host = ":" + port connURL.Host = ":" + port
} }
query.Set("sslmode", dbsslMode) query.Set("sslmode", dbsslMode)

View File

@ -77,6 +77,14 @@ func Test_getPostgreSQLConnectionString(t *testing.T) {
SSLMode: "false", SSLMode: "false",
Output: "postgres://testuser:space%20space%20%21%23$%25%5E%5E%25%5E%60%60%60-=%3F=@:5432/gitea?host=%2Ftmp%2Fpg.sock&sslmode=false", Output: "postgres://testuser:space%20space%20%21%23$%25%5E%5E%25%5E%60%60%60-=%3F=@:5432/gitea?host=%2Ftmp%2Fpg.sock&sslmode=false",
}, },
{
Host: "/tmp/pg.sock:6432",
User: "testuser",
Passwd: "pass",
Name: "gitea",
SSLMode: "false",
Output: "postgres://testuser:pass@:6432/gitea?host=%2Ftmp%2Fpg.sock&sslmode=false",
},
{ {
Host: "localhost", Host: "localhost",
User: "pgsqlusername", User: "pgsqlusername",