itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2026-03-13 17:27:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

postgres: Add support for non-default socket paths

Browse Source
This commit is contained in:
Felix Wiedemann 2020-04-14 20:51:04 +02:00 committed by Ryan Leckey
parent 5628658d3f
commit fc78f15ebf
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
sqlx-core/src/postgres/stream.rs
View File

@ -26,9 +26,20 @@ impl PgStream {
let host = url.host();
let port = url.port(5432);
#[cfg(unix)]
let stream = match host {
Some(host) => MaybeTlsStream::connect(host, port).await?,
None => MaybeTlsStream::connect_uds(format!("/var/run/postgresql/.s.PGSQL.{}", port)).await?,
let stream = {
let host = host
.map(|host| {
percent_encoding::percent_decode_str(host)
.decode_utf8()
.expect("percent-encoded hostname contained non-UTF-8 bytes")
})
.unwrap_or("/var/run/postgresql".into());
if host.starts_with("/") {
let path = format!("{}/.s.PGSQL.{}", host, port);
MaybeTlsStream::connect_uds(&path).await?
} else {
MaybeTlsStream::connect(&host, port).await?
}
};
#[cfg(not(unix))]
let stream = MaybeTlsStream::connect(host.unwrap_or("localhost"), port).await?;