package it.cavallium.dbengine.client; import it.cavallium.dbengine.client.query.QueryUtils; import it.cavallium.dbengine.client.query.current.data.QueryParams; import it.cavallium.dbengine.client.query.current.data.ScoreMode; import it.cavallium.dbengine.client.query.current.data.ScoreSort; import it.cavallium.dbengine.database.LLDocument; import it.cavallium.dbengine.database.LLItem; import it.cavallium.dbengine.database.LLLuceneIndex; import it.cavallium.dbengine.database.LLSignal; import it.cavallium.dbengine.database.LLTerm; import it.cavallium.dbengine.database.disk.LLLocalDatabaseConnection; import it.cavallium.dbengine.lucene.LuceneUtils; import it.cavallium.dbengine.lucene.analyzer.TextFieldsAnalyzer; import it.cavallium.dbengine.lucene.analyzer.TextFieldsSimilarity; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; import java.time.Duration; import java.util.Comparator; import java.util.StringJoiner; import java.util.concurrent.CompletionException; import org.apache.lucene.document.Field.Store; import reactor.core.publisher.Mono; import reactor.core.scheduler.Schedulers; public class IndicizationExample { public static void main(String[] args) { tempIndex(true) .flatMap(index -> index .addDocument(new LLTerm("id", "123"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "123", Store.YES), LLItem.newTextField("name", "Mario", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) ) .then(index.refresh()) .then(index.search(null, QueryParams .builder() .query(QueryUtils.exactSearch(TextFieldsAnalyzer.N4GramPartialString, "name", "Mario")) .limit(1) .sort(ScoreSort.of()) .scoreMode(ScoreMode.of(false, true)) .build(), "id" )) .flatMap(results -> Mono.from(results .results() .flatMap(r -> r) .doOnNext(signal -> { if (signal.isValue()) { System.out.println("Value: " + signal.getValue()); } }) .filter(LLSignal::isTotalHitsCount)) ) .doOnNext(count -> System.out.println("Total hits: " + count)) .doOnTerminate(() -> System.out.println("Completed")) .then(index.close()) ) .subscribeOn(Schedulers.parallel()) .block(); tempIndex(true) .flatMap(index -> index .addDocument(new LLTerm("id", "126"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "126", Store.YES), LLItem.newTextField("name", "Marioxq", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) ) .then(index .addDocument(new LLTerm("id", "123"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "123", Store.YES), LLItem.newTextField("name", "Mario", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) )) .then(index .addDocument(new LLTerm("id", "124"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "124", Store.YES), LLItem.newTextField("name", "Mariossi", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) )) .then(index .addDocument(new LLTerm("id", "125"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "125", Store.YES), LLItem.newTextField("name", "Mario marios", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) )) .then(index .addDocument(new LLTerm("id", "128"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "128", Store.YES), LLItem.newTextField("name", "Marion", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) )) .then(index .addDocument(new LLTerm("id", "127"), new LLDocument(new LLItem[]{ LLItem.newStringField("id", "127", Store.YES), LLItem.newTextField("name", "Mariotto", Store.NO), LLItem.newStringField("surname", "Rossi", Store.NO) }) )) .then(index.refresh()) .then(index.search(null, QueryParams .builder() .query(QueryUtils.exactSearch(TextFieldsAnalyzer.N4GramPartialString, "name", "Mario")) .limit(10) .sort(MultiSort.topScore().getQuerySort()) .scoreMode(ScoreMode.of(false, true)) .build(), "id" )) .flatMap(results -> LuceneUtils.mergeSignalStreamRaw(results .results(), MultiSort.topScoreRaw(), 10L) .doOnNext(value -> System.out.println("Value: " + value)) .then(Mono.from(results .results() .flatMap(part -> part) .filter(LLSignal::isTotalHitsCount) .map(LLSignal::getTotalHitsCount))) ) .doOnNext(count -> System.out.println("Total hits: " + count)) .doOnTerminate(() -> System.out.println("Completed")) .then(index.close()) ) .subscribeOn(Schedulers.parallel()) .block(); } public static final class CurrentCustomType { private final int number; public CurrentCustomType(int number) { this.number = number; } public int getNumber() { return number; } @Override public String toString() { return new StringJoiner(", ", CurrentCustomType.class.getSimpleName() + "[", "]") .add("number=" + number) .toString(); } } private static Mono tempIndex(boolean delete) { var wrkspcPath = Path.of("/tmp/tempdb/"); return Mono .fromCallable(() -> { if (delete && Files.exists(wrkspcPath)) { Files.walk(wrkspcPath).sorted(Comparator.reverseOrder()).forEach(file -> { try { Files.delete(file); } catch (IOException ex) { throw new CompletionException(ex); } }); } Files.createDirectories(wrkspcPath); return null; }) .subscribeOn(Schedulers.boundedElastic()) .then(new LLLocalDatabaseConnection(wrkspcPath, true).connect()) .flatMap(conn -> conn.getLuceneIndex("testindices", 10, TextFieldsAnalyzer.N4GramPartialString, TextFieldsSimilarity.NGramBM25Plus, Duration.ofSeconds(5), Duration.ofSeconds(5), false )); } }