collection相关操作

collection相关操作

import sys

from pymilvus import (
    connections,
    Collection,
)
from pymilvus.client.types import LoadState
from pymilvus.orm import utility

collection_name = "hello_milvus"
host = "192.168.230.71"
port = 19530
username = ""
password = ""

if __name__ == "__main__":
    print("start connecting to Milvus")
    connections.connect("default", host=host, port=port,user=username,password=password)

    print("test existing collection")
    #default是connection的别名
    col_exists = utility.has_collection(using="default",collection_name=collection_name)
    if not col_exists:
        print(collection_name, "不存在")
        sys.exit()
    coll = Collection(collection_name)
    load_stats = utility.load_state(using="default",collection_name=collection_name)
    if load_stats == LoadState.Loaded:
        print(collection_name, "已加载")
        print("释放collection")
        coll.release()
    elif load_stats == LoadState.NotLoad:
        print(collection_name, "未加载")
        print("加载collection")
        coll.load()
    elif load_stats == LoadState.Loading:
        print(collection_name, "加载中")
    else:
        print(collection_name, "状态未知:",load_stats)
    print("删除collection")
    coll.drop()
    print("done")

使用了utility工具类判断collection是否存在、load状态。