TiberoDB Suport

flyway-database-tibero/pom.xml

@@ -0,0 +1,72 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>org.flywaydb</groupId>
+        <artifactId>flyway-community-db-support</artifactId>
+        <version>10.16.0</version>
+    </parent>
+
+    <artifactId>flyway-database-tibero</artifactId>
+    <name>${project.artifactId}</name>
+
+    <dependencies>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>flyway-core</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.tmax.tibero</groupId>
+            <artifactId>tibero-jdbc</artifactId>
+            <version>7</version>
+            <scope>system</scope>
+            <systemPath>${basedir}/libs/tibero7-jdbc.jar</systemPath>
+        </dependency>
+        <dependency>
+            <groupId>org.junit.jupiter</groupId>
+            <artifactId>junit-jupiter-engine</artifactId>
+            <version>5.10.0</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.assertj</groupId>
+            <artifactId>assertj-core</artifactId>
+            <version>3.24.2</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>testcontainers</artifactId>
+            <version>1.18.0</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>junit-jupiter</artifactId>
+            <version>1.18.0</version>
+            <scope>test</scope>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <resources>
+            <resource>
+                <directory>src/main/resources</directory>
+                <filtering>true</filtering>
+            </resource>
+        </resources>
+        <plugins>
+            <plugin>
+                <artifactId>maven-resources-plugin</artifactId>
+            </plugin>
+            <plugin>
+                <artifactId>maven-jar-plugin</artifactId>
+            </plugin>
+        </plugins>
+    </build>
+
+</project>

flyway-database-tibero/src/main/java/org/flywaydb/community/database/TiberoDatabaseExtension.java

@@ -0,0 +1,27 @@
+package org.flywaydb.community.database;
+
+import org.flywaydb.core.api.FlywayException;
+import org.flywaydb.core.extensibility.PluginMetadata;
+import org.flywaydb.core.internal.util.FileUtils;
+
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.Objects;
+
+public class TiberoDatabaseExtension implements PluginMetadata {
+
+    public String getDescription() {
+        return "Community-contributed Tibero database support extension " + readVersion() + " by Redgate";
+    }
+
+    public static String readVersion() {
+        try {
+            return FileUtils.copyToString(
+                    Objects.requireNonNull(TiberoDatabaseExtension.class.getClassLoader()
+                            .getResourceAsStream("org/flywaydb/community/database/tibero/version.txt")),
+                    StandardCharsets.UTF_8);
+        } catch (IOException e) {
+            throw new FlywayException("Unable to read extension version: " + e.getMessage(), e);
+        }
+    }
+}

flyway-database-tibero/src/main/java/org/flywaydb/community/database/tibero/TiberoConnection.java

@@ -0,0 +1,34 @@
+package org.flywaydb.community.database.tibero;
+
+import java.sql.SQLException;
+import org.flywaydb.core.internal.database.base.Connection;
+import org.flywaydb.core.internal.database.base.Schema;
+
+public class TiberoConnection extends Connection<TiberoDatabase> {
+
+    protected TiberoConnection(TiberoDatabase database, java.sql.Connection connection) {
+        super(database, connection);
+    }
+
+    @Override
+    protected String getCurrentSchemaNameOrSearchPath() throws SQLException {
+        return jdbcTemplate.queryForString("SELECT SYS_CONTEXT('USERENV', 'CURRENT_SCHEMA') FROM DUAL");
+    }
+
+    @Override
+    public void doChangeCurrentSchemaOrSearchPathTo(String schema) throws SQLException {
+        jdbcTemplate.execute("ALTER SESSION SET CURRENT_SCHEMA=" + database.quote(schema));
+    }
+
+    @Override
+    public Schema doGetCurrentSchema() throws SQLException {
+        String currentSchema = getCurrentSchemaNameOrSearchPath();
+
+        return getSchema(currentSchema);
+    }
+
+    @Override
+    public Schema getSchema(String name) {
+        return new TiberoSchema(jdbcTemplate, database, name);
+    }
+}

flyway-database-tibero/src/main/java/org/flywaydb/community/database/tibero/TiberoDatabase.java

