Milvus¶

Introduction¶

The Testcontainers module for Milvus.

Adding this module to your project dependencies¶

Please run the following command to add the Milvus module to your Go dependencies:

go get github.com/testcontainers/testcontainers-go/modules/milvus

Usage example¶

Creating a Milvus container

ctx := context.Background()

milvusContainer, err := milvus.Run(ctx, "milvusdb/milvus:v2.3.9")
defer func() {
    if err := testcontainers.TerminateContainer(milvusContainer); err != nil {
        log.Printf("failed to terminate container: %s", err)
    }
}()
if err != nil {
    log.Printf("failed to start container: %s", err)
    return
}

Module Reference¶

Run function¶

Since v0.32.0

Info

The RunContainer(ctx, opts...) function is deprecated and will be removed in the next major release of Testcontainers for Go.

The Milvus module exposes one entrypoint function to create the Milvus container, and this function receives three parameters:

func Run(ctx context.Context, img string, opts ...testcontainers.ContainerCustomizer) (*MilvusContainer, error)

context.Context, the Go context.
string, the Docker image to use.
testcontainers.ContainerCustomizer, a variadic argument for passing options.

Image¶

Use the second argument in the Run function to set a valid Docker image. In example: Run(context.Background(), "milvusdb/milvus:v2.3.9").

Container Options¶

When starting the Milvus container, you can pass options in a variadic way to configure it.

The following options are exposed by the testcontainers package.

Basic Options¶

WithExposedPorts Since v0.37.0
WithEnv Since v0.29.0
WithWaitStrategy Since v0.20.0
WithAdditionalWaitStrategy Not available until the next release main
WithWaitStrategyAndDeadline Since v0.20.0
WithAdditionalWaitStrategyAndDeadline Not available until the next release main
WithEntrypoint Since v0.37.0
WithEntrypointArgs Since v0.37.0
WithCmd Since v0.37.0
WithCmdArgs Since v0.37.0
WithLabels Since v0.37.0

Lifecycle Options¶

WithLifecycleHooks Not available until the next release main
WithAdditionalLifecycleHooks Not available until the next release main
WithStartupCommand Since v0.25.0
WithAfterReadyCommand Since v0.28.0

Files & Mounts Options¶

Build Options¶

WithDockerfile Since v0.37.0

Logging Options¶

WithLogConsumers Since v0.28.0
WithLogConsumerConfig Not available until the next release main
WithLogger Since v0.29.0

Image Options¶

WithAlwaysPull Not available until the next release main
WithImageSubstitutors Since v0.26.0
WithImagePlatform Not available until the next release main

Networking Options¶

WithNetwork Since v0.27.0
WithNetworkByName Not available until the next release main
WithBridgeNetwork Not available until the next release main
WithNewNetwork Since v0.27.0

Advanced Options¶

WithHostPortAccess Since v0.31.0
WithConfigModifier Since v0.20.0
WithHostConfigModifier Since v0.20.0
WithEndpointSettingsModifier Since v0.20.0
CustomizeRequest Since v0.20.0
WithName Not available until the next release main
WithNoStart Not available until the next release main

Experimental Options¶

WithReuseByName Since v0.37.0

Container Methods¶

The Milvus container exposes the following methods:

ConnectionString¶

Since v0.29.0

This method returns the connection string to connect to the Milvus container, using the default 19530 port.

Get connection string

connectionStr, err := ctr.ConnectionString(ctx)

Examples¶

Creating collections¶

This example shows the usage of the Milvus module to create and retrieve collections.

Create collections

ctx := context.Background()

milvusContainer, err := milvus.Run(ctx, "milvusdb/milvus:v2.3.9")
defer func() {
    if err := testcontainers.TerminateContainer(milvusContainer); err != nil {
        log.Printf("failed to terminate container: %s", err)
    }
}()
if err != nil {
    log.Printf("failed to start container: %s", err)
    return
}

connectionStr, err := milvusContainer.ConnectionString(ctx)
if err != nil {
    log.Printf("failed to get connection string: %s", err)
    return
}

// Create a client to interact with the Milvus container
milvusClient, err := client.NewGrpcClient(context.Background(), connectionStr)
if err != nil {
    log.Print("failed to connect to Milvus:", err.Error())
    return
}
defer milvusClient.Close()

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.Printf("failed to create collection: %s", err)
    return
}

list, err := milvusClient.ListCollections(context.Background())
if err != nil {
    log.Printf("failed to list collections: %s", err)
    return
}