Merge pull request #158 from MacroConnections/sqlalchemy-connection-p…

…ooling Adding session teardown and changing pool size
CenterForCollectiveLearning · Apr 11, 2016 · ba689d9 · ba689d9
2 parents e4310a4 + 68e3fda
commit ba689d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/config.py b/config.py
@@ -21,6 +21,9 @@ class BaseConfig(object):
 
     COMPRESS = True
 
+    SQLALCHEMY_POOL_SIZE=20
+    SQLALCHEMY_MAX_OVERFLOW=100
+
     COOKIE_DOMAIN = None
     REMEMBER_COOKIE_DOMAIN = COOKIE_DOMAIN
     SESSION_COOKIE_DOMAIN = COOKIE_DOMAIN

diff --git a/dive/core.py b/dive/core.py
@@ -50,6 +50,10 @@ def create_app(**kwargs):
         allow_headers='Content-Type'
     )
 
+    @app.teardown_appcontext
+    def shutdown_session(exception=None):
+        db.session.remove()
+
     ensure_directories(app)
     return app