From 2d75c3a263e16342d17338a1d78f8b0b82182aef Mon Sep 17 00:00:00 2001
From: Zach Smith <zach@driver.xyz>
Date: Tue, 14 Feb 2017 12:38:03 -0800
Subject: Wrap in try finally

---
 src/main/scala/NamespacedCodegen.scala | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala
index 22ee890..c9f613a 100644
--- a/src/main/scala/NamespacedCodegen.scala
+++ b/src/main/scala/NamespacedCodegen.scala
@@ -32,11 +32,13 @@ object Generator {
     val parsedSchemasOpt: Option[Map[String, List[String]]] =
       schemaNames.map(SchemaParser.parse)
 
-    val dbModel: Model = Await.result(
-      dc.db.run(SchemaParser.createModel(dc.driver, parsedSchemasOpt)),
-      Duration.Inf)
-
-    dc.db.close()
+    val dbModel: Model = try {
+      Await.result(
+        dc.db.run(SchemaParser.createModel(dc.driver, parsedSchemasOpt)),
+        Duration.Inf)
+    } finally {
+      dc.db.close()
+    }
 
     val generator = new Generator(uri,
                                   pkg,
-- 
cgit v1.2.3