Java程序实现 测试数据库连接和批量SQL执行

最新推荐文章于 2023-02-02 23:43:47 发布
最新推荐文章于 2023-02-02 23:43:47 发布
阅读量471 收藏 7
点赞数 1
文章标签： jdbc mysql 数据库 java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/su2231595742/article/details/113978317
版权

Java程序实现 测试数据库连接和批量SQL执行

工具类

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.util.StringUtils;
import com.google.common.collect.Lists;
import lombok.extern.slf4j.Slf4j;
import marchsoft.exception.BadRequestException;
import javax.sql.DataSource;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.sql.*;
import java.util.List;

/**
 * @author /
 */
@Slf4j
public class SqlUtils {

	public static final String COLON = ":";


	/**
	 * 获取数据源
	 *
	 * @param jdbcUrl /
	 * @param userName /
	 * @param password /
	 * @return DataSource
	 */
	private static DataSource getDataSource(String jdbcUrl, String userName, String password) {
		DruidDataSource druidDataSource = new DruidDataSource();
		String className;
		try {
			// 为 url 选择一个适当的驱动
			className = DriverManager.getDriver(jdbcUrl.trim()).getClass().getName();
		} catch (SQLException e) {
			throw new BadRequestException("【测试连接数据库失败】Get class name error: =" + jdbcUrl);
		}
		if (StringUtils.isEmpty(className)) {
			// 如果找不到，从自定义的驱动中查找
			DataTypeEnum dataTypeEnum = DataTypeEnum.urlOf(jdbcUrl);
			if (null == dataTypeEnum) {
				throw new BadRequestException("【测试连接数据库失败】Not supported data type: jdbcUrl=" + jdbcUrl);
			}
			druidDataSource.setDriverClassName(dataTypeEnum.getDriver());
		} else {
			druidDataSource.setDriverClassName(className);
		}


		druidDataSource.setUrl(jdbcUrl);
		druidDataSource.setUsername(userName);
		druidDataSource.setPassword(password);
		// 配置获取连接等待超时的时间
		druidDataSource.setMaxWait(3000);
		// 配置初始化大小、最小、最大
		druidDataSource.setInitialSize(1);
		druidDataSource.setMinIdle(1);
		druidDataSource.setMaxActive(1);

		// 如果链接出现异常则直接判定为失败而不是一直重试
		druidDataSource.setBreakAfterAcquireFailure(true);
		try {
			druidDataSource.init();
		} catch (SQLException e) {
			log.error("【测试连接数据库失败】Exception during pool initialization", e);
			throw new RuntimeException(e.getMessage());
		}

		return druidDataSource;
	}

	/**
	 * 连接数据库
	 * @param jdbcUrl /
	 * @param userName /
	 * @param password /
	 * @return connection
	 */
	private static Connection getConnect(String jdbcUrl, String userName, String password) {
		DataSource dataSource = getDataSource(jdbcUrl, userName, password);
		Connection connection;
		try {
			connection = dataSource.getConnection();
			int timeOut = 5;
			if (null == connection || connection.isClosed() || !connection.isValid(timeOut)) {
				log.info("【测试连接数据库失败】connection is closed or invalid, retry get connection!");
				connection = dataSource.getConnection();
			}
		} catch (Exception ignored) {
			log.error("【测试连接数据库失败】create connection error, jdbcUrl: {}", jdbcUrl);
			throw new RuntimeException("【测试连接数据库失败】create connection error, jdbcUrl: " + jdbcUrl);
		}
		return connection;
	}

	private static void releaseConnection(Connection connection) {
		if (null != connection) {
			try {
				connection.close();
			} catch (Exception e) {
				log.error(e.getMessage(),e);
				log.error("【测试连接数据库失败】connection close error：" + e.getMessage());
			}
		}
	}


	public static void closeResult(ResultSet rs) {
		if (rs != null) {
			try {
				rs.close();
			} catch (Exception e) {
				log.error(e.getMessage(),e);
			}
		}
	}

	public static boolean testConnection(String jdbcUrl, String userName, String password) {
		Connection connection = null;
		try {
			connection = getConnect(jdbcUrl, userName, password);
			if (null != connection) {
				return true;
			}
		} catch (Exception e) {
			log.error("【测试连接数据库失败】Get connection failed:" + e.getMessage());
		} finally {
			releaseConnection(connection);
		}
		return false;
	}

	public static String executeFile(String jdbcUrl, String userName, String password, File sqlFile) {
		Connection connection = getConnect(jdbcUrl, userName, password);
		try {
			batchExecute(connection, readSqlList(sqlFile));
		} catch (Exception e) {
			log.error("sql脚本执行发生异常:{}",e.getMessage());
			return e.getMessage();
		}finally {
			releaseConnection(connection);
		}
		return "success";
	}


