修复搜索引擎收录不全

This commit is contained in:
2026-07-19 01:22:11 +08:00
parent c6ba83f5d7
commit 5c08ed607a
4 changed files with 26 additions and 10 deletions
@@ -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);
}
}