2 changed files with 1 additions and 69 deletions
@ -1,69 +0,0 @@
@@ -1,69 +0,0 @@
|
||||
/* |
||||
* Copyright 2021-2024 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
package org.springframework.data.r2dbc.repository.support; |
||||
|
||||
import reactor.core.publisher.Mono; |
||||
|
||||
import java.util.List; |
||||
|
||||
import org.springframework.data.domain.Page; |
||||
import org.springframework.data.domain.PageImpl; |
||||
import org.springframework.data.domain.Pageable; |
||||
import org.springframework.util.Assert; |
||||
|
||||
/** |
||||
* Support for query execution using {@link Pageable}. Using {@link ReactivePageableExecutionUtils} assumes that data |
||||
* queries are cheaper than {@code COUNT} queries and so some cases can take advantage of optimizations. |
||||
* |
||||
* @author Mark Paluch |
||||
* @since 1.4 |
||||
*/ |
||||
abstract class ReactivePageableExecutionUtils { |
||||
|
||||
private ReactivePageableExecutionUtils() {} |
||||
|
||||
/** |
||||
* Constructs a {@link Page} based on the given {@code content}, {@link Pageable} and {@link Mono} applying |
||||
* optimizations. The construction of {@link Page} omits a count query if the total can be determined based on the |
||||
* result size and {@link Pageable}. |
||||
* |
||||
* @param content must not be {@literal null}. |
||||
* @param pageable must not be {@literal null}. |
||||
* @param totalSupplier must not be {@literal null}. |
||||
* @return the {@link Page}. |
||||
*/ |
||||
public static <T> Mono<Page<T>> getPage(List<T> content, Pageable pageable, Mono<Long> totalSupplier) { |
||||
|
||||
Assert.notNull(content, "Content must not be null"); |
||||
Assert.notNull(pageable, "Pageable must not be null"); |
||||
Assert.notNull(totalSupplier, "TotalSupplier must not be null"); |
||||
|
||||
if (pageable.isUnpaged() || pageable.getOffset() == 0) { |
||||
|
||||
if (pageable.isUnpaged() || pageable.getPageSize() > content.size()) { |
||||
return Mono.just(new PageImpl<>(content, pageable, content.size())); |
||||
} |
||||
|
||||
return totalSupplier.map(total -> new PageImpl<>(content, pageable, total)); |
||||
} |
||||
|
||||
if (!content.isEmpty() && pageable.getPageSize() > content.size()) { |
||||
return Mono.just(new PageImpl<>(content, pageable, pageable.getOffset() + content.size())); |
||||
} |
||||
|
||||
return totalSupplier.map(total -> new PageImpl<>(content, pageable, total)); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue