cn.wjybxx.dson.codec.dsonlite.DsonLiteCodecImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dson-codec Show documentation
Dson对象序列化模块
There is a newer version: 2.2.0
/*
 * 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.codec.TypeArgInfo;
import cn.wjybxx.dson.codec.codecs.EnumLiteCodec;

import javax.annotation.Nonnull;
import java.util.Objects;

/**
 * @author wjybxx
 * date 2023/3/31
 */
public class DsonLiteCodecImpl {

    private final DsonLiteCodec codec;
    private final boolean isArray;

    public DsonLiteCodecImpl(DsonLiteCodec codec) {
        Objects.requireNonNull(codec.getEncoderClass(), "codecImpl.encoderClass");
        this.codec = codec;
        this.isArray = codec.isWriteAsArray();
    }

    /** 获取负责编解码的类对象 */
    @Nonnull
    public Class getEncoderClass() {
        return codec.getEncoderClass();
    }

    /** 是否编码为数组 */
    public boolean isWriteAsArray() {
        return isArray;
    }

    /**
     * 从输入流中解析指定对象。
     * 它应该创建对象，并反序列化该类及其所有超类定义的所有要序列化的字段。
     */
    public T readObject(DsonLiteObjectReader reader, TypeArgInfo typeArgInfo) {
        if (codec.autoStartEnd()) {
            T result;
            if (isArray) {
                reader.readStartArray(typeArgInfo);
                result = codec.readObject(reader, typeArgInfo);
                reader.readEndArray();
            } else {
                reader.readStartObject(typeArgInfo);
                result = codec.readObject(reader, typeArgInfo);
                reader.readEndObject();
            }
            return result;
        } else {
            return codec.readObject(reader, typeArgInfo);
        }
    }

    /**
     * 将对象写入输出流。
     * 将对象及其所有超类定义的所有要序列化的字段写入输出流。
     */
    public void writeObject(DsonLiteObjectWriter writer, T instance, TypeArgInfo typeArgInfo) {
        if (codec.autoStartEnd()) {
            if (isArray) {
                writer.writeStartArray(instance, typeArgInfo);
                codec.writeObject(writer, instance, typeArgInfo);
                writer.writeEndArray();
            } else {
                writer.writeStartObject(instance, typeArgInfo);
                codec.writeObject(writer, instance, typeArgInfo);
                writer.writeEndObject();
            }
        } else {
            codec.writeObject(writer, instance, typeArgInfo);
        }
    }

    public boolean isEnumLiteCodec() {
        return codec instanceof EnumLiteCodec;
    }

    @SuppressWarnings("unchecked")
    public T forNumber(int number) {
        if (codec instanceof EnumLiteCodec enumLiteCodec) {
            return (T) enumLiteCodec.forNumber(number);
        }
        throw new UnsupportedOperationException("unexpected forNumber method call");
    }
}