Add module for Databend (#9148)

Author: hantmac

.github/ISSUE_TEMPLATE/bug_report.yaml

@@ -23,6 +23,7 @@ body:
         - Consul
         - Couchbase
         - CrateDB
+        - Databend
         - DB2
         - Dynalite
         - Elasticsearch

.github/ISSUE_TEMPLATE/enhancement.yaml

@@ -23,6 +23,7 @@ body:
         - Consul
         - Couchbase
         - CrateDB
+        - Databend
         - DB2
         - Dynalite
         - Elasticsearch

.github/ISSUE_TEMPLATE/feature.yaml

@@ -23,6 +23,7 @@ body:
         - CrateDB
         - Consul
         - Couchbase
+        - Databend
         - DB2
         - Dynalite
         - Elasticsearch

.github/dependabot.yml

@@ -90,6 +90,10 @@ updates:
     directory: "/modules/database-commons"
     schedule:
       interval: "weekly"
+  - package-ecosystem: "gradle"
+    directory: "/modules/databend"
+    schedule:
+      interval: "weekly"
     open-pull-requests-limit: 10
   - package-ecosystem: "gradle"
     directory: "/modules/db2"

.github/labeler.yml

@@ -55,6 +55,10 @@
   - changed-files:
     - any-glob-to-any-file:
       - modules/cratedb/**/*
+"modules/databend":
+  - changed-files:
+      - any-glob-to-any-file:
+          - modules/databend/**/*
 "modules/db2":
   - changed-files:
     - any-glob-to-any-file:

.github/settings.yml

@@ -271,6 +271,9 @@ labels:
   - name: modules/yugabytedb
     color: '#006b75'
 
+  - name: modules/databend
+    color: '#006b75'
+
   - name: os/linux
     color: '#1d76db'
 

docs/modules/databases/databend.md

@@ -0,0 +1,24 @@
+# Databend Module
+
+## Adding this module to your project dependencies
+
+Add the following dependency to your `pom.xml`/`build.gradle` file:
+
+=== "Gradle"
+    ```groovy
+    testImplementation "org.testcontainers:databend:{{latest_version}}"
+    ```
+
+=== "Maven"
+    ```xml
+    <dependency>
+        <groupId>org.testcontainers</groupId>
+        <artifactId>databend</artifactId>
+        <version>{{latest_version}}</version>
+        <scope>test</scope>
+    </dependency>
+    ```
+
+!!! hint
+Adding this Testcontainers library JAR will not automatically add a database driver JAR to your project. You should ensure that your project also has a suitable database driver as a dependency.
+

mkdocs.yml

@@ -56,6 +56,7 @@ nav:
                 - modules/databases/couchbase.md
                 - modules/databases/clickhouse.md
                 - modules/databases/cratedb.md
+                - modules/databases/databend.md
                 - modules/databases/db2.md
                 - modules/databases/dynalite.md
                 - modules/databases/influxdb.md

modules/databend/build.gradle

@@ -0,0 +1,9 @@
+description = "Testcontainers :: JDBC :: Databend"
+
+dependencies {
+    api project(':jdbc')
+
+    testImplementation project(':jdbc-test')
+    testRuntimeOnly 'com.databend:databend-jdbc:0.2.9'
+    testImplementation 'org.assertj:assertj-core:3.26.3'
+}

modules/databend/src/main/java/org/testcontainers/databend/DatabendContainer.java

@@ -0,0 +1,112 @@
+package org.testcontainers.databend;
+
+import org.testcontainers.containers.JdbcDatabaseContainer;
+import org.testcontainers.containers.wait.strategy.Wait;
+import org.testcontainers.utility.DockerImageName;
+
+import java.util.HashSet;
+import java.util.Set;
+
+/**
+ * Testcontainers implementation for Databend.
+ * <p>
+ * Supported image: {@code datafuselabs/databend}
+ * <p>
+ * Exposed ports:
+ * <ul>
+ *     <li>Database: 8000</li>
+ * </ul>
+ */
+public class DatabendContainer extends JdbcDatabaseContainer<DatabendContainer> {
+
+    static final String NAME = "databend";
+
+    static final DockerImageName DOCKER_IMAGE_NAME = DockerImageName.parse("datafuselabs/databend");
+
+    private static final Integer HTTP_PORT = 8000;
+
+    private static final String DRIVER_CLASS_NAME = "com.databend.jdbc.DatabendDriver";
+
+    private static final String JDBC_URL_PREFIX = "jdbc:" + NAME + "://";
+
+    private static final String TEST_QUERY = "SELECT 1";
+
+    private String databaseName = "default";
+
+    private String username = "databend";
+
+    private String password = "databend";
+
+    public DatabendContainer(String dockerImageName) {
+        this(DockerImageName.parse(dockerImageName));
+    }
+
+    public DatabendContainer(final DockerImageName dockerImageName) {
+        super(dockerImageName);
+        dockerImageName.assertCompatibleWith(DOCKER_IMAGE_NAME);
+
+        addExposedPorts(HTTP_PORT);
+        waitingFor(Wait.forHttp("/").forResponsePredicate(response -> response.equals("Ok.")));
+    }
+
+    @Override
+    protected void configure() {
+        withEnv("QUERY_DEFAULT_USER", this.username);
+        withEnv("QUERY_DEFAULT_PASSWORD", this.password);
+    }
+
+    @Override
+    public Set<Integer> getLivenessCheckPortNumbers() {
+        return new HashSet<>(getMappedPort(HTTP_PORT));
+    }
+
+    @Override
+    public String getDriverClassName() {
+        return DRIVER_CLASS_NAME;
+    }
+
+    @Override
+    public String getJdbcUrl() {
+        return (
+            JDBC_URL_PREFIX +
+            getHost() +
+            ":" +
+            getMappedPort(HTTP_PORT) +
+            "/" +
+            this.databaseName +
+            constructUrlParameters("?", "&")
+        );
+    }
+
+    @Override
+    public String getUsername() {
+        return this.username;
+    }
+
+    @Override
+    public String getPassword() {
+        return this.password;
+    }
+
+    @Override
+    public String getDatabaseName() {
+        return this.databaseName;
+    }
+
+    @Override
+    public String getTestQueryString() {
+        return TEST_QUERY;
+    }
+
+    @Override
+    public DatabendContainer withUsername(String username) {
+        this.username = username;
+        return this;
+    }
+
+    @Override
+    public DatabendContainer withPassword(String password) {
+        this.password = password;
+        return this;
+    }
+}