You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.9 KiB
59 lines
1.9 KiB
package pgconn |
|
|
|
import ( |
|
"os" |
|
"os/user" |
|
"path/filepath" |
|
"strings" |
|
) |
|
|
|
func defaultSettings() map[string]string { |
|
settings := make(map[string]string) |
|
|
|
settings["host"] = defaultHost() |
|
settings["port"] = "5432" |
|
|
|
// Default to the OS user name. Purposely ignoring err getting user name from |
|
// OS. The client application will simply have to specify the user in that |
|
// case (which they typically will be doing anyway). |
|
user, err := user.Current() |
|
appData := os.Getenv("APPDATA") |
|
if err == nil { |
|
// Windows gives us the username here as `DOMAIN\user` or `LOCALPCNAME\user`, |
|
// but the libpq default is just the `user` portion, so we strip off the first part. |
|
username := user.Username |
|
if strings.Contains(username, "\\") { |
|
username = username[strings.LastIndex(username, "\\")+1:] |
|
} |
|
|
|
settings["user"] = username |
|
settings["passfile"] = filepath.Join(appData, "postgresql", "pgpass.conf") |
|
settings["servicefile"] = filepath.Join(user.HomeDir, ".pg_service.conf") |
|
sslcert := filepath.Join(appData, "postgresql", "postgresql.crt") |
|
sslkey := filepath.Join(appData, "postgresql", "postgresql.key") |
|
if _, err := os.Stat(sslcert); err == nil { |
|
if _, err := os.Stat(sslkey); err == nil { |
|
// Both the cert and key must be present to use them, or do not use either |
|
settings["sslcert"] = sslcert |
|
settings["sslkey"] = sslkey |
|
} |
|
} |
|
sslrootcert := filepath.Join(appData, "postgresql", "root.crt") |
|
if _, err := os.Stat(sslrootcert); err == nil { |
|
settings["sslrootcert"] = sslrootcert |
|
} |
|
} |
|
|
|
settings["target_session_attrs"] = "any" |
|
|
|
settings["min_read_buffer_size"] = "8192" |
|
|
|
return settings |
|
} |
|
|
|
// defaultHost attempts to mimic libpq's default host. libpq uses the default unix socket location on *nix and localhost |
|
// on Windows. The default socket location is compiled into libpq. Since pgx does not have access to that default it |
|
// checks the existence of common locations. |
|
func defaultHost() string { |
|
return "localhost" |
|
}
|
|
|