dnim/rust-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Setup socket

Browse Source
This commit is contained in:
Steven Fackler 2019-07-25 07:21:53 -07:00
parent 77caff91ef
commit 4396f38fcc
2 changed files with 10 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tokio-postgres/src/connect_socket.rs
View File

@ -31,7 +31,16 @@ pub async fn connect_socket(idx: usize, config: &Config) -> Result<Socket, Error
let mut error = None;
for addr in addrs {
let new_error = match connect_timeout(TcpStream::connect(&addr), config).await {
Ok(socket) => return Ok(Socket::new_tcp(socket)),
Ok(socket) => {
socket.set_nodelay(true).map_err(Error::connect)?;
if config.keepalives {
socket
.set_keepalive(Some(config.keepalives_idle))
.map_err(Error::connect)?;
}
return Ok(Socket::new_tcp(socket));
}
Err(e) => e,
};
error = Some(new_error);

36
tokio-postgres/tests/test/main.rs
View File

@ -141,42 +141,6 @@ async fn insert_select() {
}
/*
#[test]
fn insert_select() {
let _ = env_logger::try_init();
let mut runtime = Runtime::new().unwrap();
let (mut client, connection) = runtime.block_on(connect("user=postgres")).unwrap();
let connection = connection.map_err(|e| panic!("{}", e));
runtime.handle().spawn(connection).unwrap();
runtime
.block_on(
client
.simple_query("CREATE TEMPORARY TABLE foo (id SERIAL, name TEXT)")
.for_each(|_| Ok(())),
)
.unwrap();
let insert = client.prepare("INSERT INTO foo (name) VALUES ($1), ($2)");
let select = client.prepare("SELECT id, name FROM foo ORDER BY id");
let prepare = insert.join(select);
let (insert, select) = runtime.block_on(prepare).unwrap();
let insert = client
.execute(&insert, &[&"alice", &"bob"])
.map(|n| assert_eq!(n, 2));
let select = client.query(&select, &[]).collect().map(|rows| {
assert_eq!(rows.len(), 2);
assert_eq!(rows[0].get::<_, i32>(0), 1);
assert_eq!(rows[0].get::<_, &str>(1), "alice");
assert_eq!(rows[1].get::<_, i32>(0), 2);
assert_eq!(rows[1].get::<_, &str>(1), "bob");
});
let tests = insert.join(select);
runtime.block_on(tests).unwrap();
}
#[test]
fn query_portal() {
let _ = env_logger::try_init();