	/**
	 * 批量执行sql
	 * @param connection /
	 * @param sqlList /
	 */
	public static void batchExecute(Connection connection, List<String> sqlList) throws SQLException {
		Statement st = connection.createStatement();
		for (String sql : sqlList) {
			if (sql.endsWith(";")) {
				sql = sql.substring(0, sql.length() - 1);
			}
			st.addBatch(sql);
		}
		st.executeBatch();
	}

	/**
	 * 将文件中的sql语句以；为单位读取到列表中
	 * @param sqlFile /
	 * @return /
	 * @throws Exception e
	 */
	private static List<String> readSqlList(File sqlFile) throws Exception {
		List<String> sqlList = Lists.newArrayList();
		StringBuilder sb = new StringBuilder();
		try (BufferedReader reader = new BufferedReader(new InputStreamReader(
				new FileInputStream(sqlFile), StandardCharsets.UTF_8))) {
			String tmp;
			while ((tmp = reader.readLine()) != null) {
				log.info("line:{}", tmp);
				if (tmp.endsWith(";")) {
					sb.append(tmp);
					sqlList.add(sb.toString());
					sb.delete(0, sb.length());
				} else {
					sb.append(tmp);
				}
			}
			if (!"".endsWith(sb.toString().trim())) {
				sqlList.add(sb.toString());
			}
		}
		return sqlList;
	}

}

自定义数据库枚举类

import lombok.extern.slf4j.Slf4j;

/**
 * @author /
 */
@Slf4j
@SuppressWarnings({"unchecked","all"})
public enum DataTypeEnum {

    /** mysql */
    MYSQL("mysql", "mysql", "com.mysql.jdbc.Driver", "`", "`", "'", "'"),

    /** oracle */
    ORACLE("oracle", "oracle", "oracle.jdbc.driver.OracleDriver", "\"", "\"", "\"", "\""),

    /** sql server */
    SQLSERVER("sqlserver", "sqlserver", "com.microsoft.sqlserver.jdbc.SQLServerDriver", "\"", "\"", "\"", "\""),

    /** h2 */
    H2("h2", "h2", "org.h2.Driver", "`", "`", "\"", "\""),

    /** phoenix */
    PHOENIX("phoenix", "hbase phoenix", "org.apache.phoenix.jdbc.PhoenixDriver", "", "", "\"", "\""),

    /** mongo */
    MONGODB("mongo", "mongodb", "mongodb.jdbc.MongoDriver", "`", "`", "\"", "\""),

    /** sql4es */
    ELASTICSEARCH("sql4es", "elasticsearch", "nl.anchormen.sql4es.jdbc.ESDriver", "", "", "'", "'"),

    /** presto */
    PRESTO("presto", "presto", "com.facebook.presto.jdbc.PrestoDriver", "", "", "\"", "\""),

    /** moonbox */
    MOONBOX("moonbox", "moonbox", "moonbox.jdbc.MbDriver", "`", "`", "`", "`"),

    /** cassandra */
    CASSANDRA("cassandra", "cassandra", "com.github.adejanovski.cassandra.jdbc.CassandraDriver", "", "", "'", "'"),

    /** click house */
    CLICKHOUSE("clickhouse", "clickhouse", "ru.yandex.clickhouse.ClickHouseDriver", "", "", "\"", "\""),

    /** kylin */
    KYLIN("kylin", "kylin", "org.apache.kylin.jdbc.Driver", "\"", "\"", "\"", "\""),

    /** vertica */
    VERTICA("vertica", "vertica", "com.vertica.jdbc.Driver", "", "", "'", "'"),

    /** sap */
    HANA("sap", "sap hana", "com.sap.db.jdbc.Driver", "", "", "'", "'"),

    /** impala */
    IMPALA("impala", "impala", "com.cloudera.impala.jdbc41.Driver", "", "", "'", "'");

    private String feature;
    private String desc;
    private String driver;
    private String keywordPrefix;
    private String keywordSuffix;
    private String aliasPrefix;
    private String aliasSuffix;

    private static final String JDBC_URL_PREFIX = "jdbc:";

    DataTypeEnum(String feature, String desc, String driver, String keywordPrefix, String keywordSuffix, String aliasPrefix, String aliasSuffix) {
        this.feature = feature;
        this.desc = desc;
        this.driver = driver;
        this.keywordPrefix = keywordPrefix;
        this.keywordSuffix = keywordSuffix;
        this.aliasPrefix = aliasPrefix;
        this.aliasSuffix = aliasSuffix;
    }

