fix #2835 Add interface for rewriting search request parameters before processing

This commit is contained in:
Shinsuke Sugaya 2024-07-26 14:40:28 +09:00
parent 09fb29ef57
commit 8c48ac1f8a

View file

@ -16,6 +16,7 @@
package org.codelibs.fess.helper;
import java.text.NumberFormat;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
@ -72,15 +73,24 @@ public class SearchHelper {
private static final Logger logger = LogManager.getLogger(SearchHelper.class);
// ===================================================================================
// Variable
//
protected SearchRequestParamsRewriter[] searchRequestParamsRewriters = new SearchRequestParamsRewriter[0];
// ===================================================================================
// Method
// ==============
public void search(final SearchRequestParams params, final SearchRenderData data, final OptionalThing<FessUserBean> userBean) {
public void search(final SearchRequestParams searchRequestParams, final SearchRenderData data,
final OptionalThing<FessUserBean> userBean) {
final SystemHelper systemHelper = ComponentUtil.getSystemHelper();
final long startTime = systemHelper.getCurrentTimeAsLong();
final long requestedTime = startTime;
final SearchRequestParams params = rewrite(searchRequestParams);
LaRequestUtil.getOptionalRequest().ifPresent(request -> {
request.setAttribute(Constants.REQUEST_LANGUAGES, params.getLanguages());
request.setAttribute(Constants.REQUEST_QUERIES, params.getQuery());
@ -354,4 +364,21 @@ public class SearchHelper {
throw new SearchEngineClientException("Failed to update bulk data.", e);
}
}
protected SearchRequestParams rewrite(final SearchRequestParams params) {
SearchRequestParams newParams = params;
for (final SearchRequestParamsRewriter rewriter : searchRequestParamsRewriters) {
newParams = rewriter.rewrite(newParams);
}
return newParams;
}
public void addRewriter(final SearchRequestParamsRewriter rewriter) {
searchRequestParamsRewriters = Arrays.copyOf(searchRequestParamsRewriters, searchRequestParamsRewriters.length + 1);
searchRequestParamsRewriters[searchRequestParamsRewriters.length - 1] = rewriter;
}
public interface SearchRequestParamsRewriter {
SearchRequestParams rewrite(SearchRequestParams params);
}
}