@@ -0,0 +1,178 @@
+package org.flywaydb.community.database.tibero;
+
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
+
+import org.flywaydb.core.api.configuration.Configuration;
+import org.flywaydb.core.extensibility.Tier;
+import org.flywaydb.core.internal.database.base.Database;
+import org.flywaydb.core.internal.database.base.Table;
+import org.flywaydb.core.internal.jdbc.JdbcConnectionFactory;
+import org.flywaydb.core.internal.jdbc.StatementInterceptor;
+import org.flywaydb.core.internal.util.StringUtils;
+
+public class TiberoDatabase extends Database<TiberoConnection> {
+
+    public TiberoDatabase(Configuration configuration, JdbcConnectionFactory jdbcConnectionFactory,
+        StatementInterceptor statementInterceptor) {
+        super(configuration, jdbcConnectionFactory, statementInterceptor);
+    }
+
+    public static void enableTiberoTNSNameSupport() {
+        String tiberoAdminEnvVar = System.getenv("TIBERO_ADMIN");
+        String tiberoAdminSysProp = System.getProperty("TIBERO_NET_ADMIN");
+        if (StringUtils.hasLength(tiberoAdminEnvVar) && tiberoAdminSysProp == null) {
+            System.setProperty("TIBERO_NET_ADMIN", tiberoAdminEnvVar);
+        }
+    }
+
+    @Override
+    protected TiberoConnection doGetConnection(Connection connection) {
+        return new TiberoConnection(this, connection);
+    }
+
+    @Override
+    protected String doGetCurrentUser() throws SQLException {
+        return getMainConnection().getJdbcTemplate().queryForString("SELECT USER FROM DUAL");
+    }
+
+    @Override
+    public void ensureSupported(Configuration configuration) {
+        ensureDatabaseIsRecentEnough("7.0");
+        ensureDatabaseNotOlderThanOtherwiseRecommendUpgradeToFlywayEdition("7.0", Tier.PREMIUM, configuration);
+        recommendFlywayUpgradeIfNecessaryForMajorVersion("21.3");
+    }
+
+    @Override
+    public boolean supportsDdlTransactions() {
+        return false;
+    }
+
+    @Override
+    public String getBooleanTrue() {
+        return "1";
+    }
+
+    @Override
+    public String getBooleanFalse() {
+        return "0";
+    }
+
+    @Override
+    public boolean catalogIsSchema() {
+        return false;
+    }
+
+    @Override
+    public String getRawCreateScript(Table table, boolean baseline) {
+        String tablespace = configuration.getTablespace() == null
+            ? ""
+            : " TABLESPACE \"" + configuration.getTablespace() + "\"";
+
+        return "CREATE TABLE " + table + " (\n" +
+            "    \"installed_rank\" NUMBER NOT NULL,\n" +
+            "    \"version\" VARCHAR2(50),\n" +
+            "    \"description\" VARCHAR2(200) NOT NULL,\n" +
+            "    \"type\" VARCHAR2(20) NOT NULL,\n" +
+            "    \"script\" VARCHAR2(1000) NOT NULL,\n" +
+            "    \"checksum\" NUMBER,\n" +
+            "    \"installed_by\" VARCHAR2(100) NOT NULL,\n" +
+            "    \"installed_on\" TIMESTAMP DEFAULT SYSDATE NOT NULL,\n" +
+            "    \"execution_time\" NUMBER NOT NULL,\n" +
+            "    \"success\" NUMBER(1) NOT NULL,\n" +
+            "    CONSTRAINT \"" + table.getName() + "_pk\" PRIMARY KEY (\"installed_rank\")\n" +
+            ")" + tablespace + ";\n" +
+            (baseline ? getBaselineStatement(table) + ";\n" : "") +
+            "CREATE INDEX \"" + table.getSchema().getName() + "\".\"" + table.getName() + "_s_idx\" ON " + table
+            + " (\"success\");\n";
+    }
+
+    boolean isXmlDbAvailable() throws SQLException {
+        return isDataDictViewAccessible("ALL_XML_TABLES");
+    }
+
+    boolean isPrivOrRoleGranted(String name) throws SQLException {
+        return queryReturnsRows("SELECT 1 FROM SESSION_PRIVS WHERE PRIVILEGE = ? UNION ALL " +
+            "SELECT 1 FROM SESSION_ROLES WHERE ROLE = ?", name, name);
+    }
+
+    boolean queryReturnsRows(String query, String... params) throws SQLException {
+        return getMainConnection().getJdbcTemplate()
+            .queryForBoolean("SELECT CASE WHEN EXISTS(" + query + ") THEN 1 ELSE 0 END FROM DUAL", params);
+    }
+
+    private boolean isDataDictViewAccessible(String owner, String name) throws SQLException {
+        return queryReturnsRows("SELECT * FROM DBA_TAB_PRIVS WHERE OWNER = ? AND TABLE_NAME = ?" +
+            " AND PRIVILEGE = 'SELECT'", owner, name);
+    }
+
+    boolean isDataDictViewAccessible(String name) throws SQLException {
+        return isDataDictViewAccessible("SYS", name);
+    }
+
+    Set<String> getSystemSchemas() throws SQLException {
+
+        Set<String> result = new HashSet<>(Arrays.asList(
+            "SYS",
+            "SYSCAT",
+            "OUTLN",
+            "SYSGIS"
+        ));
+
+        result.addAll(getMainConnection().getJdbcTemplate().queryForStringList(
+            "SELECT USERNAME FROM DBA_USERS WHERE USERNAME LIKE 'SYS%'"
+        ));
+
+        return result;
+    }
+
+    boolean isFlashbackDataArchiveAvailable(String schemaName) throws SQLException {
+        String paramQuery = "SELECT COUNT(*) FROM V$PARAMETERS WHERE NAME LIKE '%FLASHBACK%' AND VALUE IS NOT NULL AND (LENGTH(TRIM(VALUE)) > 0 OR VALUE != '0')";
+        int paramCount = getMainConnection().getJdbcTemplate().queryForInt(paramQuery);
+
+        if (paramCount == 0) {
+            return false;
+        }
+
+        String destQuery = "SELECT VALUE FROM V$PARAMETERS WHERE NAME = 'FLASHBACK_LOG_ARCHIVE_DEST'";
+        String flashbackDest = getMainConnection().getJdbcTemplate().queryForString(destQuery);
+
+        if (flashbackDest == null || flashbackDest.trim().isEmpty()) {
+            return false;
+        }
+
+        String logModeQuery = "SELECT LOG_MODE FROM V$DATABASE";
+        String logMode = getMainConnection().getJdbcTemplate().queryForString(logModeQuery);
+
+        if (!"ARCHIVELOG".equalsIgnoreCase(logMode)) {
+            return false;
+        }
+
+        String tablespaceQuery = "SELECT COUNT(*) FROM DBA_USERS u JOIN V$TABLESPACE t ON u.DEFAULT_TABLESPACE = t.NAME WHERE u.USERNAME = ? AND t.FLASHBACK_ON = 'YES'";
+        int flashbackOnCount = getMainConnection().getJdbcTemplate()
+                .queryForInt(tablespaceQuery, schemaName.toUpperCase());
+
+        if (flashbackOnCount == 0) {
+            return false;
+        }
+
+        String privilegeQuery = "SELECT COUNT(*) FROM DBA_SYS_PRIVS WHERE GRANTEE = ? AND (PRIVILEGE = 'FLASHBACK ANY TABLE' OR PRIVILEGE = 'FLASHBACK OBJECT')";
+        int privilegeCount = getMainConnection().getJdbcTemplate()
+                .queryForInt(privilegeQuery, schemaName.toUpperCase());
+
+        return privilegeCount != 0;
+    }
+
+    String dbaOrAll(String baseName) throws SQLException {
+        return isPrivOrRoleGranted("SELECT ANY DICTIONARY") || isDataDictViewAccessible("DBA_" + baseName)
+            ? "DBA_" + baseName
+            : "ALL_" + baseName;
+    }
+
+    boolean isLocatorAvailable() throws SQLException {
+        return isDataDictViewAccessible("SYSGIS", "ALL_GEOMETRY_COLUMNS");
+    }
+}

