backend/postgres: Allow to pass application_name in the DSN

upfluence · Jan 9, 2024 · 7878a8d · 7878a8d
1 parent 52d9b8c
commit 7878a8d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/backend/postgres/config.go b/backend/postgres/config.go
@@ -36,6 +36,8 @@ type Config struct {
 	User     string
 	Password string
 
+	ApplicationName string
+
 	SSLMode SSLMode
 	SSLSNI  bool
 
@@ -151,6 +153,10 @@ func (c *Config) DSN() (string, error) {
 		return "", err
 	}
 
+	if c.ApplicationName != "" {
+		q.Add("application_name", c.ApplicationName)
+	}
+
 	u := url.URL{
 		Scheme:   "postgres",
 		User:     c.userInfo(),

diff --git a/backend/postgres/config_test.go b/backend/postgres/config_test.go
@@ -16,8 +16,8 @@ func TestDSN(t *testing.T) {
 			dsn: "postgres://localhost:5432/foobar?sslmode=verify-full&sslsni=1",
 		},
 		{
-			c:   &Config{DBName: "foobar", CACertFile: "foobar"},
-			dsn: "postgres://localhost:5432/foobar?sslmode=verify-ca&sslrootcert=foobar&sslsni=0",
+			c:   &Config{DBName: "foobar", CACertFile: "foobar", ApplicationName: "buz"},
+			dsn: "postgres://localhost:5432/foobar?application_name=buz&sslmode=verify-ca&sslrootcert=foobar&sslsni=0",
 		},
 	} {
 		dsn, err := tt.c.DSN()