admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

add Db.txWithResult method; close #I7DJBD

Browse Source
This commit is contained in:
开源海哥 2023-06-15 10:00:12 +08:00
parent ee0e3f7465
commit ddb239176a
3 changed files with 193 additions and 165 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

311
mybatis-flex-core/src/main/java/com/mybatisflex/core/row/Db.java
View File

@ -41,7 +41,8 @@ import java.util.function.Supplier;
*/ */
public class Db { public class Db {
private Db() {} private Db() {
}
private static final Map<String, RowMapperInvoker> INVOKER_MAP = new ConcurrentHashMap<>(); private static final Map<String, RowMapperInvoker> INVOKER_MAP = new ConcurrentHashMap<>();
static RowMapperInvoker defaultRowMapperInvoker; static RowMapperInvoker defaultRowMapperInvoker;
@ -70,9 +71,10 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param row 数据 * @param row 数据
*/ */
public static int insert(String schema,String tableName, Row row) { public static int insert(String schema, String tableName, Row row) {
return invoker().insert(schema,tableName, row); return invoker().insert(schema, tableName, row);
} }
/** /**
* 往 tableName 插入一条 row 数据 * 往 tableName 插入一条 row 数据
* *
@ -80,7 +82,7 @@ public class Db {
* @param row 数据 * @param row 数据
*/ */
public static int insert(String tableName, Row row) { public static int insert(String tableName, Row row) {
return invoker().insert(null,tableName, row); return invoker().insert(null, tableName, row);
} }
@ -102,8 +104,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param rows 数据 * @param rows 数据
*/ */
public static int[] insertBatch(String schema,String tableName, Collection<Row> rows) { public static int[] insertBatch(String schema, String tableName, Collection<Row> rows) {
return insertBatch(schema,tableName, rows, rows.size()); return insertBatch(schema, tableName, rows, rows.size());
} }
/** /**
@ -113,7 +115,7 @@ public class Db {
* @param rows 数据 * @param rows 数据
*/ */
public static int[] insertBatch(String tableName, Collection<Row> rows) { public static int[] insertBatch(String tableName, Collection<Row> rows) {
return insertBatch(null,tableName, rows, rows.size()); return insertBatch(null, tableName, rows, rows.size());
} }
/** /**
@ -124,13 +126,14 @@ public class Db {
* @param rows 数据 * @param rows 数据
* @param batchSize 每次提交的数据量 * @param batchSize 每次提交的数据量
*/ */
public static int[] insertBatch(String schema,String tableName, Collection<Row> rows, int batchSize) { public static int[] insertBatch(String schema, String tableName, Collection<Row> rows, int batchSize) {
List<Row> list = CollectionUtil.toList(rows); List<Row> list = CollectionUtil.toList(rows);
return executeBatch(rows.size(), batchSize, RowMapper.class, (mapper, index) -> { return executeBatch(rows.size(), batchSize, RowMapper.class, (mapper, index) -> {
Row row = list.get(index); Row row = list.get(index);
mapper.insert(schema,tableName, row); mapper.insert(schema, tableName, row);
}); });
} }
/** /**
* 批量插入数据 * 批量插入数据
* *
@ -142,29 +145,29 @@ public class Db {
List<Row> list = CollectionUtil.toList(rows); List<Row> list = CollectionUtil.toList(rows);
return executeBatch(rows.size(), batchSize, RowMapper.class, (mapper, index) -> { return executeBatch(rows.size(), batchSize, RowMapper.class, (mapper, index) -> {
Row row = list.get(index); Row row = list.get(index);
mapper.insert(null,tableName, row); mapper.insert(null, tableName, row);
}); });
} }
/** /**
* 批量插入数据，根据第一条内容来构建插入的字段，效率比 {@link #insertBatch(String ,String, Collection, int)} 高 * 批量插入数据，根据第一条内容来构建插入的字段，效率比 {@link #insertBatch(String, String, Collection, int)} 高
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param rows 数据 * @param rows 数据
*/ */
public static int insertBatchWithFirstRowColumns(String schema,String tableName, List<Row> rows) { public static int insertBatchWithFirstRowColumns(String schema, String tableName, List<Row> rows) {
return invoker().insertBatchWithFirstRowColumns(schema,tableName, rows); return invoker().insertBatchWithFirstRowColumns(schema, tableName, rows);
} }
/** /**
* 批量插入数据，根据第一条内容来构建插入的字段，效率比 {@link #insertBatch(String ,String, Collection, int)} 高 * 批量插入数据，根据第一条内容来构建插入的字段，效率比 {@link #insertBatch(String, String, Collection, int)} 高
* *
* @param tableName 表名 * @param tableName 表名
* @param rows 数据 * @param rows 数据
*/ */
public static int insertBatchWithFirstRowColumns(String tableName, List<Row> rows) { public static int insertBatchWithFirstRowColumns(String tableName, List<Row> rows) {
return invoker().insertBatchWithFirstRowColumns(null,tableName, rows); return invoker().insertBatchWithFirstRowColumns(null, tableName, rows);
} }
/** /**
@ -184,8 +187,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param row 主键 和 id值 * @param row 主键 和 id值
*/ */
public static int deleteById(String schema,String tableName, Row row) { public static int deleteById(String schema, String tableName, Row row) {
return invoker().deleteById(schema,tableName, row); return invoker().deleteById(schema, tableName, row);
} }
/** /**
@ -195,20 +198,20 @@ public class Db {
* @param row 主键 和 id值 * @param row 主键 和 id值
*/ */
public static int deleteById(String tableName, Row row) { public static int deleteById(String tableName, Row row) {
return invoker().deleteById(null,tableName, row); return invoker().deleteById(null, tableName, row);
} }
/** /**
* 根据主键来删除 1 条数据 * 根据主键来删除 1 条数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param primaryKey 主键字段名称 * @param primaryKey 主键字段名称
* @param id 主键值 * @param id 主键值
*/ */
public static int deleteById(String schema,String tableName, String primaryKey, Object id) { public static int deleteById(String schema, String tableName, String primaryKey, Object id) {
return invoker().deleteById(schema,tableName, primaryKey, id); return invoker().deleteById(schema, tableName, primaryKey, id);
} }
/** /**
@ -219,20 +222,20 @@ public class Db {
* @param id 主键值 * @param id 主键值
*/ */
public static int deleteById(String tableName, String primaryKey, Object id) { public static int deleteById(String tableName, String primaryKey, Object id) {
return invoker().deleteById(null,tableName, primaryKey, id); return invoker().deleteById(null, tableName, primaryKey, id);
} }
/** /**
* 根据 id 集合来批量删除数据 * 根据 id 集合来批量删除数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param primaryKey 主键字段名称 * @param primaryKey 主键字段名称
* @param ids id 集合 * @param ids id 集合
*/ */
public static int deleteBatchByIds(String schema,String tableName, String primaryKey, Collection<?> ids) { public static int deleteBatchByIds(String schema, String tableName, String primaryKey, Collection<?> ids) {
return invoker().deleteBatchByIds(schema,tableName, primaryKey, ids); return invoker().deleteBatchByIds(schema, tableName, primaryKey, ids);
} }
/** /**
@ -243,18 +246,18 @@ public class Db {
* @param ids id 集合 * @param ids id 集合
*/ */
public static int deleteBatchByIds(String tableName, String primaryKey, Collection<?> ids) { public static int deleteBatchByIds(String tableName, String primaryKey, Collection<?> ids) {
return invoker().deleteBatchByIds(null,tableName, primaryKey, ids); return invoker().deleteBatchByIds(null, tableName, primaryKey, ids);
} }
/** /**
* 根据 map 构建的 where 条件来删除数据 * 根据 map 构建的 where 条件来删除数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static int deleteByMap(String schema,String tableName, Map<String, Object> whereColumns) { public static int deleteByMap(String schema, String tableName, Map<String, Object> whereColumns) {
return invoker().deleteByQuery(schema,tableName, new QueryWrapper().where(whereColumns)); return invoker().deleteByQuery(schema, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
@ -264,7 +267,7 @@ public class Db {
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static int deleteByMap(String tableName, Map<String, Object> whereColumns) { public static int deleteByMap(String tableName, Map<String, Object> whereColumns) {
return invoker().deleteByQuery(null,tableName, new QueryWrapper().where(whereColumns)); return invoker().deleteByQuery(null, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
@ -274,8 +277,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param condition 条件内容 * @param condition 条件内容
*/ */
public static int deleteByCondition(String schema,String tableName, QueryCondition condition) { public static int deleteByCondition(String schema, String tableName, QueryCondition condition) {
return invoker().deleteByQuery(schema,tableName, new QueryWrapper().where(condition)); return invoker().deleteByQuery(schema, tableName, new QueryWrapper().where(condition));
} }
/** /**
@ -285,19 +288,19 @@ public class Db {
* @param condition 条件内容 * @param condition 条件内容
*/ */
public static int deleteByCondition(String tableName, QueryCondition condition) { public static int deleteByCondition(String tableName, QueryCondition condition) {
return invoker().deleteByQuery(null,tableName, new QueryWrapper().where(condition)); return invoker().deleteByQuery(null, tableName, new QueryWrapper().where(condition));
} }
/** /**
* 根据 query 构建的条件来删除数据 * 根据 query 构建的条件来删除数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper query * @param queryWrapper query
*/ */
public static int deleteByQuery(String schema,String tableName, QueryWrapper queryWrapper) { public static int deleteByQuery(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().deleteByQuery(schema,tableName, queryWrapper); return invoker().deleteByQuery(schema, tableName, queryWrapper);
} }
/** /**
@ -307,7 +310,7 @@ public class Db {
* @param queryWrapper query * @param queryWrapper query
*/ */
public static int deleteByQuery(String tableName, QueryWrapper queryWrapper) { public static int deleteByQuery(String tableName, QueryWrapper queryWrapper) {
return invoker().deleteByQuery(null,tableName, queryWrapper); return invoker().deleteByQuery(null, tableName, queryWrapper);
} }
/** /**
@ -340,8 +343,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param row id 及其内容 * @param row id 及其内容
*/ */
public static int updateById(String schema,String tableName, Row row) { public static int updateById(String schema, String tableName, Row row) {
return invoker().updateById(schema,tableName, row); return invoker().updateById(schema, tableName, row);
} }
@ -352,19 +355,19 @@ public class Db {
* @param row id 及其内容 * @param row id 及其内容
*/ */
public static int updateById(String tableName, Row row) { public static int updateById(String tableName, Row row) {
return invoker().updateById(null,tableName, row); return invoker().updateById(null, tableName, row);
} }
/** /**
* 根据 map 构建的条件来更新数据 * 根据 map 构建的条件来更新数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param data 数据内容 * @param data 数据内容
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static int updateByMap(String schema,String tableName, Row data, Map<String, Object> whereColumns) { public static int updateByMap(String schema, String tableName, Row data, Map<String, Object> whereColumns) {
return invoker().updateByQuery(schema,tableName, data, new QueryWrapper().where(whereColumns)); return invoker().updateByQuery(schema, tableName, data, new QueryWrapper().where(whereColumns));
} }
@ -376,7 +379,7 @@ public class Db {
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static int updateByMap(String tableName, Row data, Map<String, Object> whereColumns) { public static int updateByMap(String tableName, Row data, Map<String, Object> whereColumns) {
return invoker().updateByQuery(null,tableName, data, new QueryWrapper().where(whereColumns)); return invoker().updateByQuery(null, tableName, data, new QueryWrapper().where(whereColumns));
} }
/** /**
@ -387,8 +390,8 @@ public class Db {
* @param data 数据 * @param data 数据
* @param condition 条件 * @param condition 条件
*/ */
public static int updateByCondition(String schema,String tableName, Row data, QueryCondition condition) { public static int updateByCondition(String schema, String tableName, Row data, QueryCondition condition) {
return invoker().updateByQuery(schema,tableName, data, new QueryWrapper().where(condition)); return invoker().updateByQuery(schema, tableName, data, new QueryWrapper().where(condition));
} }
/** /**
@ -399,20 +402,20 @@ public class Db {
* @param condition 条件 * @param condition 条件
*/ */
public static int updateByCondition(String tableName, Row data, QueryCondition condition) { public static int updateByCondition(String tableName, Row data, QueryCondition condition) {
return invoker().updateByQuery(null,tableName, data, new QueryWrapper().where(condition)); return invoker().updateByQuery(null, tableName, data, new QueryWrapper().where(condition));
} }
/** /**
* 根据 query 构建的条件来更新数据 * 根据 query 构建的条件来更新数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param data 数据内容 * @param data 数据内容
* @param queryWrapper queryWrapper 条件 * @param queryWrapper queryWrapper 条件
*/ */
public static int updateByQuery(String schema,String tableName, Row data, QueryWrapper queryWrapper) { public static int updateByQuery(String schema, String tableName, Row data, QueryWrapper queryWrapper) {
return invoker().updateByQuery(schema,tableName, data, queryWrapper); return invoker().updateByQuery(schema, tableName, data, queryWrapper);
} }
/** /**
@ -423,7 +426,7 @@ public class Db {
* @param queryWrapper queryWrapper 条件 * @param queryWrapper queryWrapper 条件
*/ */
public static int updateByQuery(String tableName, Row data, QueryWrapper queryWrapper) { public static int updateByQuery(String tableName, Row data, QueryWrapper queryWrapper) {
return invoker().updateByQuery(null,tableName, data, queryWrapper); return invoker().updateByQuery(null, tableName, data, queryWrapper);
} }
@ -434,8 +437,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param rows 还有主键的数据 * @param rows 还有主键的数据
*/ */
public static int updateBatchById(String schema,String tableName, List<Row> rows) { public static int updateBatchById(String schema, String tableName, List<Row> rows) {
return invoker().updateBatchById(schema,tableName, rows); return invoker().updateBatchById(schema, tableName, rows);
} }
/** /**
@ -445,7 +448,7 @@ public class Db {
* @param rows 还有主键的数据 * @param rows 还有主键的数据
*/ */
public static int updateBatchById(String tableName, List<Row> rows) { public static int updateBatchById(String tableName, List<Row> rows) {
return invoker().updateBatchById(null,tableName, rows); return invoker().updateBatchById(null, tableName, rows);
} }
@ -479,16 +482,15 @@ public class Db {
/** /**
* 通过 update schema.table set field = field + 1 where ... 的这种方向更新数据库某个字段内容 * 通过 update schema.table set field = field + 1 where ... 的这种方向更新数据库某个字段内容
* *
* * @param schema 模式
* @param schema 模式 * @param tableName 表名
* @param tableName 表名 * @param fieldName 字段名
* @param fieldName 字段名 * @param value 递增值
* @param value 递增值 * @param queryWrapper 条件
* @param queryWrapper 条件 * @return 受影响行数
* @return 受影响行数
*/ */
public static int updateNumberAddByQuery(String schema,String tableName, String fieldName, Number value, QueryWrapper queryWrapper){ public static int updateNumberAddByQuery(String schema, String tableName, String fieldName, Number value, QueryWrapper queryWrapper) {
return invoker().updateNumberAddByQuery(schema,tableName, fieldName, value, queryWrapper); return invoker().updateNumberAddByQuery(schema, tableName, fieldName, value, queryWrapper);
} }
/** /**
@ -500,8 +502,8 @@ public class Db {
* @param queryWrapper * @param queryWrapper
* @return * @return
*/ */
public static int updateNumberAddByQuery(String tableName, String fieldName, Number value, QueryWrapper queryWrapper){ public static int updateNumberAddByQuery(String tableName, String fieldName, Number value, QueryWrapper queryWrapper) {
return invoker().updateNumberAddByQuery(null,tableName, fieldName, value, queryWrapper); return invoker().updateNumberAddByQuery(null, tableName, fieldName, value, queryWrapper);
} }
@ -537,8 +539,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param row 主键和 id 值 * @param row 主键和 id 值
*/ */
public static Row selectOneById(String schema,String tableName, Row row) { public static Row selectOneById(String schema, String tableName, Row row) {
return invoker().selectOneById(schema,tableName, row); return invoker().selectOneById(schema, tableName, row);
} }
/** /**
@ -548,20 +550,20 @@ public class Db {
* @param row 主键和 id 值 * @param row 主键和 id 值
*/ */
public static Row selectOneById(String tableName, Row row) { public static Row selectOneById(String tableName, Row row) {
return invoker().selectOneById(null,tableName, row); return invoker().selectOneById(null, tableName, row);
} }
/** /**
* 根据主键来查询 1 条数据 * 根据主键来查询 1 条数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param primaryKey 主键字段名称 * @param primaryKey 主键字段名称
* @param id 主键值 * @param id 主键值
*/ */
public static Row selectOneById(String schema,String tableName, String primaryKey, Object id) { public static Row selectOneById(String schema, String tableName, String primaryKey, Object id) {
return invoker().selectOneById(schema,tableName, primaryKey, id); return invoker().selectOneById(schema, tableName, primaryKey, id);
} }
/** /**
@ -572,22 +574,21 @@ public class Db {
* @param id 主键值 * @param id 主键值
*/ */
public static Row selectOneById(String tableName, String primaryKey, Object id) { public static Row selectOneById(String tableName, String primaryKey, Object id) {
return invoker().selectOneById(null,tableName, primaryKey, id); return invoker().selectOneById(null, tableName, primaryKey, id);
} }
/** /**
* 根据 map 来查询 1 条数据 * 根据 map 来查询 1 条数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param whereColumns where条件 * @param whereColumns where条件
*/ */
public static Row selectOneByMap(String schema,String tableName, Map whereColumns) { public static Row selectOneByMap(String schema, String tableName, Map whereColumns) {
return invoker().selectOneByQuery(schema,tableName, new QueryWrapper().where(whereColumns)); return invoker().selectOneByQuery(schema, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
* 根据 map 来查询 1 条数据 * 根据 map 来查询 1 条数据
* *
@ -595,7 +596,7 @@ public class Db {
* @param whereColumns where条件 * @param whereColumns where条件
*/ */
public static Row selectOneByMap(String tableName, Map whereColumns) { public static Row selectOneByMap(String tableName, Map whereColumns) {
return invoker().selectOneByQuery(null,tableName, new QueryWrapper().where(whereColumns)); return invoker().selectOneByQuery(null, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
@ -605,8 +606,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param condition 条件 * @param condition 条件
*/ */
public static Row selectOneByCondition(String schema,String tableName, QueryCondition condition) { public static Row selectOneByCondition(String schema, String tableName, QueryCondition condition) {
return invoker().selectOneByQuery(schema,tableName, new QueryWrapper().where(condition)); return invoker().selectOneByQuery(schema, tableName, new QueryWrapper().where(condition));
} }
/** /**
@ -616,19 +617,19 @@ public class Db {
* @param condition 条件 * @param condition 条件
*/ */
public static Row selectOneByCondition(String tableName, QueryCondition condition) { public static Row selectOneByCondition(String tableName, QueryCondition condition) {
return invoker().selectOneByQuery(null,tableName, new QueryWrapper().where(condition)); return invoker().selectOneByQuery(null, tableName, new QueryWrapper().where(condition));
} }
/** /**
* 根据 queryWrapper 来查询 1 条数据 * 根据 queryWrapper 来查询 1 条数据
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper queryWrapper * @param queryWrapper queryWrapper
*/ */
public static Row selectOneByQuery(String schema,String tableName, QueryWrapper queryWrapper) { public static Row selectOneByQuery(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().selectOneByQuery(schema,tableName, queryWrapper); return invoker().selectOneByQuery(schema, tableName, queryWrapper);
} }
/** /**
@ -638,7 +639,7 @@ public class Db {
* @param queryWrapper queryWrapper * @param queryWrapper queryWrapper
*/ */
public static Row selectOneByQuery(String tableName, QueryWrapper queryWrapper) { public static Row selectOneByQuery(String tableName, QueryWrapper queryWrapper) {
return invoker().selectOneByQuery(null,tableName, queryWrapper); return invoker().selectOneByQuery(null, tableName, queryWrapper);
} }
@ -652,7 +653,7 @@ public class Db {
if (queryTables == null || queryTables.isEmpty()) { if (queryTables == null || queryTables.isEmpty()) {
throw FlexExceptions.wrap("table must not be null or empty in Db.selectOneByQuery"); throw FlexExceptions.wrap("table must not be null or empty in Db.selectOneByQuery");
} }
return invoker().selectOneByQuery(null,null, queryWrapper); return invoker().selectOneByQuery(null, null, queryWrapper);
} }
@ -674,8 +675,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static List<Row> selectListByMap(String schema,String tableName, Map<String, Object> whereColumns) { public static List<Row> selectListByMap(String schema, String tableName, Map<String, Object> whereColumns) {
return invoker().selectListByQuery(schema,tableName, new QueryWrapper().where(whereColumns)); return invoker().selectListByQuery(schema, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
@ -685,20 +686,20 @@ public class Db {
* @param whereColumns where 条件 * @param whereColumns where 条件
*/ */
public static List<Row> selectListByMap(String tableName, Map<String, Object> whereColumns) { public static List<Row> selectListByMap(String tableName, Map<String, Object> whereColumns) {
return invoker().selectListByQuery(null,tableName, new QueryWrapper().where(whereColumns)); return invoker().selectListByQuery(null, tableName, new QueryWrapper().where(whereColumns));
} }
/** /**
* 根据 map 构建的条件来查询数据列表 * 根据 map 构建的条件来查询数据列表
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param whereColumns 条件 * @param whereColumns 条件
* @param count 数据量 * @param count 数据量
*/ */
public static List<Row> selectListByMap(String schema,String tableName, Map<String, Object> whereColumns, int count) { public static List<Row> selectListByMap(String schema, String tableName, Map<String, Object> whereColumns, int count) {
return invoker().selectListByQuery(schema,tableName, new QueryWrapper().where(whereColumns).limit(count)); return invoker().selectListByQuery(schema, tableName, new QueryWrapper().where(whereColumns).limit(count));
} }
/** /**
@ -709,7 +710,7 @@ public class Db {
* @param count 数据量 * @param count 数据量
*/ */
public static List<Row> selectListByMap(String tableName, Map<String, Object> whereColumns, int count) { public static List<Row> selectListByMap(String tableName, Map<String, Object> whereColumns, int count) {
return invoker().selectListByQuery(null,tableName, new QueryWrapper().where(whereColumns).limit(count)); return invoker().selectListByQuery(null, tableName, new QueryWrapper().where(whereColumns).limit(count));
} }
@ -720,8 +721,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param condition where 条件 * @param condition where 条件
*/ */
public static List<Row> selectListByCondition(String schema,String tableName, QueryCondition condition) { public static List<Row> selectListByCondition(String schema, String tableName, QueryCondition condition) {
return invoker().selectListByQuery(schema,tableName, new QueryWrapper().where(condition)); return invoker().selectListByQuery(schema, tableName, new QueryWrapper().where(condition));
} }
@ -732,8 +733,9 @@ public class Db {
* @param condition where 条件 * @param condition where 条件
*/ */
public static List<Row> selectListByCondition(String tableName, QueryCondition condition) { public static List<Row> selectListByCondition(String tableName, QueryCondition condition) {
return invoker().selectListByQuery(null,tableName, new QueryWrapper().where(condition)); return invoker().selectListByQuery(null, tableName, new QueryWrapper().where(condition));
} }
/** /**
* 根据 condition 条件来查询数据列表 * 根据 condition 条件来查询数据列表
* *
@ -742,8 +744,8 @@ public class Db {
* @param condition 条件 * @param condition 条件
* @param count 数据量 * @param count 数据量
*/ */
public static List<Row> selectListByCondition(String schema,String tableName, QueryCondition condition, int count) { public static List<Row> selectListByCondition(String schema, String tableName, QueryCondition condition, int count) {
return invoker().selectListByQuery(schema,tableName, new QueryWrapper().where(condition).limit(count)); return invoker().selectListByQuery(schema, tableName, new QueryWrapper().where(condition).limit(count));
} }
/** /**
@ -754,19 +756,19 @@ public class Db {
* @param count 数据量 * @param count 数据量
*/ */
public static List<Row> selectListByCondition(String tableName, QueryCondition condition, int count) { public static List<Row> selectListByCondition(String tableName, QueryCondition condition, int count) {
return invoker().selectListByQuery(null,tableName, new QueryWrapper().where(condition).limit(count)); return invoker().selectListByQuery(null, tableName, new QueryWrapper().where(condition).limit(count));
} }
/** /**
* 通过 query 来查询数据列表 * 通过 query 来查询数据列表
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper query 条件 * @param queryWrapper query 条件
*/ */
public static List<Row> selectListByQuery(String schema,String tableName, QueryWrapper queryWrapper) { public static List<Row> selectListByQuery(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().selectListByQuery(schema,tableName, queryWrapper); return invoker().selectListByQuery(schema, tableName, queryWrapper);
} }
@ -777,7 +779,7 @@ public class Db {
* @param queryWrapper query 条件 * @param queryWrapper query 条件
*/ */
public static List<Row> selectListByQuery(String tableName, QueryWrapper queryWrapper) { public static List<Row> selectListByQuery(String tableName, QueryWrapper queryWrapper) {
return invoker().selectListByQuery(null,tableName, queryWrapper); return invoker().selectListByQuery(null, tableName, queryWrapper);
} }
@ -791,7 +793,7 @@ public class Db {
if (queryTables == null || queryTables.isEmpty()) { if (queryTables == null || queryTables.isEmpty()) {
throw FlexExceptions.wrap("table must not be null or empty in Db.selectListByQuery"); throw FlexExceptions.wrap("table must not be null or empty in Db.selectListByQuery");
} }
return invoker().selectListByQuery(null,null, queryWrapper); return invoker().selectListByQuery(null, null, queryWrapper);
} }
/** /**
@ -800,8 +802,8 @@ public class Db {
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
*/ */
public static List<Row> selectAll(String schema,String tableName) { public static List<Row> selectAll(String schema, String tableName) {
return invoker().selectAll(schema,tableName); return invoker().selectAll(schema, tableName);
} }
/** /**
@ -810,7 +812,7 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
*/ */
public static List<Row> selectAll(String tableName) { public static List<Row> selectAll(String tableName) {
return invoker().selectAll(null,tableName); return invoker().selectAll(null, tableName);
} }
/** /**
@ -827,13 +829,13 @@ public class Db {
/** /**
* 根据 queryWrapper 查询内容，数据返回的应该只有 1 行 1 列 * 根据 queryWrapper 查询内容，数据返回的应该只有 1 行 1 列
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper query 封装 * @param queryWrapper query 封装
* @return 数据内容 * @return 数据内容
*/ */
public static Object selectObject(String schema,String tableName, QueryWrapper queryWrapper) { public static Object selectObject(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().selectObjectByQuery(schema,tableName, queryWrapper); return invoker().selectObjectByQuery(schema, tableName, queryWrapper);
} }
/** /**
@ -844,7 +846,7 @@ public class Db {
* @return 数据内容 * @return 数据内容
*/ */
public static Object selectObject(String tableName, QueryWrapper queryWrapper) { public static Object selectObject(String tableName, QueryWrapper queryWrapper) {
return invoker().selectObjectByQuery(null,tableName, queryWrapper); return invoker().selectObjectByQuery(null, tableName, queryWrapper);
} }
@ -859,7 +861,6 @@ public class Db {
} }
/** /**
* 查询某列内容，数据返回应该有 多行 1 列 * 查询某列内容，数据返回应该有 多行 1 列
* *
@ -874,13 +875,13 @@ public class Db {
/** /**
* 根据 queryWrapper 查询内容，数据返回的应该只有 1 行 1 列 * 根据 queryWrapper 查询内容，数据返回的应该只有 1 行 1 列
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper query 封装 * @param queryWrapper query 封装
* @return 数据内容 * @return 数据内容
*/ */
public static Object selectObjectList(String schema,String tableName, QueryWrapper queryWrapper) { public static Object selectObjectList(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().selectObjectListByQuery(schema,tableName, queryWrapper); return invoker().selectObjectListByQuery(schema, tableName, queryWrapper);
} }
/** /**
@ -891,7 +892,7 @@ public class Db {
* @return 数据内容 * @return 数据内容
*/ */
public static Object selectObjectList(String tableName, QueryWrapper queryWrapper) { public static Object selectObjectList(String tableName, QueryWrapper queryWrapper) {
return invoker().selectObjectListByQuery(null,tableName, queryWrapper); return invoker().selectObjectListByQuery(null, tableName, queryWrapper);
} }
@ -925,8 +926,8 @@ public class Db {
* @param tableName 表名 * @param tableName 表名
* @param condition 条件 * @param condition 条件
*/ */
public static long selectCountByCondition(String schema,String tableName, QueryCondition condition) { public static long selectCountByCondition(String schema, String tableName, QueryCondition condition) {
return invoker().selectCountByQuery(schema,tableName, new QueryWrapper().where(condition)); return invoker().selectCountByQuery(schema, tableName, new QueryWrapper().where(condition));
} }
/** /**
@ -936,19 +937,19 @@ public class Db {
* @param condition 条件 * @param condition 条件
*/ */
public static long selectCountByCondition(String tableName, QueryCondition condition) { public static long selectCountByCondition(String tableName, QueryCondition condition) {
return invoker().selectCountByQuery(null,tableName, new QueryWrapper().where(condition)); return invoker().selectCountByQuery(null, tableName, new QueryWrapper().where(condition));
} }
/** /**
* 根据 query 构建的条件来查询数据量 * 根据 query 构建的条件来查询数据量
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param queryWrapper query 条件 * @param queryWrapper query 条件
*/ */
public static long selectCountByQuery(String schema,String tableName, QueryWrapper queryWrapper) { public static long selectCountByQuery(String schema, String tableName, QueryWrapper queryWrapper) {
return invoker().selectCountByQuery(schema,tableName, queryWrapper); return invoker().selectCountByQuery(schema, tableName, queryWrapper);
} }
/** /**
@ -958,7 +959,7 @@ public class Db {
* @param queryWrapper query 条件 * @param queryWrapper query 条件
*/ */
public static long selectCountByQuery(String tableName, QueryWrapper queryWrapper) { public static long selectCountByQuery(String tableName, QueryWrapper queryWrapper) {
return invoker().selectCountByQuery(null,tableName, queryWrapper); return invoker().selectCountByQuery(null, tableName, queryWrapper);
} }
@ -973,21 +974,21 @@ public class Db {
if (queryTables == null || queryTables.isEmpty()) { if (queryTables == null || queryTables.isEmpty()) {
throw FlexExceptions.wrap("Query tables must not be null or empty in Db.selectCountByQuery"); throw FlexExceptions.wrap("Query tables must not be null or empty in Db.selectCountByQuery");
} }
return invoker().selectCountByQuery(null,null, queryWrapper); return invoker().selectCountByQuery(null, null, queryWrapper);
} }
/** /**
* 分页查询 * 分页查询
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param pageNumber 当前的页码 * @param pageNumber 当前的页码
* @param pageSize 每页的数据量 * @param pageSize 每页的数据量
* @param condition 条件 * @param condition 条件
*/ */
public static Page<Row> paginate(String schema,String tableName, int pageNumber, int pageSize, QueryCondition condition) { public static Page<Row> paginate(String schema, String tableName, int pageNumber, int pageSize, QueryCondition condition) {
return invoker().paginate(schema,tableName, new Page<>(pageNumber, pageSize), QueryWrapper.create().where(condition)); return invoker().paginate(schema, tableName, new Page<>(pageNumber, pageSize), QueryWrapper.create().where(condition));
} }
@ -1000,22 +1001,22 @@ public class Db {
* @param condition 条件 * @param condition 条件
*/ */
public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, QueryCondition condition) { public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, QueryCondition condition) {
return invoker().paginate(null,tableName, new Page<>(pageNumber, pageSize), QueryWrapper.create().where(condition)); return invoker().paginate(null, tableName, new Page<>(pageNumber, pageSize), QueryWrapper.create().where(condition));
} }
/** /**
* 分页查询 * 分页查询
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param pageNumber 当前的页码 * @param pageNumber 当前的页码
* @param pageSize 每页的数据量 * @param pageSize 每页的数据量
* @param totalRow 数据总量 * @param totalRow 数据总量
* @param condition 条件 * @param condition 条件
*/ */
public static Page<Row> paginate(String schema,String tableName, int pageNumber, int pageSize, int totalRow, QueryCondition condition) { public static Page<Row> paginate(String schema, String tableName, int pageNumber, int pageSize, int totalRow, QueryCondition condition) {
return invoker().paginate(schema,tableName, new Page<>(pageNumber, pageSize, totalRow), QueryWrapper.create().where(condition)); return invoker().paginate(schema, tableName, new Page<>(pageNumber, pageSize, totalRow), QueryWrapper.create().where(condition));
} }
/** /**
@ -1028,21 +1029,21 @@ public class Db {
* @param condition 条件 * @param condition 条件
*/ */
public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, int totalRow, QueryCondition condition) { public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, int totalRow, QueryCondition condition) {
return invoker().paginate(null,tableName, new Page<>(pageNumber, pageSize, totalRow), QueryWrapper.create().where(condition)); return invoker().paginate(null, tableName, new Page<>(pageNumber, pageSize, totalRow), QueryWrapper.create().where(condition));
} }
/** /**
* 分页查询 * 分页查询
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param pageNumber 当前的页码 * @param pageNumber 当前的页码
* @param pageSize 每页的数据量 * @param pageSize 每页的数据量
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String schema,String tableName, int pageNumber, int pageSize, QueryWrapper queryWrapper) { public static Page<Row> paginate(String schema, String tableName, int pageNumber, int pageSize, QueryWrapper queryWrapper) {
return invoker().paginate(schema,tableName, new Page<>(pageNumber, pageSize), queryWrapper); return invoker().paginate(schema, tableName, new Page<>(pageNumber, pageSize), queryWrapper);
} }
/** /**
@ -1054,22 +1055,22 @@ public class Db {
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, QueryWrapper queryWrapper) { public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, QueryWrapper queryWrapper) {
return invoker().paginate(null,tableName, new Page<>(pageNumber, pageSize), queryWrapper); return invoker().paginate(null, tableName, new Page<>(pageNumber, pageSize), queryWrapper);
} }
/** /**
* 分页查询 * 分页查询
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param pageNumber 当前的页码 * @param pageNumber 当前的页码
* @param pageSize 每页的数据量 * @param pageSize 每页的数据量
* @param totalRow 数据总量 * @param totalRow 数据总量
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String schema,String tableName, int pageNumber, int pageSize, int totalRow, QueryWrapper queryWrapper) { public static Page<Row> paginate(String schema, String tableName, int pageNumber, int pageSize, int totalRow, QueryWrapper queryWrapper) {
return invoker().paginate(schema,tableName, new Page<>(pageNumber, pageSize, totalRow), queryWrapper); return invoker().paginate(schema, tableName, new Page<>(pageNumber, pageSize, totalRow), queryWrapper);
} }
/** /**
@ -1082,20 +1083,20 @@ public class Db {
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, int totalRow, QueryWrapper queryWrapper) { public static Page<Row> paginate(String tableName, int pageNumber, int pageSize, int totalRow, QueryWrapper queryWrapper) {
return invoker().paginate(null,tableName, new Page<>(pageNumber, pageSize, totalRow), queryWrapper); return invoker().paginate(null, tableName, new Page<>(pageNumber, pageSize, totalRow), queryWrapper);
} }
/** /**
* 分页查询 * 分页查询
* *
* @param schema 模式 * @param schema 模式
* @param tableName 表名 * @param tableName 表名
* @param page page 对象，若 page 有 totalCount 值，则不会再去查询分类的数据总量 * @param page page 对象，若 page 有 totalCount 值，则不会再去查询分类的数据总量
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String schema,String tableName, Page<Row> page, QueryWrapper queryWrapper) { public static Page<Row> paginate(String schema, String tableName, Page<Row> page, QueryWrapper queryWrapper) {
return invoker().paginate(schema,tableName, page, queryWrapper); return invoker().paginate(schema, tableName, page, queryWrapper);
} }
/** /**
@ -1106,22 +1107,36 @@ public class Db {
* @param queryWrapper 条件 * @param queryWrapper 条件
*/ */
public static Page<Row> paginate(String tableName, Page<Row> page, QueryWrapper queryWrapper) { public static Page<Row> paginate(String tableName, Page<Row> page, QueryWrapper queryWrapper) {
return invoker().paginate(null,tableName, page, queryWrapper); return invoker().paginate(null, tableName, page, queryWrapper);
} }
/** /**
* 进行事务操作 * 进行事务操作，返回 null 或者 false 或者 抛出异常，事务回滚
*
* @param supplier
*/ */
public static boolean tx(Supplier<Boolean> supplier) { public static boolean tx(Supplier<Boolean> supplier) {
return tx(supplier, Propagation.REQUIRED); return tx(supplier, Propagation.REQUIRED);
} }
/**
* 进行事务操作，返回 null 或者 false 或者 抛出异常， 事务回滚
*/
public static boolean tx(Supplier<Boolean> supplier, Propagation propagation) { public static boolean tx(Supplier<Boolean> supplier, Propagation propagation) {
Boolean result = TransactionalManager.exec(supplier, propagation); Boolean result = TransactionalManager.exec(supplier, propagation, false);
return result != null && result; return result != null && result;
} }
/**
* 进行事务操作，和返回结果无关，只有抛出异常时，事务回滚
*/
public static <T> T txWithResult(Supplier<T> supplier) {
return txWithResult(supplier, Propagation.REQUIRED);
}
/**
* 进行事务操作，和返回结果无关，只有抛出异常时，事务回滚
*/
public static <T> T txWithResult(Supplier<T> supplier, Propagation propagation) {
return TransactionalManager.exec(supplier, propagation, true);
}
} }

37
mybatis-flex-core/src/main/java/com/mybatisflex/core/transaction/TransactionalManager.java
View File

@ -30,7 +30,8 @@ import java.util.function.Supplier;
*/ */
public class TransactionalManager { public class TransactionalManager {
private TransactionalManager() {} private TransactionalManager() {
}
private static final Log log = LogFactory.getLog(TransactionalManager.class); private static final Log log = LogFactory.getLog(TransactionalManager.class);
@ -55,7 +56,7 @@ public class TransactionalManager {
} }
public static Boolean exec(Supplier<Boolean> supplier, Propagation propagation) { public static <T> T exec(Supplier<T> supplier, Propagation propagation, boolean withResult) {
//上一级事务的id，支持事务嵌套 //上一级事务的id，支持事务嵌套
String currentXID = TransactionContext.getXID(); String currentXID = TransactionContext.getXID();
try { try {
@ -65,7 +66,7 @@ public class TransactionalManager {
if (currentXID != null) { if (currentXID != null) {
return supplier.get(); return supplier.get();
} else { } else {
return execNewTransactional(supplier); return execNewTransactional(supplier, withResult);
} }
@ -85,7 +86,7 @@ public class TransactionalManager {
//始终以新事务的方式运行，若存在当前事务，则暂停（挂起）当前事务。 //始终以新事务的方式运行，若存在当前事务，则暂停（挂起）当前事务。
case REQUIRES_NEW: case REQUIRES_NEW:
return execNewTransactional(supplier); return execNewTransactional(supplier, withResult);
//以非事务的方式运行，若存在当前事务，则暂停（挂起）当前事务。 //以非事务的方式运行，若存在当前事务，则暂停（挂起）当前事务。
@ -118,24 +119,32 @@ public class TransactionalManager {
} }
} }
private static Boolean execNewTransactional(Supplier<Boolean> supplier) { private static <T> T execNewTransactional(Supplier<T> supplier, boolean withResult) {
String xid = startTransactional(); String xid = startTransactional();
Boolean success = false; T result = null;
boolean rollbacked = false; boolean isRollback = false;
try { try {
success = supplier.get(); result = supplier.get();
} catch (Throwable e) { } catch (Throwable e) {
rollbacked = true; isRollback = true;
rollback(xid); rollback(xid);
throw new TransactionException(e.getMessage(), e); throw new TransactionException(e.getMessage(), e);
} finally { } finally {
if (success != null && success) { if (!isRollback) {
commit(xid); if (!withResult) {
} else if (!rollbacked) { if (result instanceof Boolean && (Boolean) result) {
rollback(xid); commit(xid);
}
//null or false
else {
rollback(xid);
}
} else {
commit(xid);
}
} }
} }
return success; return result;
} }

10
mybatis-flex-test/mybatis-flex-spring-boot-test/src/test/java/com/mybatisflex/test/common/ReflectTest.java
View File

@ -23,6 +23,7 @@ import org.junit.jupiter.api.Test;
import java.lang.reflect.Field; import java.lang.reflect.Field;
import java.lang.reflect.Type; import java.lang.reflect.Type;
import java.util.List;
/** /**
* @author 王帅 * @author 王帅
@ -32,9 +33,12 @@ class ReflectTest {
@Test @Test
void test() { void test() {
Field field = ClassUtil.getAllFields(Account.class, f -> f.getName().equals("list")).get(0); List<Field> allFields = ClassUtil.getAllFields(Account.class);
Type type = TypeParameterResolver.resolveFieldType(field, Account.class); for (Field field : allFields) {
System.out.println(type); Type type = TypeParameterResolver.resolveFieldType(field, Account.class);
System.out.println("field: " + field+"----->Type:" + type);
}
} }
} }