com.qcloud.cos.internal.eventstreaming.HeaderValue Maven / Gradle / Ivy
/*
* Copyright 2010-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
* According to cos feature, we modify some class，comment, field name, etc.
*/
package com.qcloud.cos.internal.eventstreaming;
import static com.qcloud.cos.internal.eventstreaming.HeaderType.TIMESTAMP;
import static com.qcloud.cos.internal.eventstreaming.HeaderType.fromTypeId;
import static com.qcloud.cos.internal.eventstreaming.Utils.writeBytes;
import static com.qcloud.cos.internal.eventstreaming.Utils.writeString;
import com.qcloud.cos.utils.Base64;
import com.qcloud.cos.utils.ValidationUtils;
import java.io.DataOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.Date;
import java.util.UUID;
import org.joda.time.DateTime;
/**
* A typed header value. The underlying value can be obtained by calling the
* appropriate getter.
*/
public abstract class HeaderValue {
public static HeaderValue fromBoolean(boolean value) {
return new BooleanValue(value);
}
public static HeaderValue fromByte(byte value) {
return new ByteValue(value);
}
public static HeaderValue fromShort(short value) {
return new ShortValue(value);
}
public static HeaderValue fromInteger(int value) {
return new IntegerValue(value);
}
public static HeaderValue fromLong(long value) {
return new LongValue(value);
}
public static HeaderValue fromByteArray(byte[] bytes) {
return new ByteArrayValue(bytes);
}
public static HeaderValue fromByteBuffer(ByteBuffer buf) {
buf = buf.duplicate();
byte[] bytes = new byte[buf.remaining()];
buf.get(bytes);
return fromByteArray(bytes);
}
public static HeaderValue fromString(String string) {
return new StringValue(string);
}
public static HeaderValue fromTimestamp(DateTime value) {
return new TimestampValue(value);
}
public static HeaderValue fromDate(Date value) {
return new TimestampValue(new DateTime(value));
}
public static HeaderValue fromUuid(UUID value) {
return new UuidValue(value);
}
protected HeaderValue() {}
public abstract HeaderType getType();
public boolean getBoolean() {
throw new IllegalStateException();
}
public byte getByte() {
throw new IllegalStateException("Expected byte, but type was " + getType().name());
}
public short getShort() {
throw new IllegalStateException("Expected short, but type was " + getType().name());
}
public int getInteger() {
throw new IllegalStateException("Expected integer, but type was " + getType().name());
}
public long getLong() {
throw new IllegalStateException("Expected long, but type was " + getType().name());
}
public byte[] getByteArray() {
throw new IllegalStateException();
}
public final ByteBuffer getByteBuffer() {
return ByteBuffer.wrap(getByteArray());
}
public String getString() {
throw new IllegalStateException();
}
public DateTime getTimestamp() {
throw new IllegalStateException("Expected timestamp, but type was " + getType().name());
}
public Date getDate() {
return getTimestamp().toDate();
}
public UUID getUuid() {
throw new IllegalStateException("Expected UUID, but type was " + getType().name());
}
void encode(DataOutputStream dos) throws IOException {
dos.writeByte(getType().headerTypeId);
encodeValue(dos);
}
abstract void encodeValue(DataOutputStream dos) throws IOException;
static HeaderValue decode(ByteBuffer buf) {
HeaderType type = fromTypeId(buf.get());
switch (type) {
case TRUE:
return new BooleanValue(true);
case FALSE:
return new BooleanValue(false);
case BYTE:
return new ByteValue(buf.get());
case SHORT:
return new ShortValue(buf.getShort());
case INTEGER:
return fromInteger(buf.getInt());
case LONG:
return new LongValue(buf.getLong());
case BYTE_ARRAY:
return fromByteArray(Utils.readBytes(buf));
case STRING:
return fromString(Utils.readString(buf));
case TIMESTAMP:
return TimestampValue.decode(buf);
case UUID:
return UuidValue.decode(buf);
default:
throw new IllegalStateException();
}
}
private static final class BooleanValue extends HeaderValue {
private final boolean value;
private BooleanValue(boolean value) {
this.value = value;
}
@Override
public HeaderType getType() {
if (value) {
return HeaderType.TRUE;
} else {
return HeaderType.FALSE;
}
}
@Override
public boolean getBoolean() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) {}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
BooleanValue that = (BooleanValue) o;
return value == that.value;
}
@Override
public int hashCode() {
if (value) {
return 1;
} else {
return 0;
}
}
@Override
public String toString() {
return String.valueOf(value);
}
}
private static final class ByteValue extends HeaderValue {
private final byte value;
private ByteValue(byte value) {
this.value = value;
}
@Override
public HeaderType getType() {
return HeaderType.BYTE;
}
@Override
public byte getByte() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) {}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ByteValue that = (ByteValue) o;
return value == that.value;
}
@Override
public int hashCode() {
return (int) value;
}
@Override
public String toString() {
return String.valueOf(value);
}
}
private static final class ShortValue extends HeaderValue {
private final short value;
private ShortValue(short value) {
this.value = value;
}
@Override
public HeaderType getType() {
return HeaderType.SHORT;
}
@Override
public short getShort() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) {}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ShortValue that = (ShortValue) o;
return value == that.value;
}
@Override
public int hashCode() {
return (int) value;
}
@Override
public String toString() {
return String.valueOf(value);
}
}
private static final class IntegerValue extends HeaderValue {
private final int value;
private IntegerValue(int value) {
this.value = value;
}
@Override
public HeaderType getType() {
return HeaderType.INTEGER;
}
@Override
public int getInteger() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
dos.writeInt(value);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
IntegerValue that = (IntegerValue) o;
return value == that.value;
}
@Override
public int hashCode() {
return value;
}
@Override
public String toString() {
return String.valueOf(value);
}
}
private static final class LongValue extends HeaderValue {
private final long value;
private LongValue(long value) {
this.value = value;
}
@Override
public HeaderType getType() {
return HeaderType.LONG;
}
@Override
public long getLong() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
dos.writeLong(value);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
LongValue longValue = (LongValue) o;
return value == longValue.value;
}
@Override
public int hashCode() {
return (int) (value ^ (value >>> 32));
}
@Override
public String toString() {
return String.valueOf(value);
}
}
private static final class ByteArrayValue extends HeaderValue {
private final byte[] value;
private ByteArrayValue(byte[] value) {
this.value = ValidationUtils.assertNotNull(value, "value");
}
@Override
public HeaderType getType() {
return HeaderType.BYTE_ARRAY;
}
@Override
public byte[] getByteArray() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
writeBytes(dos, value);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
ByteArrayValue that = (ByteArrayValue) o;
return Arrays.equals(value, that.value);
}
@Override
public int hashCode() {
return Arrays.hashCode(value);
}
@Override
public String toString() {
return Base64.encodeAsString(value);
}
}
private static final class StringValue extends HeaderValue {
private final String value;
private StringValue(String value) {
this.value = ValidationUtils.assertNotNull(value, "value");
}
@Override
public HeaderType getType() {
return HeaderType.STRING;
}
@Override
public String getString() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
writeString(dos, value);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
StringValue that = (StringValue) o;
return value.equals(that.value);
}
@Override
public int hashCode() {
return value.hashCode();
}
@Override
public String toString() {
return '"' + value + '"';
}
}
private static final class TimestampValue extends HeaderValue {
private final DateTime value;
private TimestampValue(DateTime value) {
this.value = ValidationUtils.assertNotNull(value, "value");
}
static TimestampValue decode(ByteBuffer buf) {
long epochMillis = buf.getLong();
return new TimestampValue(new DateTime(epochMillis));
}
@Override
public HeaderType getType() {
return TIMESTAMP;
}
@Override
public DateTime getTimestamp() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
dos.writeLong(value.getMillis());
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
TimestampValue that = (TimestampValue) o;
return value.equals(that.value);
}
@Override
public int hashCode() {
return value.hashCode();
}
@Override
public String toString() {
return value.toString();
}
}
private static final class UuidValue extends HeaderValue {
private final UUID value;
private UuidValue(UUID value) {
this.value = ValidationUtils.assertNotNull(value, "value");
}
static UuidValue decode(ByteBuffer buf) {
long msb = buf.getLong();
long lsb = buf.getLong();
return new UuidValue(new UUID(msb, lsb));
}
@Override
public HeaderType getType() {
return HeaderType.UUID;
}
@Override
public UUID getUuid() {
return value;
}
@Override
void encodeValue(DataOutputStream dos) throws IOException {
dos.writeLong(value.getMostSignificantBits());
dos.writeLong(value.getLeastSignificantBits());
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
UuidValue uuidValue = (UuidValue) o;
return value.equals(uuidValue.value);
}
@Override
public int hashCode() {
return value.hashCode();
}
@Override
public String toString() {
return value.toString();
}
}
}