    public static DataTypeEnum urlOf(String jdbcUrl) {
        // 将url的 字符串转换为小写，并删除头尾的空白符
        String url = jdbcUrl.toLowerCase().trim();
        for (DataTypeEnum dataTypeEnum : values()) {
            // 通过判断 url 的开头，找到对应的数据库驱动
            if (url.startsWith(JDBC_URL_PREFIX + dataTypeEnum.feature)) {
//                try {
//                    Class<?> aClass = Class.forName(dataTypeEnum.getDriver());
//                    if (null == aClass) {
//                        throw new RuntimeException("Unable to get driver instance for jdbcUrl: " + jdbcUrl);
//                    }
//                } catch (ClassNotFoundException e) {
//                    throw new RuntimeException("Unable to get driver instance: " + jdbcUrl);
//                }
                return dataTypeEnum;
            }
        }
        return null;
    }

    public String getFeature() {
        return feature;
    }

    public String getDesc() {
        return desc;
    }

    public String getDriver() {
        return driver;
    }

    public String getKeywordPrefix() {
        return keywordPrefix;
    }

    public String getKeywordSuffix() {
        return keywordSuffix;
    }

    public String getAliasPrefix() {
        return aliasPrefix;
    }

    public String getAliasSuffix() {
        return aliasSuffix;
    }
}

测试：

public Result<Object> testConnect(@RequestBody Database resources){
    boolean connect = databaseService.testConnect(resources);
    if (!connect){
        log.error(StrUtil.format("【测试连接数据库失败 /api/database/test-connect】操作人id：{}，测试连接目标，Database：{}"), SecurityUtils.getCurrentUserId(),
                resources);
        throw new BadRequestException("测试连接数据库失败");
    }
    log.info(StrUtil.format("【测试数据库连接成功 /api/database/test-connect】操作人id：{}，测试连接目标：{}"), SecurityUtils.getCurrentUserId(),
            resources);
    return Result.success();
}

@Override
public boolean testConnect(Database resources) {
    try{
        return SqlUtils.testConnection(resources.getJdbcUrl(), resources.getUserName(), resources.getPwd());
    }catch (Exception e){
        log.error(e.getMessage());
        return false;
    }
}
渣渣苏
关注
  • 1
    点赞
  • 踩
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
java批量执行SQL
04-04
java批量执行SQL java批量执行SQL.txt
java程序连接数据库并批量执行.sql文件
没有伞的孩子必须努力奔跑！—小林
04-12 3959
在做软件安装程序的时候;数据库信息都是我们自己配的，配完之后连接数据库-->执行我们准备好的指定的sql文件建表建库。我们这里主要用到的方法是：mybatis ScriptRunner import org.apache.commons.configuration.Configuration;import org.apache.commons.configuration.Properti
Java批量执行sql
PlanetWalker的博客
05-22 1276
直接上代码 import java.sql.Connection; import java.sql.SQLException; import java.sql.Statement; import java.util.Date; public class sqlExecute { public void executeBatch() throws Exception { DbUtil dbUtil = new DbUtil(); Connection con =
java 批量执行 sql_JDBC批量执行SQL
weixin_35610224的博客
02-21 1396
批量执行SQL，效率比较高，但执行SQL太多，谨防内存溢出。代码案例一：package com.what21.jdbc.demo05;import java.sql.Connection;import java.sql.DriverManager;import java.sql.SQLException;import java.sql.Statement;public class JDBCDemo...
java批量执行sql语句_JAVA如何对数据库执行批量操作
weixin_28985877的博客
02-12 3502
转载自： http://linduo168.iteye.com/blog/1887328在传统关系型数据库当中，如何对数据库执行批量操作呢 ? 我们了解到在JDBC的底层有 executeBatch操作 。实现批处理有两种方式，第一种方式：Statement.addBatch(sql) (其实是将sql语句 放在了一个 list 集合中。)第二种方式：PreparedStatement.ad...
java连接mysql数据库实现单条插入和批量插入
08-25
Java 连接 MySQL 数据库实现单条插入和批量插入 Java 连接 MySQL 数据库实现单条插入和批量插入是 Java 编程语言中的一种常见操作。在实际项目中，我们经常需要将数据插入到 MySQL 数据库中，以便于存储和管理数据...
Java实现批量导入excel表格数据到数据库中的方法
08-29
我们可以使用Java语言提供的数据库连接池来连接数据库，然后使用SQL语句将数据导入到数据库中。 6. 实现数据导入 下面是一个简单的Java实现批量导入excel表格数据到数据库中的方法的示例代码： ```java public ...
java实现批量导入.csv文件到mysql数据库
08-25
首先，我们需要使用Java的File和Scanner类来读取.csv文件，然后使用MySQL的JDBC驱动来连接数据库，并使用PreparedStatement来批量执行SQL语句。 Java代码实现 首先，我们需要import必要的Java包，包括java.io.File...
Java使用JDBC连接postgresql数据库示例
08-26
Java 使用 JDBC 连接 PostgreSQL 数据库示例 Java 是一种广泛使用的编程语言，...通过加载 PostgreSQL 的 JDBC 驱动程序、获取数据库连接、执行 SQL 语句等步骤，可以实现 Java 应用程序与 PostgreSQL 数据库的集成。
批量执行SQL脚本（附源码）.zip
05-22
在编程实现上，可能会使用如Java的JDBC（Java Database Connectivity）或者Python的psycopg2（PostgreSQL adapter）等库来建立与数据库的连接，然后执行SQL命令。源码的分析和学习将有助于开发者更好地理解如何高效...
java用addBatch()批量执行sql语句实例
Top5软件工程硕士，先后在京东、字节从事多年Java后端开发、实时和离线大数据开发
08-26 1万+
Test .java import java.sql.Connection; import java.sql.DriverManager; import java.sql.ResultSet; import java.sql.Statement; public class Test { static Connection con; // 声明Connection对象 static State
ssm框架中使用了druid连接池无法一次执行多条sql的问题
weixin_44059948的博客
12-01 1171
今天试图在一个ssm框架的项目中在mybatis insert中执行多条sql，出现异常：Caused by: java.sql.SQLException: sql injection violation, multi-statement not allow，特此记录一下 找了许久，发现是druid的wall sql监控问题，wall监控下，默认是不支持批量sql执行的，经过调试发现报错位置为multiStatementAllow设置为false抛出的异常。 解决方法 1. 关闭wall监控 ssm框架.
Java一次执行多条SQL语句
热门推荐
opera95的博客
10-11 2万+
Java 一次执行多条SQL 数据库 java sql 多条 jdbc 概要： 本实例将演示如何一次执行多条SQL语句，这些SQL语句可以是插入语句、更新语句和删除语句。本实例以myuser数据库中的staff表为例。 | |目录 1技术要点 2代码实现 3程序解读 1技术要点 一次执行多条SQL的技术要点如下： Data
clickhouse语句_Clickhouse 批量执行SQL文件等常用语句
weixin_39877182的博客
12-20 4899
有一堆或者太长的SQL需要执行，可以写成一个文件，批量执行：clickhouse-client --user 用户名 --password 密码 -d 数据库 --multiquery < /root/temp.sql查看SQL的执行计划：clickhouse-client -h localhost --send_logs_level=trace <</dev/null导入为...
MyBatis + Druid 启动批量SQL功能
最新发布
℡メ㏑╭ァ小凯
02-02 535
DruidConfig类中修改。Mysql 连接字符串新增。
解决 druid批量执行sql语句报错问题
weixin_44197039的博客
07-28 1617
java.sql.SQLException: sql injection violation, multi-statement not allow(sql注入冲突，不允许多语句)
ClickHouse命令行中执行多行的一句sql
x²＋(y－√³x²)²＝1的博客
05-23 1182
在 clickhouse 中执行很长的一句 SQL 的时候，默认无法换行，否则就会报错。 但是如果在启动命令行的时候，加上一个参数 -m ，便可以轻松的解决这个问题 启动的时候，在后面加一个 -m即可 clickhouse-client --host=localhost --port=9900 -m ...
h2调用mysql数据_H2数据库安装与使用(单点)
weixin_30851655的博客
02-15 389
H2数据库是一款 Java 嵌入式数据库，之前在搭建分布式定时任务(Spring+Quartz)时，需要选择一款关系型数据库作为任务协调。H2数据库够轻量(就一个 Jar 包)，使用起来还不错。下面是具体安装使用的一些备忘：安装部署H2数据库版本： 1.4.1961、下载与解压[root@localhost ~]# wget http://www.h2database.com/h2-2017-06...
单元测试之h2数据库的快速使用!(附实例)
weixin_44841849的博客
07-26 1588
假设你现在是有一个能够链接数据库的项目,而你现在想用h2数据库去单元测试,那么下面几个步骤是你应该做的!(另外不想跟着步骤做的直接去我github上把实例clone下来自己看也行) 1.pom 加入h2依赖 <dependency> <groupId>com.h2database</groupId> <artifactId>h2</artifactId> <versio
Java Statement接口：数据库连接与SQL操作指南
学习JDBC时，首先要理解JDBC的概念，它是一个Java API，允许Java应用程序与各种数据库系统进行连接和数据操作。数据库连接的四个关键步骤包括：建立连接、发送SQL命令、接收和处理结果，以及最终关闭连接。SQL...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值