package it.cavallium.dbengine.database.collections;
import it.cavallium.dbengine.client.CompositeSnapshot;
import it.cavallium.dbengine.database.serialization.Serializer;
import java.util.Optional;
import java.util.function.Function;
import org.jetbrains.annotations.Nullable;
import reactor.core.publisher.Mono;
@SuppressWarnings("unused")
public class DatabaseSingleMapped implements DatabaseStageEntry {
private final DatabaseStageEntry serializedSingle;
private final Serializer serializer;
public DatabaseSingleMapped(DatabaseStageEntry serializedSingle, Serializer serializer) {
this.serializedSingle = serializedSingle;
this.serializer = serializer;
}
@Override
public Mono get(@Nullable CompositeSnapshot snapshot, boolean existsAlmostCertainly) {
return serializedSingle.get(snapshot, existsAlmostCertainly).map(this::deserialize);
}
@Override
public Mono getOrDefault(@Nullable CompositeSnapshot snapshot, Mono defaultValue) {
return serializedSingle.get(snapshot).map(this::deserialize).switchIfEmpty(defaultValue);
}
@Override
public Mono set(A value) {
return serializedSingle.set(serialize(value));
}
@Override
public Mono setAndGetPrevious(A value) {
return serializedSingle.setAndGetPrevious(serialize(value)).map(this::deserialize);
}
@Override
public Mono setAndGetStatus(A value) {
return serializedSingle.setAndGetStatus(serialize(value));
}
@Override
public Mono update(Function, Optional> updater, boolean existsAlmostCertainly) {
return serializedSingle
.update(oldValue -> updater.apply(oldValue.map(this::deserialize)).map(this::serialize), existsAlmostCertainly);
}
@Override
public Mono clear() {
return serializedSingle.clear();
}
@Override
public Mono clearAndGetPrevious() {
return serializedSingle.clearAndGetPrevious().map(this::deserialize);
}
@Override
public Mono clearAndGetStatus() {
return serializedSingle.clearAndGetStatus();
}
@Override
public Mono close() {
return serializedSingle.close();
}
@Override
public Mono leavesCount(@Nullable CompositeSnapshot snapshot, boolean fast) {
return serializedSingle.leavesCount(snapshot, fast);
}
@Override
public Mono isEmpty(@Nullable CompositeSnapshot snapshot) {
return serializedSingle.isEmpty(snapshot);
}
@Override
public DatabaseStageEntry entry() {
return this;
}
//todo: temporary wrapper. convert the whole class to buffers
private A deserialize(B bytes) {
return serializer.deserialize(bytes);
}
//todo: temporary wrapper. convert the whole class to buffers
private B serialize(A bytes) {
return serializer.serialize(bytes);
}
}