Merge pull request ibis-project#7 from maxmzkr/close-con-on-tmp-fail

Close after unsuccessful ensure temp directory
maxmzkr · Sep 30, 2016 · cbf12c6 · cbf12c6
2 parents b9aa053 + 4ce692f
commit cbf12c6
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ibis/impala/api.py b/ibis/impala/api.py
@@ -95,9 +95,13 @@ def connect(host='localhost', port=21050, database='default', timeout=45,
     }
 
     con = ImpalaConnection(pool_size=pool_size, **params)
-    client = ImpalaClient(con, hdfs_client=hdfs_client)
+    try:
+        client = ImpalaClient(con, hdfs_client=hdfs_client)
 
-    if options.default_backend is None:
-        options.default_backend = client
+        if options.default_backend is None:
+            options.default_backend = client
+    except:
+        con.close()
+        raise
 
     return client