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();
     }