diff options
-rw-r--r-- | src/main/scala/NamespacedCodegen.scala | 12 |
1 files 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, |