From 8a0eab05023a261f35cef39ca6be019d8a7a61ec Mon Sep 17 00:00:00 2001
From: Joe Wilm <joe@jwilm.com>
Date: Fri, 27 Apr 2018 16:52:56 -0700
Subject: [PATCH] Handle option being None

---
 postgres/src/lib.rs | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/postgres/src/lib.rs b/postgres/src/lib.rs
index 61ed3d27..08f6168d 100644
--- a/postgres/src/lib.rs
+++ b/postgres/src/lib.rs
@@ -959,13 +959,9 @@ impl InnerConnection {
                     columns = Some(self.parse_cols(Some(body))?);
                 }
                 backend::Message::CommandComplete(_) => {
-                    result.push(
-                        TextRows::new(
-                            // TODO is it safe to unwrap here?
-                            columns.take().unwrap(),
-                            mem::replace(&mut rows, Vec::new())
-                        )
-                    );
+                    if let Some(cols) = columns.take() {
+                        result.push(TextRows::new(cols, mem::replace(&mut rows, Vec::new())));
+                    }
                 }
                 _ => {}
             }