mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-08-09 10:08:26 +00:00
d3be16ffe8
* perf(perms): 资产授权列表关联数据改为 `prefetch_related` * perf(perms): 优化一波 * dispatch_mapping_node_tasks.delay * perf: 在做一些优化 * perf: 再优化一波 * perf(perms): 授权更改节点慢的问题 * fix: 修改一处bug * perf(perms): ungrouped 资产数量计算方式 * fix: 修复dispatch data中的bug * fix(assets): add_nodes_assets_to_system_users celery task * fix: 修复ungrouped的bug * feat(nodes): 添加 favorite 节点 * feat(node): 添加 favorite api * fix: 修复clean keys的bug Co-authored-by: xinwen <coderWen@126.com> Co-authored-by: ibuler <ibuler@qq.com>
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
|
|
from kombu import Exchange, Queue
|
|
from celery import Celery
|
|
from celery.schedules import crontab
|
|
|
|
# set the default Django settings module for the 'celery' program.
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'jumpserver.settings')
|
|
from jumpserver import settings
|
|
# from django.conf import settings
|
|
|
|
app = Celery('jumpserver')
|
|
|
|
configs = {k: v for k, v in settings.__dict__.items() if k.startswith('CELERY')}
|
|
# Using a string here means the worker will not have to
|
|
# pickle the object when using Windows.
|
|
# app.config_from_object('django.conf:settings', namespace='CELERY')
|
|
configs["CELERY_QUEUES"] = [
|
|
Queue("celery", Exchange("celery"), routing_key="celery"),
|
|
Queue("ansible", Exchange("ansible"), routing_key="ansible"),
|
|
Queue("celery_node_tree", Exchange("celery_node_tree"), routing_key="celery_node_tree")
|
|
]
|
|
configs["CELERY_ROUTES"] = {
|
|
"ops.tasks.run_ansible_task": {'exchange': 'ansible', 'routing_key': 'ansible'},
|
|
}
|
|
|
|
app.namespace = 'CELERY'
|
|
app.conf.update(configs)
|
|
app.autodiscover_tasks(lambda: [app_config.split('.')[0] for app_config in settings.INSTALLED_APPS])
|
|
|
|
app.conf.beat_schedule = {
|
|
'check-asset-permission-expired': {
|
|
'task': 'perms.tasks.check_asset_permission_expired',
|
|
'schedule': settings.PERM_EXPIRED_CHECK_PERIODIC,
|
|
'args': ()
|
|
},
|
|
'check-node-assets-amount': {
|
|
'task': 'assets.tasks.nodes_amount.check_node_assets_amount_celery_task',
|
|
'schedule': crontab(minute=0, hour=0),
|
|
'args': ()
|
|
},
|
|
}
|