All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.wildfly.clustering.marshalling.protostream.Scalar Maven / Gradle / Ivy
/*
* Copyright The WildFly Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.wildfly.clustering.marshalling.protostream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import java.util.OptionalInt;
import org.infinispan.protostream.descriptors.WireType;
/**
* Enumeration of common scalar marshaller implementations.
* @author Paul Ferraro
*/
public enum Scalar implements ScalarMarshaller {
ANY(new ScalarMarshaller<>() {
@Override
public Object readFrom(ProtoStreamReader reader) throws IOException {
return reader.readAny();
}
@Override
public void writeTo(ProtoStreamWriter writer, Object value) throws IOException {
writer.writeAnyNoTag(value);
}
@Override
public Class getJavaClass() {
return Object.class;
}
@Override
public WireType getWireType() {
return WireType.LENGTH_DELIMITED;
}
}),
BOOLEAN(new ScalarMarshaller() {
@Override
public Boolean readFrom(ProtoStreamReader reader) throws IOException {
return reader.readUInt32() != 0;
}
@Override
public void writeTo(ProtoStreamWriter writer, Boolean value) throws IOException {
writer.writeVarint32(value ? 1 : 0);
}
@Override
public Class getJavaClass() {
return Boolean.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
BYTE(new ScalarMarshaller() {
@Override
public Byte readFrom(ProtoStreamReader reader) throws IOException {
return (byte) reader.readSInt32();
}
@Override
public void writeTo(ProtoStreamWriter writer, Byte value) throws IOException {
// Roll the bits in order to move the sign bit from position 31 to position 0, to reduce the wire length of negative numbers.
// See https://developers.google.com/protocol-buffers/docs/encoding
writer.writeVarint32((value << 1) ^ (value >> (Integer.SIZE - 1)));
}
@Override
public Class getJavaClass() {
return Byte.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
SHORT(new ScalarMarshaller() {
@Override
public Short readFrom(ProtoStreamReader reader) throws IOException {
return (short) reader.readSInt32();
}
@Override
public void writeTo(ProtoStreamWriter writer, Short value) throws IOException {
// Roll the bits in order to move the sign bit from position 31 to position 0, to reduce the wire length of negative numbers.
// See https://developers.google.com/protocol-buffers/docs/encoding
writer.writeVarint32((value << 1) ^ (value >> (Integer.SIZE - 1)));
}
@Override
public Class getJavaClass() {
return Short.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
INTEGER(new ScalarMarshaller() {
@Override
public Integer readFrom(ProtoStreamReader reader) throws IOException {
return reader.readSInt32();
}
@Override
public void writeTo(ProtoStreamWriter writer, Integer value) throws IOException {
// Roll the bits in order to move the sign bit from position 31 to position 0, to reduce the wire length of negative numbers.
// See https://developers.google.com/protocol-buffers/docs/encoding
writer.writeVarint32((value << 1) ^ (value >> (Integer.SIZE - 1)));
}
@Override
public Class getJavaClass() {
return Integer.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
LONG(new ScalarMarshaller() {
@Override
public Long readFrom(ProtoStreamReader reader) throws IOException {
return reader.readSInt64();
}
@Override
public void writeTo(ProtoStreamWriter writer, Long value) throws IOException {
// Roll the bits in order to move the sign bit from position 63 to position 0, to reduce the wire length of negative numbers.
// See https://developers.google.com/protocol-buffers/docs/encoding
writer.writeVarint64((value << 1) ^ (value >> (Long.SIZE - 1)));
}
@Override
public Class getJavaClass() {
return Long.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
FLOAT(new ScalarMarshaller() {
@Override
public Float readFrom(ProtoStreamReader reader) throws IOException {
return reader.readFloat();
}
@Override
public void writeTo(ProtoStreamWriter writer, Float value) throws IOException {
int bits = Float.floatToRawIntBits(value);
byte[] bytes = new byte[Float.BYTES];
for (int i = 0; i < bytes.length; ++i) {
int index = i * Byte.SIZE;
bytes[i] = (byte) ((bits >> index) & 0xFF);
}
writer.writeRawBytes(bytes, 0, bytes.length);
}
@Override
public Class getJavaClass() {
return Float.class;
}
@Override
public WireType getWireType() {
return WireType.FIXED32;
}
}),
DOUBLE(new ScalarMarshaller() {
@Override
public Double readFrom(ProtoStreamReader reader) throws IOException {
return reader.readDouble();
}
@Override
public void writeTo(ProtoStreamWriter writer, Double value) throws IOException {
long bits = Double.doubleToRawLongBits(value);
byte[] bytes = new byte[Double.BYTES];
for (int i = 0; i < bytes.length; ++i) {
int index = i * Byte.SIZE;
if (index < Integer.SIZE) {
bytes[i] = (byte) ((bits >> index) & 0xFF);
} else {
bytes[i] = (byte) ((int) (bits >> index) & 0xFF);
}
}
writer.writeRawBytes(bytes, 0, bytes.length);
}
@Override
public Class getJavaClass() {
return Double.class;
}
@Override
public WireType getWireType() {
return WireType.FIXED64;
}
}),
CHARACTER(new ScalarMarshaller() {
@Override
public Character readFrom(ProtoStreamReader reader) throws IOException {
return (char) reader.readUInt32();
}
@Override
public void writeTo(ProtoStreamWriter writer, Character value) throws IOException {
writer.writeVarint32(value);
}
@Override
public Class getJavaClass() {
return Character.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
BYTE_BUFFER(new ScalarMarshaller() {
@Override
public ByteBuffer readFrom(ProtoStreamReader reader) throws IOException {
return reader.readByteBuffer();
}
@Override
public void writeTo(ProtoStreamWriter writer, ByteBuffer buffer) throws IOException {
if (buffer.hasArray()) {
int length = buffer.remaining();
writer.writeVarint32(length);
writer.writeRawBytes(buffer.array(), buffer.arrayOffset(), length);
} else {
// Don't mess with existing position
ByteBuffer copy = buffer.asReadOnlyBuffer();
int length = copy.remaining();
writer.writeVarint32(length);
byte[] bytes = new byte[length];
copy.get(bytes, copy.position(), length);
writer.writeRawBytes(bytes, 0, length);
}
}
@Override
public Class getJavaClass() {
return ByteBuffer.class;
}
@Override
public WireType getWireType() {
return WireType.LENGTH_DELIMITED;
}
}),
BYTE_ARRAY(new ScalarMarshaller() {
@Override
public byte[] readFrom(ProtoStreamReader reader) throws IOException {
return reader.readByteArray();
}
@Override
public void writeTo(ProtoStreamWriter writer, byte[] value) throws IOException {
writer.writeVarint32(value.length);
writer.writeRawBytes(value, 0, value.length);
}
@Override
public Class getJavaClass() {
return byte[].class;
}
@Override
public WireType getWireType() {
return WireType.LENGTH_DELIMITED;
}
}),
STRING(new ScalarMarshaller() {
@Override
public String readFrom(ProtoStreamReader reader) throws IOException {
return StandardCharsets.UTF_8.decode(reader.readByteBuffer()).toString();
}
@Override
public void writeTo(ProtoStreamWriter writer, String value) throws IOException {
BYTE_BUFFER.writeTo(writer, StandardCharsets.UTF_8.encode(value));
}
@Override
public Class getJavaClass() {
return String.class;
}
@Override
public WireType getWireType() {
return WireType.LENGTH_DELIMITED;
}
}),
REFERENCE(new ScalarMarshaller() {
@Override
public Reference readFrom(ProtoStreamReader reader) throws IOException {
return new Reference(reader.readUInt32());
}
@Override
public void writeTo(ProtoStreamWriter writer, Reference value) throws IOException {
writer.writeVarint32(value.getAsInt());
}
@Override
public Class getJavaClass() {
return Reference.class;
}
@Override
public WireType getWireType() {
return WireType.VARINT;
}
}),
;
private final ScalarMarshaller marshaller;
Scalar(ScalarMarshaller marshaller) {
this.marshaller = marshaller;
}
@Override
public Class getJavaClass() {
return this.marshaller.getJavaClass();
}
@Override
public WireType getWireType() {
return this.marshaller.getWireType();
}
@Override
public Object readFrom(ProtoStreamReader reader) throws IOException {
return this.marshaller.readFrom(reader);
}
@Override
public void writeTo(ProtoStreamWriter writer, Object value) throws IOException {
this.cast(Object.class).writeTo(writer, value);
}
@Override
public OptionalInt size(ProtoStreamSizeOperation operation, Object value) {
return this.cast(Object.class).size(operation, value);
}
@SuppressWarnings("unchecked")
public ScalarMarshaller cast(Class type) {
if (!type.isAssignableFrom(this.getJavaClass())) {
throw new IllegalArgumentException(type.getName());
}
return (ScalarMarshaller) this.marshaller;
}
}
