From b4221e39068c92ecdaa2cc625f5c8418a4a0a319 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=B4=E7=99=BB=E6=95=B0?=
 <11035577+wu-dengshu@user.noreply.gitee.com>
Date: Mon, 9 Dec 2024 14:51:26 +0800
Subject: [PATCH] =?UTF-8?q?milvus=20=E4=BF=AE=E6=94=B9=E5=88=86=E5=8C=BA?=
 =?UTF-8?q?=E5=86=99=E5=85=A5=E6=B5=8B=E8=AF=95=E7=89=88=E6=9C=ACv1.2=20?=
 =?UTF-8?q?=E4=BF=AE=E4=B8=BA=E4=B8=BA=E6=AF=8F=E6=AC=A1=E5=88=A0=E9=99=A4?=
 =?UTF-8?q?=E5=89=8D=E9=9D=A2=E6=89=80=E6=9C=89=E7=9A=84=E5=88=86=E5=8C=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zzb_data_prod/Mil_unit.py | 18 +++++++++++-------
 zzb_data_prod/app.py      | 16 ++++++++--------
 2 files changed, 19 insertions(+), 15 deletions(-)

diff --git a/zzb_data_prod/Mil_unit.py b/zzb_data_prod/Mil_unit.py
index 869bc5b..ea30889 100644
--- a/zzb_data_prod/Mil_unit.py
+++ b/zzb_data_prod/Mil_unit.py
@@ -23,14 +23,18 @@ def create_partition_by_hour(current_hour):
         partition.load()
 
     # 删除前2个小时的分区
-    previous_hour = (now - timedelta(hours=2)).strftime("%Y%m%d%H")
-    previous_partition_name = f"partition_{previous_hour}"
-    if collection.has_partition(previous_partition_name):
 
-        pre_partition = collection.partition(previous_partition_name)
-        pre_partition.release()
-        collection.drop_partition(previous_partition_name)
-        print(f"Dropped partition: {previous_partition_name}")
+    # 获取所有分区
+    partitions = collection.partitions
+
+    # 删除所有分区（除了默认分区）
+    for partition in partitions:
+        name = partition.name
+        if name not in ["_default", partition_name]:  # 保留默认分区
+            pre_partition = collection.partition(name)
+            pre_partition.release()
+            collection.drop_partition(name)
+            print(f"Partition '{name}' deleted.")
 
 
 
diff --git a/zzb_data_prod/app.py b/zzb_data_prod/app.py
index 24d58af..bca3cf2 100644
--- a/zzb_data_prod/app.py
+++ b/zzb_data_prod/app.py
@@ -224,13 +224,13 @@ app.post("/parser/start",
 # 运行 FastAPI 应用
 if __name__ == "__main__":
     # 服务器启动服务
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=config.PORT)
+    # import uvicorn
+    # uvicorn.run(app, host="0.0.0.0", port=config.PORT)
 
     # 本地调试任务
-    # job_queue.put({
-    #  'file_path' : '1.pdf',
-    #  'file_id' : '2122'
-    #  })
-    #
-    # run_job()
+    job_queue.put({
+     'file_path' : '3.pdf',
+     'file_id' : '2122'
+     })
+
+    run_job()