java生成db文件
工作中遇到一个需求，就是需要把mysql里的一些表数据生成db文件，给客户端使用，客户端使用sqlite数据库；
首先我们需要在项目中添加Sqlite JDBC 依赖
<dependency> <groupId>org.xerial</groupId> <artifactId>sqlite-jdbc</artifactId> <version>3.23.1</version> </dependency>
生成DB文件工具类
package cn.kayun.util; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.lang.reflect.Field; import java.sql.Connection; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.Statement; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.List; import java.util.Locale; /** * Created by root on 2018/7/2 0002. */ public class DBUtil { private static final Logger logger = LoggerFactory.getLogger(DBUtil.class); private static Connection connection = null; private static Statement statement = null; private static String username = "kanyun"; private static String password = "kanyun"; /** * @describe: 设置连接 * @params: * @Author: Kanyun * @Date: 2018/7/12 9:54 */ public static void setConnection() { try { // 声明驱动类型 Class.forName("org.sqlite.JDBC"); // 设置sqlite db文件存放基本目录 String path = DBUtil.class.getClassLoader().getResource("").getPath(); // 设置 sqlite文件路径，等同于mysql连接地址(jdbc:mysql://127.0.0.1:3306/test) String url = "jdbc:sqlite:" + path + "data.db"; // 获取连接 connection = DriverManager.getConnection(url, username, password); // 声明 statement = connection.createStatement(); } catch (Exception e) { throw new RuntimeException("建立Sqlite连接失败"); } } /** * @describe: 创建表 * @params: tableName: 要创建的表的名称 className：项目中Pojo的类名(需要注意的是该类名需要加上包名 如 com.xxx.xxx.pojo.xxx) * @Author: Kanyun * @Date: 2018/7/12 9:56 */ public synchronized static void create(String tableName, String className) { try { statement.executeUpdate("DROP TABLE IF EXISTS " + tableName + ";"); // 通过反射，获取传入类名的字段信息 Field[] fields = Class.forName(className).getDeclaredFields(); StringBuffer sb = new StringBuffer(); String reg = " "; sb.append("create table " + tableName + " ("); for (Field field : fields) { // 设置使用反射时,可以访问私有变量,当pojo的变量设置为private修饰时,isAccessible()得到的值是false，必须要改成true才可以访问 field.setAccessible(true); // 通过获得的字段,获取字段类型,和字段名,来定义创建表的字段名和字段类型 if (field.getType().getName().equals(java.lang.Long.class.getName())) { sb.append(reg + field.getName() + " bigint(20) "); } else if (field.getType().getName().equals(java.lang.String.class.getName())) { sb.append(reg + field.getName() + " varchar(255) "); } else if (field.getType().getName().equals(java.util.Date.class.getName())) { sb.append(reg + field.getName() + " datetime "); } else { sb.append(reg + field.getName() + " int(11) "); } reg = ","; } sb.append(") ;"); statement.executeUpdate(sb.toString()); } catch (Exception e) { logger.error("建表失败：" + e); throw new RuntimeException("建表失败，表名称：" + tableName); } } /** * @describe: 表中插入数据 * @params: tableName：表名 list:待插入的对象集合 需要注意的是插入的对象要跟表名对应 * @Author: Kanyun * @Date: 2018/7/12 10:03 */ public synchronized static <T> int insert(String tableName, List<T> list) { StringBuffer declaration = new StringBuffer(); List<String> dataSqlArray = new ArrayList<>(); int count = 0; try { list.stream().map(t -> { if (dataSqlArray.size() < 1) { logger.info("定义语句"); Field[] fields = t.getClass().getDeclaredFields(); declaration.append(" ("); String reg = ""; for (Field field : fields) { field.setAccessible(true); declaration.append(reg + field.getName()); reg = ","; } declaration.append(")"); } return t; }).forEach((t) -> { Field[] fields = t.getClass().getDeclaredFields(); StringBuffer oneData = new StringBuffer(); String reg = ""; oneData.append(" ("); for (Field field : fields) { try { field.setAccessible(true); if (null == field.get(t) || "".equals(field.get(t))) { oneData.append(reg + null); } else if (field.getType().getName().equals(java.lang.String.class.getName())) { // 传入的对象中字段是字符串时,插入数据库需要转义单引号 oneData.append(reg + "\'" + field.get(t) + "\'"); // 传入的对象中字段是日期类型时,先将日期转换为 yyyy-MM-dd 的形式(我这里,传入对象的日期对象格式为：Wed Jul 11 14:34:07 CST 2018),在将其转换成字符串，注意转义单引号 } else if (field.getType().getName().equals(java.util.Date.class.getName())) { String dd = field.get(t).toString(); SimpleDateFormat sdf1 = new SimpleDateFormat("EEE MMM dd HH:mm:ss z yyyy", Locale.ENGLISH); SimpleDateFormat sdf2 = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); String date = sdf2.format(sdf1.parse(dd)); oneData.append(reg + "\'" + date + "\'"); } else { oneData.append(reg + field.get(t)); } reg = ","; } catch (Exception e) { e.printStackTrace(); } } oneData.append(")"); dataSqlArray.add(oneData.toString()); }); String dataSql = StringUtils.join(dataSqlArray, ","); String retSQL = "INSERT INTO " + tableName + declaration.toString() + " VALUES " + dataSql + ";"; logger.info("待插入SQL: {}",retSQL); PreparedStatement prep = connection.prepareStatement(retSQL); // 设置自动提交 connection.setAutoCommit(true); count = prep.executeUpdate(); } catch (Exception e) { logger.error("插入失败：" , e); } return count; } /** * @describe: 关闭链接 * @params: * @Author: Kanyun * @Date: 2018/7/12 10:11 */ public static void endConnection() { try { connection.close(); } catch (Exception e) { e.printStackTrace(); } } }
生成的DB文件在classess目录下.
需要注意的是，在创建链接的过程中,虽然可以传入用户名,密码，但是使用一些工具（比如Navicat）还是可以查看db文件的内容的，因为免费版的SQLite有一个致命缺点：不支持加密
如果需要使用sqlite加密，一是数据本身进行加密，也就是往表中插入的数据进行加密，二是可以加密db文件本身
更多加密方式，参考：https://www.cnblogs.com/ligun123/p/5206942.html
