add create tidb serverless job control (#10467)

Co-authored-by: crazywoola <427733928@qq.com>
This commit is contained in:
Jyong
2024-11-08 18:48:12 +08:00
committed by GitHub
parent 25ca0278dd
commit 445dcfe4d0
6 changed files with 14 additions and 0 deletions

View File

@@ -12,6 +12,8 @@ from models.dataset import TidbAuthBinding
@app.celery.task(queue="dataset")
def create_tidb_serverless_task():
click.echo(click.style("Start create tidb serverless task.", fg="green"))
if not dify_config.CREATE_TIDB_SERVICE_JOB_ENABLED:
return
tidb_serverless_number = dify_config.TIDB_SERVERLESS_NUMBER
start_at = time.perf_counter()
while True: