fix: 修复forUpdate无效问题

hs-web · Aug 11, 2023 · 5c33e3e · 5c33e3e
1 parent cd92582
commit 5c33e3e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...rdb/src/main/java/org/hswebframework/ezorm/rdb/mapping/defaults/DefaultReactiveQuery.java b/...rdb/src/main/java/org/hswebframework/ezorm/rdb/mapping/defaults/DefaultReactiveQuery.java
@@ -52,6 +52,7 @@ public Flux<T> fetch() {
  .where(param.getTerms())
  .orderBy(getSortOrder())
  .when(param.isPaging(), query -> query.paging(param.getPageIndex(), param.getPageSize()))
+ .when(param.isForUpdate(), QueryOperator::forUpdate)
  .fetch(eventWrapper(tableMetadata, wrapper, executorType("reactive"), type("fetch")))
  .reactive())
  .contextWrite(ctx->ctx.put(Logger.class,logger));
@@ -67,9 +68,11 @@ public Mono<T> fetchOne() {
  .select(getSelectColumn())
  .where(param.getTerms())
  .orderBy(getSortOrder())
- .paging(0, 1)
+ //.paging(0, 1)
+ .when(param.isForUpdate(), QueryOperator::forUpdate)
  .fetch(eventWrapper(tableMetadata, wrapper, executorType("reactive"), type("fetchOne")))
- .reactive())
+ .reactive()
+ .take(1))
  .contextWrite(ctx-> ctx.put(Logger.class,logger))
  .singleOrEmpty();
  }