32 lines
834 B
Java
32 lines
834 B
Java
|
package it.cavallium.dbengine.client;
|
||
|
|
|
||
|
|
import io.netty.buffer.ByteBuf;
|
||
|
|
import it.cavallium.dbengine.database.serialization.Serializer;
|
||
|
|
import org.jetbrains.annotations.NotNull;
|
||
|
|
|
||
|
|
public class MappedSerializer<A, B> implements Serializer<B, ByteBuf> {
|
||
|
|
|
||
|
|
private final Serializer<A, ByteBuf> serializer;
|
||
|
|
private final Mapper<A, B> keyMapper;
|
||
|
|
|
||
|
|
public MappedSerializer(Serializer<A, ByteBuf> serializer,
|
||
|
|
Mapper<A, B> keyMapper) {
|
||
|
|
this.serializer = serializer;
|
||
|
|
this.keyMapper = keyMapper;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public @NotNull B deserialize(@NotNull ByteBuf serialized) {
|
||
|
|
try {
|
||
|
|
return keyMapper.map(serializer.deserialize(serialized.retain()));
|
||
|
|
} finally {
|
||
|
|
serialized.release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public @NotNull ByteBuf serialize(@NotNull B deserialized) {
|
||
|
|
return serializer.serialize(keyMapper.unmap(deserialized));
|
||
|
|
}
|
||
|
|
}
|