85 lines
2.4 KiB
Java
85 lines
2.4 KiB
Java
package it.cavallium.dbengine.database.collections;
|
|
|
|
import it.cavallium.dbengine.buffers.Buf;
|
|
import it.cavallium.dbengine.buffers.BufDataInput;
|
|
import it.cavallium.dbengine.buffers.BufDataOutput;
|
|
import it.cavallium.dbengine.database.LLKeyValueDatabaseStructure;
|
|
import it.cavallium.dbengine.database.LLSingleton;
|
|
import it.cavallium.dbengine.database.LLSnapshot;
|
|
import it.cavallium.dbengine.database.UpdateReturnMode;
|
|
import it.cavallium.dbengine.database.serialization.SerializerFixedBinaryLength;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class DatabaseLong implements LLKeyValueDatabaseStructure {
|
|
|
|
private final LLSingleton singleton;
|
|
private final SerializerFixedBinaryLength<Long> serializer;
|
|
private final SerializerFixedBinaryLength<Integer> bugSerializer;
|
|
|
|
public DatabaseLong(LLSingleton singleton) {
|
|
this.singleton = singleton;
|
|
this.serializer = SerializerFixedBinaryLength.longSerializer();
|
|
this.bugSerializer = SerializerFixedBinaryLength.intSerializer();
|
|
}
|
|
|
|
public Long get(@Nullable LLSnapshot snapshot) {
|
|
var result = BufDataInput.create(singleton.get(snapshot));
|
|
if (result.available() == 4) {
|
|
return (long) (int) bugSerializer.deserialize(result);
|
|
} else {
|
|
return serializer.deserialize(result);
|
|
}
|
|
}
|
|
|
|
public Long incrementAndGet() {
|
|
return addAnd(1, UpdateReturnMode.GET_NEW_VALUE);
|
|
}
|
|
|
|
public Long getAndIncrement() {
|
|
return addAnd(1, UpdateReturnMode.GET_OLD_VALUE);
|
|
}
|
|
|
|
public Long decrementAndGet() {
|
|
return addAnd(-1, UpdateReturnMode.GET_NEW_VALUE);
|
|
}
|
|
|
|
public Long getAndDecrement() {
|
|
return addAnd(-1, UpdateReturnMode.GET_OLD_VALUE);
|
|
}
|
|
|
|
public Long addAndGet(long count) {
|
|
return addAnd(count, UpdateReturnMode.GET_NEW_VALUE);
|
|
}
|
|
|
|
public Long getAndAdd(long count) {
|
|
return addAnd(count, UpdateReturnMode.GET_OLD_VALUE);
|
|
}
|
|
|
|
private Long addAnd(long count, UpdateReturnMode updateReturnMode) {
|
|
var result = singleton.update(prev -> {
|
|
if (prev != null) {
|
|
var prevLong = prev.getLong(0);
|
|
var buf = Buf.createZeroes(Long.BYTES);
|
|
buf.setLong(0, prevLong + count);
|
|
return buf;
|
|
} else {
|
|
var buf = Buf.createZeroes(Long.BYTES);
|
|
buf.setLong(0, count);
|
|
return buf;
|
|
}
|
|
}, updateReturnMode);
|
|
return result.getLong(0);
|
|
}
|
|
|
|
public void set(long value) {
|
|
var buf = BufDataOutput.createLimited(Long.BYTES);
|
|
serializer.serialize(value, buf);
|
|
singleton.set(buf.asList());
|
|
}
|
|
|
|
@Override
|
|
public String getDatabaseName() {
|
|
return singleton.getDatabaseName();
|
|
}
|
|
}
|