Don't we need a DbConnectionsMiddleware ?

[dperetti]

... like django-dramatiq has ?

[jarekwg]

Ha yes. We actually wrote our own in our production code. Should definitely live in this repo.
Feel free to add it, otherwise we'll get round to it eventually...

[pedrospdc]

@jarekwg any chance you could paste here a snippet of your middleware? I'm happy to add it myself :)

[jarekwg]

Here you are @pedrospdc

class DbConnectionsMiddleware(Middleware):
    """This middleware cleans up db connections on worker shutdown.
    """

    def _close_connections(self, *args, **kwargs):
        for conn in db.connections.all():
            conn.close()

    before_process_message = _close_connections
    after_process_message = _close_connections

    before_consumer_thread_shutdown = _close_connections
    before_worker_thread_shutdown = _close_connections
    before_worker_shutdown = _close_connections

Thanks in advance! :)