57 lines
2.0 KiB
Java
57 lines
2.0 KiB
Java
package it.cavallium.dbengine.client;
|
|
|
|
import com.google.common.collect.Multimap;
|
|
import com.google.common.collect.Multimaps;
|
|
import it.cavallium.dbengine.database.LLIndexRequest;
|
|
import it.cavallium.dbengine.database.LLSoftUpdateDocument;
|
|
import it.cavallium.dbengine.database.LLUpdateDocument;
|
|
import it.cavallium.dbengine.database.LLTerm;
|
|
import it.cavallium.dbengine.database.LLUpdateFields;
|
|
import it.cavallium.dbengine.database.LLUtils;
|
|
import it.cavallium.dbengine.rpc.current.data.IndicizerAnalyzers;
|
|
import it.cavallium.dbengine.rpc.current.data.IndicizerSimilarities;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import org.apache.lucene.index.IndexableField;
|
|
import org.apache.lucene.util.BytesRef;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import reactor.core.publisher.Flux;
|
|
import reactor.core.publisher.Mono;
|
|
import reactor.util.function.Tuple2;
|
|
|
|
public abstract class Indicizer<T, U> {
|
|
|
|
/**
|
|
* Transform a value to an IndexRequest.
|
|
*/
|
|
public abstract @NotNull Mono<? extends LLIndexRequest> toIndexRequest(@NotNull T key, @NotNull U value);
|
|
|
|
public final @NotNull Mono<LLUpdateDocument> toDocument(@NotNull T key, @NotNull U value) {
|
|
return toIndexRequest(key, value).map(req -> {
|
|
if (req instanceof LLUpdateFields updateFields) {
|
|
return new LLUpdateDocument(updateFields.items());
|
|
} else if (req instanceof LLUpdateDocument updateDocument) {
|
|
return updateDocument;
|
|
} else if (req instanceof LLSoftUpdateDocument softUpdateDocument) {
|
|
return new LLUpdateDocument(softUpdateDocument.items());
|
|
} else {
|
|
throw new UnsupportedOperationException("Unexpected request type: " + req);
|
|
}
|
|
});
|
|
}
|
|
|
|
public abstract @NotNull LLTerm toIndex(@NotNull T key);
|
|
|
|
public abstract @NotNull String getKeyFieldName();
|
|
|
|
public abstract @NotNull T getKey(IndexableField key);
|
|
|
|
public abstract IndicizerAnalyzers getPerFieldAnalyzer();
|
|
|
|
public abstract IndicizerSimilarities getPerFieldSimilarity();
|
|
|
|
public Multimap<String, String> getMoreLikeThisDocumentFields(T key, U value) {
|
|
return Multimaps.forMap(Map.of());
|
|
}
|
|
}
|