Go Milvusの管理

コレクションの作成

このトピックでは、Milvusでコレクションを作成する方法について説明します。

コレクションは1つ以上のパーティションで構成されています。新しいコレクションを作成する際、Milvusは自動的に_defaultという名前のデフォルトパーティションを作成します。

次の例では、book_idというプライマリキーのフィールド、word_countというINT64スカラーフィールド、そしてbook_introという2次元の浮動小数点ベクトルフィールドを含む、2つのシャードを持つbookというコレクションを作成します。実際のアプリケーションでは、この例よりも高い次元のベクトルを使用する場合があります。

コレクションスキーマの定義

作成するコレクションには、プライマリキーフィールドとベクトルフィールドが含まれている必要があります。プライマリキーフィールドはINT64およびVarCharのデータ型をサポートしています。

ヒント: コレクションのスキーマ定義は、MYSQLでテーブル構造を定義するのと似ています。

まず、フィールドスキーマ、コレクションスキーマ、およびコレクション名を含む必要なパラメータを準備します。

コレクションスキーマを定義する前に、コレクションの各フィールド用のスキーマを作成します。Milvusでは、データ挿入の複雑さを簡素化するため、各スカラーフィールドにデフォルト値を指定することができます（プライマリキーフィールドを除く）。これにより、データ挿入時にフィールドが空の場合、フィールドスキーマ作成時に構成されたデフォルト値が使用されます。

var (
    collectionName = "book"
    )
schema := &entity.Schema{
  CollectionName: collectionName,
  Description:    "Test book search",
  Fields: []*entity.Field{
    {
      Name:       "book_id",
      DataType:   entity.FieldTypeInt64,
      PrimaryKey: true,
      AutoID:     false,
    },
    {
      Name:       "word_count",
      DataType:   entity.FieldTypeInt64,
      PrimaryKey: false,
      AutoID:     false,
    },
    {
      Name:     "book_intro",
      DataType: entity.FieldTypeFloatVector,
      TypeParams: map[string]string{
          "dim": "2",
      },
    },
  },
  EnableDynamicField: true
}

定義された構造に基づいてコレクションを作成する

次に、指定されたスキーマを使用してコレクションを作成します。

err = milvusClient.CreateCollection(
    context.Background(), // ctx
    schema,
    2, // shardNum
)
if err != nil {
    log.Fatal("Failed to create collection: ", err.Error())
}

コレクションの制限事項

リソース構成

パラメータ default_value

• default_value は、配列でないJSONスカラーフィールドにのみ適用されます。
• default_value はプライマリキーには適用されません。
• default_value のデータ型は、dtype で指定されたデータ型と同じでなければならず、それ以外の場合はエラーが発生する可能性があります。
• auto_id を使用している場合、残りのすべてのフィールドをデフォルト値を使用するように設定することはできません。つまり、挿入または更新操作を行う際には、少なくとも1つのフィールドに値を指定する必要があります。それ以外の場合、エラーが発生する可能性があります。

コレクションの存在を確認する

collExists, err := milvusClient.HasCollection(ctx, collectionName)
if err != nil {
    log.Fatal("コレクションの存在を確認できませんでした：", err.Error())
}

コレクションを削除する

_ = milvusClient.DropCollection(ctx, collectionName)

すべてのコレクションをクエリする

collections, err := milvusClient.ListCollections(ctx)