工具类 - 检测SQL文件是Oracle还是Mysql

package jna;

/**
 * @author wushaopei
 * @create 2023-04-27 9:28
 */
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class SqlTypeChecker {
    public static void main(String[] args) {
        String fileName = "test.sql";
        String sqlContent = readSqlFile(fileName);

        if (isOracle(sqlContent)) {
            System.out.println("The SQL script is for Oracle database.");
        } else if (isMySQL(sqlContent)) {
            System.out.println("The SQL script is for MySQL database.");
        } else {
            System.out.println("Unable to determine the database type.");
        }
    }

    private static String readSqlFile(String fileName) {
        StringBuilder sb = new StringBuilder();
        try (BufferedReader br = new BufferedReader(new FileReader(fileName))) {
            String line;
            while ((line = br.readLine()) != null) {
                sb.append(line);
                sb.append("\n");
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return sb.toString();
    }

    private static boolean isOracle(String sqlContent) {
        Pattern pattern = Pattern.compile("(?i)\\bDUAL\\b");
        Matcher matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            return true;
        }

        pattern = Pattern.compile("(?i)\\bCREATE\\s+TABLE\\b");
        matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            pattern = Pattern.compile("(?i)\\bSEQUENCE\\b");
            matcher = pattern.matcher(sqlContent);
            if (matcher.find()) {
                return true;
            }

            pattern = Pattern.compile("(?i)\\bCONSTRAINT\\b.*\\bPRIMARY\\s+KEY\\b");
            matcher = pattern.matcher(sqlContent);
            if (matcher.find()) {
                return true;
            }
        }

        pattern = Pattern.compile("(?i)\\bTRUNCATE\\s+TABLE\\b");
        matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            return true;
        }

        return false;
    }

    private static boolean isMySQL(String sqlContent) {
        Pattern pattern = Pattern.compile("(?i)\\bSELECT\\s+1\\b");
        Matcher matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            return true;
        }

        pattern = Pattern.compile("(?i)\\bCREATE\\s+TABLE\\b");
        matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            pattern = Pattern.compile("(?i)\\bENGINE\\b.*\\bInnoDB\\b");
            matcher = pattern.matcher(sqlContent);
            if (matcher.find()) {
                return true;
            }

            pattern = Pattern.compile("(?i)\\bPRIMARY\\s+KEY\\b");
            matcher = pattern.matcher(sqlContent);
            if (matcher.find()) {
                return true;
            }
        }

        pattern = Pattern.compile("(?i)\\bTRUNCATE\\s+TABLE\\b");
        matcher = pattern.matcher(sqlContent);
        if (matcher.find()) {
            return true;
        }

        return false;
    }
}

上述代码中,readSqlFile方法用于从文件中读取SQL脚本,isOracleisMySQL方法用于检查SQL脚本中是否包含特定的语法和关键字,以确定它是Oracle还是MySQL的。最后在main方法中调用这些方法并打印结果。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值