修复搜索引擎收录不全
This commit is contained in:
@@ -9,6 +9,4 @@ package run.halo.moments;
|
||||
public enum ModelConst {
|
||||
;
|
||||
public static final String TEMPLATE_ID = "_templateId";
|
||||
public static final Integer DEFAULT_PAGE_SIZE = 10;
|
||||
public static final Integer SEARCH_DEFAULT_PAGE_SIZE = 200;
|
||||
}
|
||||
|
||||
@@ -1,17 +1,18 @@
|
||||
package run.halo.moments.search;
|
||||
|
||||
import static run.halo.moments.ModelConst.SEARCH_DEFAULT_PAGE_SIZE;
|
||||
import static run.halo.app.extension.index.query.QueryFactory.equal;
|
||||
import static run.halo.app.extension.index.query.QueryFactory.isNull;
|
||||
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import org.springframework.data.domain.Sort;
|
||||
import org.springframework.stereotype.Component;
|
||||
import reactor.core.publisher.Flux;
|
||||
import reactor.core.publisher.Mono;
|
||||
import run.halo.app.extension.ListOptions;
|
||||
import run.halo.app.extension.ListResult;
|
||||
import run.halo.app.extension.PageRequest;
|
||||
import run.halo.app.extension.PageRequestImpl;
|
||||
import run.halo.app.extension.ReactiveExtensionClient;
|
||||
import run.halo.app.extension.index.query.QueryFactory;
|
||||
import run.halo.app.extension.router.selector.FieldSelector;
|
||||
import run.halo.app.search.HaloDocument;
|
||||
import run.halo.app.search.HaloDocumentsProvider;
|
||||
@@ -26,6 +27,8 @@ public class MomentHaloDocumentsProvider implements HaloDocumentsProvider {
|
||||
|
||||
public static final String MOMENT_DOCUMENT_TYPE = "moment.moment.halo.run";
|
||||
|
||||
private static final int PAGE_SIZE = 200;
|
||||
|
||||
private final ReactiveExtensionClient client;
|
||||
|
||||
private final DocumentConverter converter;
|
||||
@@ -33,14 +36,16 @@ public class MomentHaloDocumentsProvider implements HaloDocumentsProvider {
|
||||
@Override
|
||||
public Flux<HaloDocument> fetchAll() {
|
||||
var options = new ListOptions();
|
||||
var notDeleted = QueryFactory.isNull("metadata.deletionTimestamp");
|
||||
var approved = QueryFactory.equal("spec.approved", "true");
|
||||
var notDeleted = isNull("metadata.deletionTimestamp");
|
||||
var approved = equal("spec.approved", Boolean.TRUE.toString());
|
||||
options.setFieldSelector(FieldSelector.of(notDeleted).andQuery(approved));
|
||||
var pageRequest = createPageRequest();
|
||||
// make sure the moments are approved and not deleted.
|
||||
return client.listBy(Moment.class, options, pageRequest)
|
||||
.map(ListResult::getItems)
|
||||
.flatMapMany(Flux::fromIterable)
|
||||
.expand(result -> result.hasNext()
|
||||
? client.listBy(Moment.class, options, nextPage(result, pageRequest.getSort()))
|
||||
: Mono.empty())
|
||||
.flatMap(result -> Flux.fromIterable(result.getItems()))
|
||||
.flatMap(converter::convert);
|
||||
}
|
||||
|
||||
@@ -50,7 +55,11 @@ public class MomentHaloDocumentsProvider implements HaloDocumentsProvider {
|
||||
}
|
||||
|
||||
private PageRequest createPageRequest() {
|
||||
return PageRequestImpl.of(1, SEARCH_DEFAULT_PAGE_SIZE,
|
||||
return PageRequestImpl.of(1, PAGE_SIZE,
|
||||
Sort.by("metadata.creationTimestamp", "metadata.name"));
|
||||
}
|
||||
|
||||
private static PageRequest nextPage(ListResult<Moment> result, Sort sort) {
|
||||
return PageRequestImpl.of(result.getPage() + 1, result.getSize(), sort);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user