cn.wjybxx.dson.codec.codecs.MapCodec 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.codecs;

import cn.wjybxx.dson.DsonType;
import cn.wjybxx.dson.codec.DuplexCodec;
import cn.wjybxx.dson.codec.TypeArgInfo;
import cn.wjybxx.dson.codec.dson.DsonCodecScanIgnore;
import cn.wjybxx.dson.codec.dson.DsonObjectReader;
import cn.wjybxx.dson.codec.dson.DsonObjectWriter;
import cn.wjybxx.dson.codec.dsonlite.DsonLiteCodecScanIgnore;
import cn.wjybxx.dson.codec.dsonlite.DsonLiteObjectReader;
import cn.wjybxx.dson.codec.dsonlite.DsonLiteObjectWriter;
import cn.wjybxx.dson.text.ObjectStyle;

import javax.annotation.Nonnull;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.function.Supplier;

/**
 * @author wjybxx
 * date 2023/4/4
 */
@SuppressWarnings("rawtypes")
@DsonLiteCodecScanIgnore
@DsonCodecScanIgnore
public class MapCodec implements DuplexCodec {

    final Class clazz;
    final Supplier factory;

    public MapCodec(Class clazz, Supplier factory) {
        this.clazz = Objects.requireNonNull(clazz);
        this.factory = factory;
    }

    @Override
    public boolean autoStartEnd() {
        return false;
    }

    @Nonnull
    @Override
    public Class getEncoderClass() {
        return clazz;
    }

    @SuppressWarnings("unchecked")
    private Map newMap(TypeArgInfo typeArgInfo) {
        if (typeArgInfo.factory != null) {
            return (Map) typeArgInfo.factory.get();
        }
        if (factory != null) {
            return (Map) factory.get();
        }
        return new LinkedHashMap<>();
    }

    @Override
    public void writeObject(DsonLiteObjectWriter writer, T instance, TypeArgInfo typeArgInfo) {
        TypeArgInfo ketArgInfo = TypeArgInfo.of(typeArgInfo.typeArg1);
        TypeArgInfo valueArgInfo = TypeArgInfo.of(typeArgInfo.typeArg2);

        writer.writeStartArray(instance, typeArgInfo);
        @SuppressWarnings("unchecked") Set> entrySet = instance.entrySet();
        for (Map.Entry entry : entrySet) {
            writer.writeObject(0, entry.getKey(), ketArgInfo);
            writer.writeObject(0, entry.getValue(), valueArgInfo);
        }
        writer.writeEndArray();
    }

    @Override
    public T readObject(DsonLiteObjectReader reader, TypeArgInfo typeArgInfo) {
        Map result = newMap(typeArgInfo);
        TypeArgInfo ketArgInfo = TypeArgInfo.of(typeArgInfo.typeArg1);
        TypeArgInfo valueArgInfo = TypeArgInfo.of(typeArgInfo.typeArg2);

        reader.readStartArray(typeArgInfo);
        while (reader.readDsonType() != DsonType.END_OF_OBJECT) {
            Object key = reader.readObject(0, ketArgInfo);
            Object value = reader.readObject(0, valueArgInfo);
            result.put(key, value);
        }
        reader.readEndArray();
        return clazz.cast(result);
    }

    @Override
    public void writeObject(DsonObjectWriter writer, T instance, TypeArgInfo typeArgInfo, ObjectStyle style) {
        TypeArgInfo valueArgInfo = TypeArgInfo.of(typeArgInfo.typeArg2);
        @SuppressWarnings("unchecked") Set> entrySet = instance.entrySet();

        if (writer.options().writeMapAsDocument) {
            writer.writeStartObject(instance, typeArgInfo, style);
            for (Map.Entry entry : entrySet) {
                String keyString = writer.encodeKey(entry.getKey());
                Object value = entry.getValue();
                if (value == null) {
                    // map写为普通的Object的时候，必须要写入Null，否则containsKey会异常；要强制写入Null必须先写入Name
                    writer.writeName(keyString);
                    writer.writeNull(keyString);
                } else {
                    writer.writeObject(keyString, value, valueArgInfo, null);
                }
            }
            writer.writeEndObject();
        } else {
            TypeArgInfo ketArgInfo = TypeArgInfo.of(typeArgInfo.typeArg1);
            writer.writeStartArray(instance, typeArgInfo, style);
            for (Map.Entry entry : entrySet) {
                writer.writeObject(null, entry.getKey(), ketArgInfo, null);
                writer.writeObject(null, entry.getValue(), valueArgInfo, null);
            }
            writer.writeEndArray();
        }
    }

    @Override
    public T readObject(DsonObjectReader reader, TypeArgInfo typeArgInfo) {
        Map result = newMap(typeArgInfo);
        TypeArgInfo valueArgInfo = TypeArgInfo.of(typeArgInfo.typeArg2);

        if (reader.options().writeMapAsDocument) {
            reader.readStartObject(typeArgInfo);
            while (reader.readDsonType() != DsonType.END_OF_OBJECT) {
                String keyString = reader.readName();
                Object key = reader.decodeKey(keyString, typeArgInfo.typeArg1);
                Object value = reader.readObject(keyString, valueArgInfo);
                result.put(key, value);
            }
            reader.readEndObject();
        } else {
            TypeArgInfo ketArgInfo = TypeArgInfo.of(typeArgInfo.typeArg1);
            reader.readStartArray(typeArgInfo);
            while (reader.readDsonType() != DsonType.END_OF_OBJECT) {
                Object key = reader.readObject(null, ketArgInfo);
                Object value = reader.readObject(null, valueArgInfo);
                result.put(key, value);
            }
            reader.readEndArray();
        }
        return clazz.cast(result);
    }

}