Add Connection::close

Author: sfackler

postgres-tokio/src/lib.rs

@@ -581,6 +581,15 @@ impl Connection {
             .boxed()
     }
 
+    pub fn close(self) -> BoxFuture<(), Error> {
+        let mut terminate = vec![];
+        frontend::terminate(&mut terminate);
+        self.0.send(terminate)
+            .map(|_| ())
+            .map_err(Error::Io)
+            .boxed()
+    }
+
     pub fn cancel_data(&self) -> CancelData {
         self.0.cancel_data
     }

postgres-tokio/src/test.rs

@@ -8,9 +8,9 @@ use error::{Error, ConnectError, SqlState};
 fn basic() {
     let mut l = Core::new().unwrap();
     let handle = l.handle();
-    let done = Connection::connect("postgres://postgres@localhost", &handle);
-    let conn = l.run(done).unwrap();
-    assert!(conn.cancel_data().process_id != 0);
+    let done = Connection::connect("postgres://postgres@localhost", &handle)
+        .then(|c| c.unwrap().close());
+    l.run(done).unwrap();
 }
 
 #[test]