88 lines
2.6 KiB
Java
88 lines
2.6 KiB
Java
package it.cavallium.dbengine.database.indicizer;
|
|
|
|
import it.cavallium.dbengine.database.DatabaseMemoryMode;
|
|
import it.cavallium.dbengine.database.LLScoreMode;
|
|
import it.cavallium.dbengine.database.LLSearchResult;
|
|
import it.cavallium.dbengine.database.LLSort;
|
|
import it.cavallium.dbengine.database.collections.Joiner;
|
|
import it.cavallium.dbengine.database.collections.Joiner.ValueGetter;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import it.cavallium.dbengine.client.CompositeSnapshot;
|
|
import reactor.core.publisher.Mono;
|
|
|
|
public class JoinedIndicizerWriter<KEY, DBTYPE, JOINEDTYPE> implements LuceneIndicizerWriter<KEY, DBTYPE> {
|
|
|
|
private final LuceneIndicizerWriter<KEY, JOINEDTYPE> indicizerWriter;
|
|
private final Joiner<KEY, DBTYPE, JOINEDTYPE> joiner;
|
|
private final ValueGetter<KEY, DBTYPE> dbValueGetter;
|
|
|
|
public JoinedIndicizerWriter(LuceneIndicizerWriter<KEY, JOINEDTYPE> indicizerWriter,
|
|
Joiner<KEY, DBTYPE, JOINEDTYPE> joiner,
|
|
ValueGetter<KEY, DBTYPE> dbValueGetter) {
|
|
this.indicizerWriter = indicizerWriter;
|
|
this.joiner = joiner;
|
|
this.dbValueGetter = dbValueGetter;
|
|
}
|
|
|
|
@Override
|
|
public Mono<Void> add(KEY key, DBTYPE value) {
|
|
return joiner
|
|
.join(dbValueGetter, value)
|
|
.flatMap(joinedValue -> this.indicizerWriter.add(key, joinedValue));
|
|
}
|
|
|
|
@Override
|
|
public Mono<Void> remove(KEY key) {
|
|
return this.indicizerWriter.remove(key);
|
|
}
|
|
|
|
@Override
|
|
public Mono<Void> update(KEY key, DBTYPE value) {
|
|
return joiner
|
|
.join(dbValueGetter, value)
|
|
.flatMap(joinedValue -> this.indicizerWriter.update(key, joinedValue));
|
|
}
|
|
|
|
@Override
|
|
public Mono<Void> clearIndex() {
|
|
return this.indicizerWriter.clearIndex();
|
|
}
|
|
|
|
@Override
|
|
public Mono<LLSearchResult> moreLikeThis(@Nullable CompositeSnapshot snapshot, DBTYPE mltDocumentValue, int limit) {
|
|
return joiner
|
|
.join(dbValueGetter, mltDocumentValue)
|
|
.flatMap(val -> this.indicizerWriter.moreLikeThis(snapshot, val, limit));
|
|
}
|
|
|
|
@Override
|
|
public Mono<LLSearchResult> search(@Nullable CompositeSnapshot snapshot,
|
|
String query,
|
|
int limit,
|
|
@Nullable LLSort sort,
|
|
LLScoreMode scoreMode) {
|
|
return this.indicizerWriter
|
|
.search(snapshot, query, limit, sort, scoreMode);
|
|
}
|
|
|
|
@Override
|
|
public Mono<Long> count(@Nullable CompositeSnapshot snapshot, String query) {
|
|
return this.indicizerWriter.count(snapshot, query);
|
|
}
|
|
|
|
@Override
|
|
public Mono<Void> close() {
|
|
return this.indicizerWriter.close();
|
|
}
|
|
|
|
@Override
|
|
public Mono<KEY> getKey(String key) {
|
|
return this.indicizerWriter.getKey(key);
|
|
}
|
|
|
|
@Override
|
|
public DatabaseMemoryMode getMemoryMode() {
|
|
return this.indicizerWriter.getMemoryMode();
|
|
}
|
|
}
|