From 71b8f56e1e119362cee885903dcc6e1aac17665c Mon Sep 17 00:00:00 2001 From: 502647092 Date: Tue, 27 Sep 2016 15:13:11 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E5=BA=93=E5=9F=BA=E6=9C=AC=E6=93=8D=E4=BD=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 502647092 --- .../java/pw/yumc/YumCore/sql/DataBase.java | 503 ++++++++++++++++++ 1 file changed, 503 insertions(+) create mode 100644 src/main/java/pw/yumc/YumCore/sql/DataBase.java diff --git a/src/main/java/pw/yumc/YumCore/sql/DataBase.java b/src/main/java/pw/yumc/YumCore/sql/DataBase.java new file mode 100644 index 0000000..133ce69 --- /dev/null +++ b/src/main/java/pw/yumc/YumCore/sql/DataBase.java @@ -0,0 +1,503 @@ +package pw.yumc.YumCore.sql; + +import java.sql.Connection; +import java.sql.DatabaseMetaData; +import java.sql.PreparedStatement; +import java.sql.ResultSet; +import java.sql.SQLException; +import java.sql.Statement; +import java.util.ArrayList; +import java.util.Collection; +import java.util.LinkedList; +import java.util.List; + +import org.bukkit.configuration.ConfigurationSection; +import org.bukkit.plugin.Plugin; + +import pw.yumc.YumCore.bukkit.Log; +import pw.yumc.YumCore.sql.core.DataBaseCore; +import pw.yumc.YumCore.sql.core.KeyValue; +import pw.yumc.YumCore.sql.core.MySQLCore; +import pw.yumc.YumCore.sql.core.SQLiteCore; + +/** + * 数据库管理类 + * + * @since 2015年12月14日 下午1:26:06 + * @author 喵♂呜 + * + */ +public class DataBase { + private final DataBaseCore dataBaseCore; + + /** + * 初始化数据库管理 + * + * @param core + * 数据库核心 + */ + public DataBase(final DataBaseCore core) { + this.dataBaseCore = core; + } + + public static DataBase create(final Plugin plugin, final ConfigurationSection dbConfig) { + final ConfigurationSection cfg = dbConfig.getConfigurationSection("MySQL"); + if (dbConfig.getString("FileSystem").equalsIgnoreCase("MySQL")) { + plugin.getLogger().info("已启用MySQL保存数据,开始连接数据库..."); + return new DataBase(new MySQLCore(cfg)); + } + return new DataBase(new SQLiteCore(plugin, cfg)); + } + + /** + * 关闭数据库连接 + * + * @return 是否关闭成功 + */ + public boolean close() { + try { + this.dataBaseCore.getConnection().close(); + return true; + } catch (final SQLException e) { + Log.debug("数据库链接关闭失败!", e); + return false; + } + } + + /** + * 复制当前数据核心的数据库到指定的数据库 + * 此方法将不会删除当前数据库原有数据 + * 此方法可能花费较长的时间 + * + * 注意: 当前方法将不会创建表在新的数据库内 需要自行创建数据表 + * + * @param db + * 接受数据的数据库核心 + * @return 是否转换成功 + */ + public boolean copyTo(final DataBaseCore db) { + try { + final String src = this.dataBaseCore.getConnection().getMetaData().getURL(); + final String des = db.getConnection().getMetaData().getURL(); + Log.info("开始从源 " + src + " 复制数据到 " + des + " ..."); + ResultSet rs = this.dataBaseCore.getConnection().getMetaData().getTables(null, null, "%", null); + final List tables = new LinkedList<>(); + while (rs.next()) { + tables.add(rs.getString("TABLE_NAME")); + } + info("源数据库中有 " + tables.size() + " 张数据表 ..."); + rs.close(); + int s = 0; + final long start = System.currentTimeMillis(); + for (final String table : tables) { + Log.info("开始复制源数据库中的表 " + table + " ..."); + if (table.toLowerCase().startsWith("sqlite_autoindex_")) { + continue; + } + Log.info("清空目标数据库中的表 " + table + " ..."); + db.execute("DELETE FROM " + table); + rs = this.dataBaseCore.query("SELECT * FROM " + table); + int n = 0; + String query = "INSERT INTO " + table + " VALUES ("; + for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) { + query += "?, "; + } + query = query.substring(0, query.length() - 2) + ")"; + + final Connection con = db.getConnection(); + try { + con.setAutoCommit(false); + final PreparedStatement ps = con.prepareStatement(query); + long time = System.currentTimeMillis(); + while (rs.next()) { + n++; + for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) { + ps.setObject(i, rs.getObject(i)); + } + ps.addBatch(); + if (n % 100 == 0) { + try { + ps.executeBatch(); + con.commit(); + } catch (final SQLException e) { + info("#===================================================="); + info("#数据复制区段(不是ID!) " + (n - 100) + "-" + n + " 出现错误..."); + info("#错误信息如下: "); + e.printStackTrace(); + info("#===================================================="); + } + } + if (System.currentTimeMillis() - time > 500) { + info("已复制 " + n + " 条记录..."); + time = System.currentTimeMillis(); + } + } + s += n; + ps.executeBatch(); + con.commit(); + info("数据表 " + table + " 复制完成 共 " + n + " 条记录..."); + } catch (final Exception e) { + e.printStackTrace(); + } finally { + con.setAutoCommit(true); + } + rs.close(); + } + info("成功从 " + src + " 复制 " + s + " 条数据到 " + des + " 耗时 " + (System.currentTimeMillis() - start) / 1000 + " 秒..."); + db.getConnection().close(); + this.dataBaseCore.getConnection().close(); + return true; + } catch (final SQLException e) { + return false; + } + } + + /** + * 创建数据表 + * + * @param tableName + * 表名 + * @param fields + * 字段参数 + * @param Conditions + * -附加值 + * @return 运行结果 + */ + public boolean createTables(final String tableName, final KeyValue fields, final String Conditions) { + try { + this.dataBaseCore.createTables(tableName, fields, Conditions); + return isTableExists(tableName); + } catch (final Exception e) { + sqlerr("创建数据表 " + tableName + " 异常(内部方法)...", e); + return false; + } + } + + /** + * 对数据库表中的记录进行删除操作 + * + * @param tableName + * 表名 + * @param fields + * 删除条件 + * @return 受到影响的数据条目 + */ + public int dbDelete(final String tableName, final KeyValue fields) { + final String sql = "DELETE FROM `" + tableName + "` WHERE " + fields.toWhereString(); + try { + return this.dataBaseCore.update(sql); + } catch (final Exception e) { + sqlerr(sql, e); + return 0; + } + } + + /** + * 判断数据库某个值是否存在! + * + * @param tableName + * 数据库表名 + * @param fields + * 选择条件 + * @return 首个符合条件的结果 + */ + public boolean dbExist(final String tableName, final KeyValue fields) { + final String sql = "SELECT * FROM " + tableName + " WHERE " + fields.toWhereString(); + try { + return this.dataBaseCore.query(sql).next(); + } catch (final Exception e) { + sqlerr(sql, e); + return false; + } + } + + /** + * 对数据库表进行插入操作 + * + * @param tabName + * 表名 + * @param fields + * 带键值的 + * @return 受到影响的数据条目 + */ + public int dbInsert(final String tabName, final KeyValue fields) { + final String sql = "INSERT INTO `" + tabName + "` " + fields.toInsertString(); + try { + return this.dataBaseCore.update(sql); + } catch (final Exception e) { + sqlerr(sql, e); + return 0; + } + + } + + // @SuppressWarnings("unchecked") + // public List dbSelect(final Class> model, final KeyValue selCondition) { + // final List modellist = new ArrayList<>(); + // final String sql = "SELECT " + toKeys(model) + " FROM `" + model.getAnnotation(Entity.class).name() + "`" + (selCondition == null ? "" : " WHERE " + selCondition.toWhereString()); + // try { + // final ResultSet dbresult = this.dataBaseCore.execute(sql); + // while (dbresult.next()) { + // final M m = (M) model.newInstance(); + // final Field[] fields = model.getDeclaredFields(); + // for (final Field col : fields) { + // col.set(m, dbresult.getObject(col.getName())); + // } + // modellist.add(m); + // } + // } catch (final InstantiationException e) { + // info("模型类实例化失败!"); + // e.printStackTrace(); + // } catch (final Exception e) { + // sqlerr(sql, e); + // } + // return modellist; + // } + + /** + * 对数据库表进行选择操作！ + * + * @param tableName + * 数据库表名 + * @param fields + * 读取的字段 + * @param selCondition + * 选择条件 + * @return 一个含有KeyValue的List（列表） + */ + public List dbSelect(final String tableName, final KeyValue fields, final KeyValue selCondition) { + final String sql = "SELECT " + fields.toKeys() + " FROM `" + tableName + "`" + (selCondition == null ? "" : " WHERE " + selCondition.toWhereString()); + final List kvlist = new ArrayList<>(); + try { + final ResultSet dbresult = this.dataBaseCore.query(sql); + while (dbresult.next()) { + final KeyValue kv = new KeyValue(); + for (final String col : fields.getKeys()) { + kv.add(col, dbresult.getString(col.toString())); + } + kvlist.add(kv); + } + } catch (final Exception e) { + sqlerr(sql, e); + } + return kvlist; + } + + /** + * 对数据库表进行选择操作！ + * + * @param tableName + * 数据库表名 + * @param selCondition + * 选择条件 + * @param fields + * 读取的字段 + * @return 一个含有KeyValue的List（列表） + */ + public List dbSelect(final String tableName, final KeyValue selCondition, final String... fields) { + final String sql = "SELECT " + getKeys(fields) + " FROM `" + tableName + "`" + (selCondition == null ? "" : " WHERE " + selCondition.toWhereString()); + final List kvlist = new ArrayList<>(); + try { + final ResultSet dbresult = this.dataBaseCore.query(sql); + while (dbresult.next()) { + final KeyValue kv = new KeyValue(); + for (final String col : fields) { + kv.add(col, dbresult.getString(col.toString())); + } + kvlist.add(kv); + } + } catch (final Exception e) { + sqlerr(sql, e); + } + return kvlist; + } + + /** + * 对数据库表进行选择操作！ + * + * @param tableName + * 数据库表名 + * @param fields + * 字段名 + * @param selConditions + * 选择条件 + * @return 首个符合条件的结果 + */ + public String dbSelectFirst(final String tableName, final String fields, final KeyValue selConditions) { + final String sql = "SELECT " + fields + " FROM " + tableName + " WHERE " + selConditions.toWhereString() + " limit 1"; + try { + final ResultSet dbresult = this.dataBaseCore.query(sql); + if (dbresult.next()) { + return dbresult.getString(fields); + } + } catch (final Exception e) { + sqlerr(sql, e); + } + return null; + } + + /** + * 对数据库表中记录进行更新操作 + * + * @param tabName + * 表名 + * @param fields + * 字段参数 + * @param upCondition + * 更新条件 + * @return 受到影响的数据条目 + */ + public int dbUpdate(final String tabName, final KeyValue fields, final KeyValue upCondition) { + final String sql = "UPDATE `" + tabName + "` SET " + fields.toUpdateString() + " WHERE " + upCondition.toWhereString(); + try { + return this.dataBaseCore.update(sql); + } catch (final Exception e) { + sqlerr(sql, e); + return 0; + } + } + + /** + * 获得当前使用的数据库核心 + * + * @return 数据库核心 + */ + public DataBaseCore getDataBaseCore() { + return this.dataBaseCore; + } + + /** + * 字段数组转字符串 + * + * @param fields + * 字段数组 + * @return 字段字符串 + */ + public String getKeys(final String... fields) { + final StringBuilder sb = new StringBuilder(); + for (final String string : fields) { + sb.append("`"); + sb.append(string); + sb.append("`, "); + } + return sb.toString().substring(0, sb.length() - 2); + } + + public boolean isFieldExists(final String tableName, final KeyValue fields) { + final DatabaseMetaData dbm; + final ResultSet tables; + try { + dbm = this.dataBaseCore.getConnection().getMetaData(); + tables = dbm.getTables(null, null, tableName, null); + if (tables.next()) { + final ResultSet f = dbm.getColumns(null, null, tableName, fields.getKeys()[0]); + return f.next(); + } + } catch (final SQLException e) { + sqlerr("判断 表名:" + tableName + " 字段名:" + fields.getKeys()[0] + " 是否存在时出错!", e); + } + return false; + } + + /** + * 判断数据表是否存在 + * + * @param tableName + * 表名 + * @return 是否存在 + */ + public boolean isTableExists(final String tableName) { + try { + final DatabaseMetaData dbm = this.dataBaseCore.getConnection().getMetaData(); + final ResultSet tables = dbm.getTables(null, null, tableName, null); + return tables.next(); + } catch (final SQLException e) { + sqlerr("判断 表名:" + tableName + " 是否存在时出错!", e); + return false; + } + } + + /** + * 批量执行SQL语句 + * + * @param sqls + * SQL语句列表 + */ + public void runSqlList(final Collection sqls) { + final Connection con = getDataBaseCore().getConnection(); + final long start = System.currentTimeMillis(); + try { + long time = System.currentTimeMillis(); + con.setAutoCommit(false); + final Statement st = con.createStatement(); + int i = 0; + for (final String sql : sqls) { + st.addBatch(sql); + i++; + if (i % 100 == 0) { + st.executeBatch(); + con.commit(); + if (System.currentTimeMillis() - time > 500) { + info("已执行 " + i + " 条语句..."); + time = System.currentTimeMillis(); + } + } + } + st.executeBatch(); + con.commit(); + info("执行SQL完毕 总计: " + sqls.size() + " 条 耗时: " + start + "ms!"); + } catch (final SQLException e) { + try { + con.rollback(); + sqlerr("执行SQL数组发生错误 数据已回滚...", e); + } catch (final SQLException e1) { + sqlerr("执行SQL数组发生错误 警告! 数据回滚失败...", e1); + } + } finally { + try { + con.setAutoCommit(true); + } catch (final SQLException e) { + } + } + } + + /** + * 输出SQL错误 + * + * @param sql + * SQL语句 + * @param e + * 错误异常 + */ + public void sqlerr(final String sql, final Exception e) { + info("数据库操作出错: " + e.getMessage()); + info("SQL查询语句: " + sql); + Log.debug(this.getClass().getName()); + Log.debug(e); + } + + /** + * 测试数据库连接 + * + * @return 是否连接成功 + */ + public boolean testConnect() { + return this.dataBaseCore.getConnection() != null; + } + + private void info(final String info) { + Log.info(info); + } + + // private String toKeys(final Class> model) { + // final Field[] fields = model.getDeclaredFields(); + // final StringBuilder sb = new StringBuilder(); + // for (final Field next : fields) { + // sb.append("`"); + // sb.append(next.getName()); + // sb.append("`, "); + // } + // return sb.toString().substring(0, sb.length() - 2); + // } + +}