flyway-database-tibero/src/main/java/org/flywaydb/community/database/tibero/TiberoDatabaseType.java

@@ -0,0 +1,56 @@
+package org.flywaydb.community.database.tibero;
+
+import java.sql.Connection;
+import java.sql.Types;
+import org.flywaydb.core.api.ResourceProvider;
+import org.flywaydb.core.api.configuration.Configuration;
+import org.flywaydb.core.internal.database.base.BaseDatabaseType;
+import org.flywaydb.core.internal.database.base.CommunityDatabaseType;
+import org.flywaydb.core.internal.database.base.Database;
+import org.flywaydb.core.internal.jdbc.JdbcConnectionFactory;
+import org.flywaydb.core.internal.jdbc.StatementInterceptor;
+import org.flywaydb.core.internal.parser.Parser;
+import org.flywaydb.core.internal.parser.ParsingContext;
+
+public class TiberoDatabaseType extends BaseDatabaseType implements CommunityDatabaseType {
+
+    @Override
+    public String getName() {
+        return "Tibero";
+    }
+
+    @Override
+    public int getNullType() {
+        return Types.VARCHAR;
+    }
+
+    @Override
+    public boolean handlesJDBCUrl(String url) {
+        return url.startsWith("jdbc:tibero");
+    }
+
+    @Override
+    public String getDriverClass(String url, ClassLoader classLoader) {
+        return "com.tmax.tibero.jdbc.TbDriver";
+    }
+
+    @Override
+    public boolean handlesDatabaseProductNameAndVersion(String databaseProductName,
+        String databaseProductVersion, Connection connection) {
+        return true;
+    }
+
+    @Override
+    public Database createDatabase(Configuration configuration,
+        JdbcConnectionFactory jdbcConnectionFactory, StatementInterceptor statementInterceptor) {
+        TiberoDatabase.enableTiberoTNSNameSupport();
+
+        return new TiberoDatabase(configuration, jdbcConnectionFactory, statementInterceptor);
+    }
+
+    @Override
+    public Parser createParser(Configuration configuration, ResourceProvider resourceProvider,
+        ParsingContext parsingContext) {
+        return new TiberoParser(configuration, parsingContext, 3);
+    }
+}

flyway-database-tibero/src/main/java/org/flywaydb/community/database/tibero/TiberoParser.java

@@ -0,0 +1,12 @@
+package org.flywaydb.community.database.tibero;
+
+import org.flywaydb.core.api.configuration.Configuration;
+import org.flywaydb.core.internal.parser.Parser;
+import org.flywaydb.core.internal.parser.ParsingContext;
+
+public class TiberoParser extends Parser {
+
+    protected TiberoParser(Configuration configuration, ParsingContext parsingContext, int peekDepth) {
+        super(configuration, parsingContext, peekDepth);
+    }
+}