Add Milvus module (#8352)

Author: eddumelendez

.github/ISSUE_TEMPLATE/bug_report.yaml

@@ -33,6 +33,7 @@ body:
         - Kafka
         - LocalStack
         - MariaDB
+        - Milvus
         - MinIO
         - MockServer
         - MongoDB

.github/ISSUE_TEMPLATE/enhancement.yaml

@@ -33,6 +33,7 @@ body:
         - Kafka
         - LocalStack
         - MariaDB
+        - Milvus
         - MinIO
         - MockServer
         - MongoDB

.github/ISSUE_TEMPLATE/feature.yaml

@@ -33,6 +33,7 @@ body:
         - Kafka
         - LocalStack
         - MariaDB
+        - Milvus
         - MinIO
         - MockServer
         - MongoDB

.github/dependabot.yml

@@ -166,6 +166,11 @@ updates:
     ignore:
       - dependency-name: "org.mariadb:r2dbc-mariadb"
         update-types: [ "version-update:semver-minor" ]
+  - package-ecosystem: "gradle"
+    directory: "/modules/milvus"
+    schedule:
+      interval: "weekly"
+    open-pull-requests-limit: 10
   - package-ecosystem: "gradle"
     directory: "/modules/minio"
     schedule:

.github/labeler.yml

@@ -99,6 +99,10 @@
   - changed-files:
     - any-glob-to-any-file:
       - modules/mariadb/**/*
+"modules/milvus":
+  - changed-files:
+    - any-glob-to-any-file:
+      - modules/milvus/**/*
 "modules/minio":
   - changed-files:
     - any-glob-to-any-file:

.github/settings.yml

@@ -169,6 +169,9 @@ labels:
   - name: modules/mariadb
     color: '#006b75'
 
+  - name: modules/milvus
+    color: '#006b75'
+
   - name: modules/minio
     color: '#006b75'
 

docs/modules/milvus.md

@@ -0,0 +1,36 @@
+# Milvus
+
+Testcontainers module for [Milvus](https://hub.docker.com/r/milvusdb/milvus).
+
+## Milvus's usage examples
+
+You can start a Milvus container instance from any Java application by using:
+
+<!--codeinclude-->
+[Default config](../../modules/milvus/src/test/java/org/testcontainers/milvus/MilvusContainerTest.java) inside_block:milvusContainer
+<!--/codeinclude-->
+
+With external Etcd:
+
+<!--codeinclude-->
+[External Etcd](../../modules/milvus/src/test/java/org/testcontainers/milvus/MilvusContainerTest.java) inside_block:externalEtcd
+<!--/codeinclude-->
+
+## Adding this module to your project dependencies
+
+Add the following dependency to your `pom.xml`/`build.gradle` file:
+
+=== "Gradle"
+    ```groovy
+    testImplementation "org.testcontainers:milvus:{{latest_version}}"
+    ```
+
+=== "Maven"
+    ```xml
+    <dependency>
+        <groupId>org.testcontainers</groupId>
+        <artifactId>milvus</artifactId>
+        <version>{{latest_version}}</version>
+        <scope>test</scope>
+    </dependency>
+    ```

mkdocs.yml

@@ -84,6 +84,7 @@ nav:
           - modules/k3s.md
           - modules/kafka.md
           - modules/localstack.md
+          - modules/milvus.md
           - modules/minio.md
           - modules/mockserver.md
           - modules/nginx.md

modules/milvus/build.gradle

@@ -0,0 +1,8 @@
+description = "Testcontainers :: ActiveMQ"
+
+dependencies {
+    api project(':testcontainers')
+
+    testImplementation 'org.assertj:assertj-core:3.25.1'
+    testImplementation 'io.milvus:milvus-sdk-java:2.3.4'
+}

modules/milvus/src/main/java/org/testcontainers/milvus/MilvusContainer.java

@@ -0,0 +1,61 @@
+package org.testcontainers.milvus;
+
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.wait.strategy.Wait;
+import org.testcontainers.utility.DockerImageName;
+import org.testcontainers.utility.MountableFile;
+
+/**
+ * Testcontainers implementation for Milvus.
+ * <p>
+ * Supported image: {@code milvusdb/milvus}
+ * <p>
+ * Exposed ports:
+ * <ul>
+ *     <li>Management port: 9091</li>
+ *     <li>HTTP: 19530</li>
+ * </ul>
+ */
+public class MilvusContainer extends GenericContainer<MilvusContainer> {
+
+    private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse("milvusdb/milvus");
+
+    private String etcdEndpoint;
+
+    public MilvusContainer(String image) {
+        this(DockerImageName.parse(image));
+    }
+
+    public MilvusContainer(DockerImageName dockerImageName) {
+        super(dockerImageName);
+        dockerImageName.assertCompatibleWith(DEFAULT_IMAGE_NAME);
+        withExposedPorts(9091, 19530);
+        waitingFor(Wait.forHttp("/healthz").forPort(9091));
+        withCommand("milvus", "run", "standalone");
+        withCopyFileToContainer(
+            MountableFile.forClasspathResource("testcontainers/embedEtcd.yaml"),
+            "/milvus/configs/embedEtcd.yaml"
+        );
+        withEnv("COMMON_STORAGETYPE", "local");
+    }
+
+    @Override
+    protected void configure() {
+        if (this.etcdEndpoint == null) {
+            withEnv("ETCD_USE_EMBED", "true");
+            withEnv("ETCD_DATA_DIR", "/var/lib/milvus/etcd");
+            withEnv("ETCD_CONFIG_PATH", "/milvus/configs/embedEtcd.yaml");
+        } else {
+            withEnv("ETCD_ENDPOINTS", this.etcdEndpoint);
+        }
+    }
+
+    public MilvusContainer withEtcdEndpoint(String etcdEndpoint) {
+        this.etcdEndpoint = etcdEndpoint;
+        return this;
+    }
+
+    public String getEndpoint() {
+        return "http://" + getHost() + ":" + getMappedPort(19530);
+    }
+}