67 lines
2.2 KiB
Java
67 lines
2.2 KiB
Java
package it.cavallium.dbengine.tests;
|
|
|
|
import com.google.common.primitives.Ints;
|
|
import com.google.common.primitives.Longs;
|
|
import it.cavallium.dbengine.client.Indicizer;
|
|
import it.cavallium.dbengine.database.LLUpdateDocument;
|
|
import it.cavallium.dbengine.database.LLItem;
|
|
import it.cavallium.dbengine.database.LLTerm;
|
|
import it.cavallium.dbengine.lucene.analyzer.TextFieldsAnalyzer;
|
|
import it.cavallium.dbengine.lucene.analyzer.TextFieldsSimilarity;
|
|
import it.cavallium.dbengine.rpc.current.data.IndicizerAnalyzers;
|
|
import it.cavallium.dbengine.rpc.current.data.IndicizerSimilarities;
|
|
import java.util.LinkedList;
|
|
import org.apache.lucene.document.Field;
|
|
import org.apache.lucene.document.Field.Store;
|
|
import org.apache.lucene.index.IndexableField;
|
|
import org.apache.lucene.util.BytesRef;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
public class StringIndicizer extends Indicizer<String, String> {
|
|
|
|
@Override
|
|
public @NotNull LLUpdateDocument toIndexRequest(@NotNull String key, @NotNull String value) {
|
|
var fields = new LinkedList<LLItem>();
|
|
fields.add(LLItem.newStringField("uid", key, Field.Store.YES));
|
|
fields.add(LLItem.newTextField("text", value, Store.NO));
|
|
@SuppressWarnings("UnstableApiUsage")
|
|
var numInt = Ints.tryParse(value);
|
|
if (numInt != null) {
|
|
fields.add(LLItem.newIntPoint("intpoint", numInt));
|
|
fields.add(LLItem.newNumericDocValuesField("intsort", numInt));
|
|
}
|
|
@SuppressWarnings("UnstableApiUsage")
|
|
var numLong = Longs.tryParse(value);
|
|
if (numLong != null) {
|
|
fields.add(LLItem.newLongPoint("longpoint", numLong));
|
|
fields.add(LLItem.newNumericDocValuesField("longsort", numLong));
|
|
}
|
|
return new LLUpdateDocument(fields);
|
|
}
|
|
|
|
@Override
|
|
public @NotNull LLTerm toIndex(@NotNull String key) {
|
|
return new LLTerm("uid", key);
|
|
}
|
|
|
|
@Override
|
|
public @NotNull String getKeyFieldName() {
|
|
return "uid";
|
|
}
|
|
|
|
@Override
|
|
public @NotNull String getKey(IndexableField key) {
|
|
return key.stringValue();
|
|
}
|
|
|
|
@Override
|
|
public IndicizerAnalyzers getPerFieldAnalyzer() {
|
|
return it.cavallium.dbengine.client.IndicizerAnalyzers.of(TextFieldsAnalyzer.ICUCollationKey);
|
|
}
|
|
|
|
@Override
|
|
public IndicizerSimilarities getPerFieldSimilarity() {
|
|
return it.cavallium.dbengine.client.IndicizerSimilarities.of(TextFieldsSimilarity.BM25Standard);
|
|
}
|
|
}
|