package org.bson.codecs;
import org.bson.BSONException;
import org.bson.BsonBinary;
import org.bson.BsonBinarySubType;
import org.bson.BsonReader;
import org.bson.BsonWriter;
import org.bson.UuidRepresentation;
import org.bson.codecs.configuration.CodecConfigurationException;
import org.bson.internal.UuidHelper;
import java.util.UUID;
import static org.bson.assertions.Assertions.notNull;
public class UuidCodec implements Codec<UUID> {
private final UuidRepresentation uuidRepresentation;
public UuidCodec(final UuidRepresentation uuidRepresentation) {
notNull("uuidRepresentation", uuidRepresentation);
this.uuidRepresentation = uuidRepresentation;
}
public UuidCodec() {
this.uuidRepresentation = UuidRepresentation.JAVA_LEGACY;
}
public UuidRepresentation getUuidRepresentation() {
return uuidRepresentation;
}
@Override
public void encode(final BsonWriter writer, final UUID value, final EncoderContext encoderContext) {
if (uuidRepresentation == UuidRepresentation.UNSPECIFIED) {
throw new CodecConfigurationException("The uuidRepresentation has not been specified, so the UUID cannot be encoded.");
}
byte[] binaryData = UuidHelper.encodeUuidToBinary(value, uuidRepresentation);
if (uuidRepresentation == UuidRepresentation.STANDARD) {
writer.writeBinaryData(new BsonBinary(BsonBinarySubType.UUID_STANDARD, binaryData));
} else {
writer.writeBinaryData(new BsonBinary(BsonBinarySubType.UUID_LEGACY, binaryData));
}
}
@Override
public UUID decode(final BsonReader reader, final DecoderContext decoderContext) {
byte subType = reader.peekBinarySubType();
if (subType != BsonBinarySubType.UUID_LEGACY.getValue() && subType != BsonBinarySubType.UUID_STANDARD.getValue()) {
throw new BSONException("Unexpected BsonBinarySubType");
}
byte[] bytes = reader.readBinaryData().getData();
return UuidHelper.decodeBinaryToUuid(bytes, subType, uuidRepresentation);
}
@Override
public Class<UUID> getEncoderClass() {
return UUID.class;
}
@Override
public String toString() {
return "UuidCodec{"
+ "uuidRepresentation=" + uuidRepresentation
+ '}';
}
}