Django:Postgres连接没有关闭

时间:2016-02-26 14:43:09

标签: python django postgresql celery webfaction

我遇到的问题是我的django应用程序会累积postgres连接。似乎大约每30分钟建立一个新的连接并且旧的连接不会关闭(见屏幕)。 一段时间后,所有连接都被阻止,最大连接数设置为100。

有谁知道造成这个问题的原因是什么?

enter image description here

我在整合了一些celery tasks之后发现了这一点。所以我很确定它与芹菜有关。

所以我尝试使用after_return方法在每个任务后手动关闭连接:

from django.db import connection

class DBTask(Task):
    abstract = True

    def after_return(self, *args, **kwargs):
        connection.close()

@task(name='example', base=DBTask)
def example_task(value):
    # do some stuff

但这也没有用。 也许我完全错了,它根本不与celery相关。

我的数据库配置:

DATABASES = {
    'default': {
        'ENGINE': 'django.contrib.gis.db.backends.postgis',
        'NAME': 'production', 
        'USER': 'production', 
        'HOST': 'some.host', 
        'CONN_MAX_AGE': 0,
    },
}

已安装的软件包:

  • django 1.8.9
  • pyscopg2 2.6.1
  • 芹菜3.1.20
  • django-celery 3.1.17

该应用程序部署在webfaction(可能有帮助)

我也看过这个question,但设置CONN_MAX_AGE: 0并没有帮助。

更新

尝试在每个芹菜任务结束时添加connection.close(),但连接数仍在增加。

更新2:

尝试在芹菜文件的顶部添加connection.close(),但这也没有帮助。

更新3:

以下是我在芹菜任务中实际使用的代码:

celery_tasks.py

@task(name='push_notifications', base=DBTask)
def push_notifications_task(user_id):
    user = CustomUser.objects.get(id=user_id)
    PusherAPI().push_notifications(user)
    connection.close()

models.py

class PusherAPI(object):

    def push_notifications(self, user):
        from .serializers import NotificationSerializer
        self.pusher.trigger(
            'user_%s' % user.slug,
            'notifications',
            NotificationSerializer(user).data
        )

serializers.py

class NotificationSerializer(object):

    def __init__(self, user=None):
        if user is None:
            self.user = get_current_user()
        else:
            self.user = user

    @property
    def data(self):
        # get notifications from db
        notifications = self.user.notifications.unread()
        # create the notification dict
        ...
        return note_dict

唯一的数据库查询位于CustomUser.objects.get(id=user_id)notifications = self.user.notifications.unread()

1 个答案:

答案 0 :(得分:2)

确保它实际上是未关闭的旧连接,而不是新连接堆积,因为应用程序的某些部分无法处理负载。看看各个连接,例如与<script src="https://ajax.googleapis.com/ajax/libs/jquery/2.1.1/jquery.min.js"></script> <input type="text" class="emailError" />