mutil thread

api/controllers/console/datasets/datasets_segments.py

@@ -16,7 +16,6 @@ from models.dataset import DocumentSegment
 
 from libs.helper import TimestampField
 from services.dataset_service import DatasetService, DocumentService, SegmentService
-from tasks.test_task import test_task
 from tasks.enable_segment_to_index_task import enable_segment_to_index_task
 from tasks.remove_segment_from_index_task import remove_segment_from_index_task
 
@@ -285,15 +284,6 @@ class DatasetDocumentSegmentUpdateApi(Resource):
         }, 200
 
 
-class DatasetDocumentTest(Resource):
-    @setup_required
-    @login_required
-    @account_initialization_required
-    def patch(self):
-        test_task.delay()
-        return 200
-
-
 api.add_resource(DatasetDocumentSegmentListApi,
                  '/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments')
 api.add_resource(DatasetDocumentSegmentApi,
@@ -302,5 +292,4 @@ api.add_resource(DatasetDocumentSegmentAddApi,
                  '/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segment')
 api.add_resource(DatasetDocumentSegmentUpdateApi,
                  '/datasets/<uuid:dataset_id>/documents/<uuid:document_id>/segments/<uuid:segment_id>')
-api.add_resource(DatasetDocumentTest,
-                 '/datasets/test')
+

api/tasks/test_task.py

-import threading
-from time import sleep, ctime
-from typing import List
-
-from celery import shared_task
-
-
-
-
-@shared_task
-def test_task():
-    """
-    Clean dataset when dataset deleted.
-
-    Usage: test_task.delay(dataset_id, tenant_id, indexing_technique, index_struct)
-    """
-    print('---开始---:%s' % ctime())
-
-    def smoke(count: List):
-        for i in range(3):
-            print("smoke...%d" % i)
-            count.append("smoke...%d" % i)
-            sleep(1)
-
-    def drunk(count: List):
-        for i in range(3):
-            print("drink...%d" % i)
-            count.append("drink...%d" % i)
-            sleep(10)
-    count = []
-    threads = []
-    for i in range(3):
-        t1 = threading.Thread(target=smoke, kwargs={'count': count})
-        t2 = threading.Thread(target=drunk, kwargs={'count': count})
-        threads.append(t1)
-        threads.append(t2)
-        t1.start()
-        t2.start()
-    for thread in threads:
-        thread.join()
-    print(str(count))
-    # sleep(5) #
-    print('---结束---:%s' % ctime())
\ No newline at end of file