java生成db文件

谁借莪１个温暖的怀抱￠ 2022-02-27 02:42 217阅读 0赞

[https://www.cnblogs.com/kanyun/p/9298163.html][https_www.cnblogs.com_kanyun_p_9298163.html]

工作中遇到一个需求，就是需要把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 + ";";
                System.out.println(retSQL);
                PreparedStatement prep = connection.prepareStatement(retSQL);
    //            设置自动提交
                connection.setAutoCommit(true);
                count = prep.executeUpdate();
            } catch (Exception e) {
                logger.error("插入失败：" + e);
                e.printStackTrace();
            }
            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

[https_www.cnblogs.com_kanyun_p_9298163.html]: https://www.cnblogs.com/kanyun/p/9298163.html