zzb
/
pdf_code
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

milvus 修改分区写入测试版本v1.2 修为为每次删除前面所有的分区

This commit is contained in:
吴登数 2024-12-09 15:09:54 +08:00
parent b4221e3906
commit 15e33eadf7
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
zzb_data_prod/Mil_unit.py
View File

@ -10,10 +10,6 @@ def create_partition_by_hour(current_hour):
collection_name = "pdf_measure_v4" collection_name = "pdf_measure_v4"
collection = Collection(collection_name) collection = Collection(collection_name)
# 获取当前时间
now = datetime.now()
# 创建当前小时的分区 # 创建当前小时的分区
partition_name = f"partition_{current_hour}" partition_name = f"partition_{current_hour}"
if not collection.has_partition(partition_name): if not collection.has_partition(partition_name):
@ -22,12 +18,9 @@ def create_partition_by_hour(current_hour):
partition = collection.partition(partition_name) partition = collection.partition(partition_name)
partition.load() partition.load()
# 删除前2个小时的分区
# 获取所有分区 # 获取所有分区
partitions = collection.partitions partitions = collection.partitions
# 删除所有分区（除了默认分区和当前分区）
# 删除所有分区（除了默认分区）
for partition in partitions: for partition in partitions:
name = partition.name name = partition.name
if name not in ["_default", partition_name]: # 保留默认分区 if name not in ["_default", partition_name]: # 保留默认分区