cn.wjybxx.dson.codec.dsonlite.DsonLiteCodecHelper Maven / Gradle / Ivy
/*
* Copyright 2023-2024 wjybxx([email protected])
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License 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.
*/
package cn.wjybxx.dson.codec.dsonlite;
import cn.wjybxx.dson.*;
import cn.wjybxx.dson.codec.DsonCodecException;
import cn.wjybxx.dson.types.*;
/**
* 1.int扩展之间可以相互转换，当int的扩展不可以直接转换为其它数值类型
* 2.long扩展之间可以相互转换，但long的扩展不可直接转换为其它数值类型
* 3.String扩展之间也可以相互转换
*
* @author wjybxx
* date - 2023/4/17
*/
class DsonLiteCodecHelper {
static DsonType readOrGetDsonType(DsonLiteReader reader) {
if (reader.isAtType()) {
return reader.readDsonType();
} else {
return reader.getCurrentDsonType();
}
}
static int readInt(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> reader.readInt32(name);
case INT64 -> (int) reader.readInt64(name);
case FLOAT -> (int) reader.readFloat(name);
case DOUBLE -> (int) reader.readDouble(name);
case BOOLEAN -> reader.readBoolean(name) ? 1 : 0;
case NULL -> {
reader.readNull(name);
yield 0;
}
case EXT_INT32 -> reader.readExtInt32(name).getValue();
default -> throw DsonCodecException.incompatible(Integer.class, dsonType);
};
}
static long readLong(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> reader.readInt32(name);
case INT64 -> reader.readInt64(name);
case FLOAT -> (long) reader.readFloat(name);
case DOUBLE -> (long) reader.readDouble(name);
case BOOLEAN -> reader.readBoolean(name) ? 1 : 0;
case NULL -> {
reader.readNull(name);
yield 0;
}
case EXT_INT64 -> reader.readExtInt64(name).getValue();
default -> throw DsonCodecException.incompatible(Long.class, dsonType);
};
}
static float readFloat(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> reader.readInt32(name);
case INT64 -> reader.readInt64(name);
case FLOAT -> reader.readFloat(name);
case DOUBLE -> (float) reader.readDouble(name);
case BOOLEAN -> reader.readBoolean(name) ? 1 : 0;
case NULL -> {
reader.readNull(name);
yield 0;
}
default -> throw DsonCodecException.incompatible(Float.class, dsonType);
};
}
static double readDouble(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> reader.readInt32(name);
case INT64 -> reader.readInt64(name);
case FLOAT -> reader.readFloat(name);
case DOUBLE -> reader.readDouble(name);
case BOOLEAN -> reader.readBoolean(name) ? 1 : 0;
case NULL -> {
reader.readNull(name);
yield 0;
}
case EXT_DOUBLE -> reader.readExtDouble(name).getValue();
default -> throw DsonCodecException.incompatible(Double.class, dsonType);
};
}
static boolean readBool(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> reader.readInt32(name) != 0;
case INT64 -> reader.readInt64(name) != 0;
case FLOAT -> reader.readFloat(name) != 0;
case DOUBLE -> reader.readDouble(name) != 0;
case BOOLEAN -> reader.readBoolean(name);
case NULL -> {
reader.readNull(name);
yield false;
}
default -> throw DsonCodecException.incompatible(Boolean.class, dsonType);
};
}
static String readString(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case STRING -> reader.readString(name);
case EXT_STRING -> reader.readExtString(name).getValue();
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(String.class, dsonType);
};
}
static void readNull(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
if (dsonType != DsonType.NULL) {
throw DsonCodecException.incompatible(DsonType.NULL, dsonType);
}
reader.readNull(name);
}
static Binary readBinary(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case BINARY -> reader.readBinary(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(DsonBinary.class, dsonType);
};
}
static ExtInt32 readExtInt32(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT32 -> new ExtInt32(0, reader.readInt32(name));
case EXT_INT32 -> reader.readExtInt32(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(DsonExtInt64.class, dsonType);
};
}
static ExtInt64 readExtInt64(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case INT64 -> new ExtInt64(0, reader.readInt64(name));
case EXT_INT64 -> reader.readExtInt64(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(DsonExtInt64.class, dsonType);
};
}
static ExtDouble readExtDouble(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case DOUBLE -> new ExtDouble(0, reader.readDouble(name));
case EXT_DOUBLE -> reader.readExtDouble(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(DsonExtDouble.class, dsonType);
};
}
static ExtString readExtString(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case STRING -> new ExtString(0, reader.readString(name));
case EXT_STRING -> reader.readExtString(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(DsonExtString.class, dsonType);
};
}
static ObjectRef readRef(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case REFERENCE -> reader.readRef(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(ObjectRef.class, dsonType);
};
}
static OffsetTimestamp readTimestamp(DsonLiteReader reader, int name) {
DsonType dsonType = readOrGetDsonType(reader);
return switch (dsonType) {
case TIMESTAMP -> reader.readTimestamp(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw DsonCodecException.incompatible(OffsetTimestamp.class, dsonType);
};
}
//
static Object readPrimitive(DsonLiteReader reader, int name, Class declared) {
if (declared == int.class) {
return readInt(reader, name);
}
if (declared == long.class) {
return readLong(reader, name);
}
if (declared == float.class) {
return readFloat(reader, name);
}
if (declared == double.class) {
return readDouble(reader, name);
}
if (declared == boolean.class) {
return readBool(reader, name);
}
if (declared == short.class) {
return (short) readInt(reader, name);
}
if (declared == char.class) {
return (char) readInt(reader, name);
}
if (declared == byte.class) {
return (byte) readInt(reader, name);
}
throw new AssertionError();
}
public static Object readValue(DsonLiteReader reader, DsonType dsonType, int name) {
return switch (dsonType) {
case INT32 -> reader.readInt32(name);
case INT64 -> reader.readInt64(name);
case FLOAT -> reader.readFloat(name);
case DOUBLE -> reader.readDouble(name);
case BOOLEAN -> reader.readBoolean(name);
case STRING -> reader.readString(name);
case BINARY -> reader.readBinary(name);
case EXT_STRING -> reader.readExtString(name);
case EXT_INT32 -> reader.readExtInt32(name);
case EXT_INT64 -> reader.readExtInt64(name);
case REFERENCE -> reader.readRef(name);
case TIMESTAMP -> reader.readTimestamp(name);
case NULL -> {
reader.readNull(name);
yield null;
}
default -> throw new AssertionError(dsonType); // null和容器都前面测试了
};
}
}© 2015 - 2024 Weber Informatics LLC | Privacy Policy