diff --git a/sdk/database/helper/connutil/sql.go b/sdk/database/helper/connutil/sql.go index 08f610ac7c..1dcf4eb294 100644 --- a/sdk/database/helper/connutil/sql.go +++ b/sdk/database/helper/connutil/sql.go @@ -128,13 +128,19 @@ func (c *SQLConnectionProducer) Connection(ctx context.Context) (interface{}, er // Otherwise, attempt to make connection conn := c.ConnectionURL - // Ensure timezone is set to UTC for all the connections + // PostgreSQL specific settings if strings.HasPrefix(conn, "postgres://") || strings.HasPrefix(conn, "postgresql://") { + // Ensure timezone is set to UTC for all the connections if strings.Contains(conn, "?") { conn += "&timezone=UTC" } else { conn += "?timezone=UTC" } + + // Ensure a reasonable application_name is set + if !strings.Contains(conn, "application_name") { + conn += "&application_name=vault" + } } var err error diff --git a/vendor/github.com/hashicorp/vault/sdk/database/helper/connutil/sql.go b/vendor/github.com/hashicorp/vault/sdk/database/helper/connutil/sql.go index 08f610ac7c..1dcf4eb294 100644 --- a/vendor/github.com/hashicorp/vault/sdk/database/helper/connutil/sql.go +++ b/vendor/github.com/hashicorp/vault/sdk/database/helper/connutil/sql.go @@ -128,13 +128,19 @@ func (c *SQLConnectionProducer) Connection(ctx context.Context) (interface{}, er // Otherwise, attempt to make connection conn := c.ConnectionURL - // Ensure timezone is set to UTC for all the connections + // PostgreSQL specific settings if strings.HasPrefix(conn, "postgres://") || strings.HasPrefix(conn, "postgresql://") { + // Ensure timezone is set to UTC for all the connections if strings.Contains(conn, "?") { conn += "&timezone=UTC" } else { conn += "?timezone=UTC" } + + // Ensure a reasonable application_name is set + if !strings.Contains(conn, "application_name") { + conn += "&application_name=vault" + } } var err error