shade.com.alibaba.fastjson2.writer.ObjectWriterImplMap Maven / Gradle / Ivy
package com.alibaba.fastjson2.writer;
import com.alibaba.fastjson2.*;
import com.alibaba.fastjson2.filter.*;
import com.alibaba.fastjson2.util.*;
import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.math.BigDecimal;
import java.nio.charset.StandardCharsets;
import java.time.temporal.Temporal;
import java.util.*;
import static com.alibaba.fastjson2.JSONWriter.Feature.BrowserCompatible;
import static com.alibaba.fastjson2.JSONWriter.Feature.WriteNonStringKeyAsString;
import static com.alibaba.fastjson2.util.JDKUtils.UNSAFE;
import static com.alibaba.fastjson2.util.TypeUtils.CLASS_JSON_OBJECT_1x;
public final class ObjectWriterImplMap
extends ObjectWriterPrimitiveImpl {
static final byte[] TYPE_NAME_JSONObject1O = JSONB.toBytes("JO10");
static final long TYPE_HASH_JSONObject1O = Fnv.hashCode64("JO10");
static final ObjectWriterImplMap INSTANCE = new ObjectWriterImplMap(String.class, Object.class, JSONObject.class, JSONObject.class, 0);
static final ObjectWriterImplMap INSTANCE_1x;
static {
if (CLASS_JSON_OBJECT_1x == null) {
INSTANCE_1x = null;
} else {
INSTANCE_1x = new ObjectWriterImplMap(String.class, Object.class, CLASS_JSON_OBJECT_1x, CLASS_JSON_OBJECT_1x, 0);
}
}
final Type objectType;
final Class objectClass;
final Type keyType;
final Type valueType;
final String format;
final boolean valueTypeRefDetect;
volatile ObjectWriter keyWriter;
volatile ObjectWriter valueWriter;
final byte[] jsonbTypeInfo;
final long typeNameHash;
final long features;
final boolean jsonObject1; // fastjson 1 JSONObject
final Field jsonObject1InnerMap;
final long jsonObject1InnerMapOffset;
final char[] typeInfoUTF16;
final byte[] typeInfoUTF8;
public ObjectWriterImplMap(Class objectClass, long features) {
this(null, null, objectClass, objectClass, features);
}
public ObjectWriterImplMap(Type keyType, Type valueType, Class objectClass, Type objectType, long features) {
this(keyType, valueType, null, objectClass, objectType, features);
}
public ObjectWriterImplMap(Type keyType, Type valueType, String format, Class objectClass, Type objectType, long features) {
this.keyType = keyType;
this.valueType = valueType;
this.format = format;
this.objectClass = objectClass;
this.objectType = objectType;
this.features = features;
if (valueType == null) {
this.valueTypeRefDetect = true;
} else {
this.valueTypeRefDetect = !ObjectWriterProvider.isNotReferenceDetect(TypeUtils.getClass(valueType));
}
String typeName = TypeUtils.getTypeName(objectClass);
String typeInfoStr = "\"@type\":\"" + objectClass.getName() + "\"";
this.typeInfoUTF16 = typeInfoStr.toCharArray();
this.typeInfoUTF8 = typeInfoStr.getBytes(StandardCharsets.UTF_8);
jsonObject1 = "JO1".equals(typeName);
this.jsonbTypeInfo = JSONB.toBytes(typeName);
this.typeNameHash = Fnv.hashCode64(typeName);
long jsonObject1InnerMapOffset = -1;
if (jsonObject1) {
jsonObject1InnerMap = BeanUtils.getDeclaredField(objectClass, "map");
if (jsonObject1InnerMap != null) {
jsonObject1InnerMap.setAccessible(true);
jsonObject1InnerMapOffset = UNSAFE.objectFieldOffset(jsonObject1InnerMap);
}
} else {
jsonObject1InnerMap = null;
}
this.jsonObject1InnerMapOffset = jsonObject1InnerMapOffset;
}
public static ObjectWriterImplMap of(Class objectClass) {
if (objectClass == JSONObject.class) {
return INSTANCE;
}
if (objectClass == CLASS_JSON_OBJECT_1x) {
return INSTANCE_1x;
}
return new ObjectWriterImplMap(null, null, objectClass, objectClass, 0);
}
public static ObjectWriterImplMap of(Type type) {
Class objectClass = TypeUtils.getClass(type);
return new ObjectWriterImplMap(objectClass, 0);
}
public static ObjectWriterImplMap of(Type type, Class defineClass) {
return of(type, null, defineClass);
}
public static ObjectWriterImplMap of(Type type, String format, Class defineClass) {
Type keyType = null, valueType = null;
if (type instanceof ParameterizedType) {
ParameterizedType parameterizedType = (ParameterizedType) type;
Type rawType = parameterizedType.getRawType();
Type[] actualTypeArguments = parameterizedType.getActualTypeArguments();
if (actualTypeArguments.length == 2) {
keyType = actualTypeArguments[0];
valueType = actualTypeArguments[1];
}
}
return new ObjectWriterImplMap(keyType, valueType, format, defineClass, type, 0);
}
@Override
public void writeArrayMappingJSONB(JSONWriter jsonWriter,
Object object,
Object fieldName,
Type fieldType,
long features) {
Map map = (Map) object;
jsonWriter.startObject();
boolean writeNulls = jsonWriter.isWriteNulls();
for (Map.Entry entry : (Iterable>) map.entrySet()) {
String key = entry.getKey();
Object value = entry.getValue();
if (value == null) {
if (writeNulls) {
jsonWriter.writeString(key);
jsonWriter.writeNull();
}
continue;
}
jsonWriter.writeString(key);
Class valueType = value.getClass();
if (valueType == String.class) {
jsonWriter.writeString((String) value);
} else {
ObjectWriter valueWriter = jsonWriter.getObjectWriter(valueType);
valueWriter.writeJSONB(jsonWriter, value, key, this.valueType, this.features);
}
}
jsonWriter.endObject();
}
@Override
public void writeJSONB(JSONWriter jsonWriter, Object object, Object fieldName, Type fieldType, long features) {
if ((fieldType == this.objectType && jsonWriter.isWriteMapTypeInfo(object, objectClass, features))
|| jsonWriter.isWriteTypeInfo(object, fieldType, features)
) {
boolean ordered = false;
if (jsonObject1InnerMap != null) {
if (jsonObject1InnerMapOffset != -1) {
Object innerMap = UNSAFE.getObject(object, jsonObject1InnerMapOffset);
ordered = innerMap instanceof LinkedHashMap;
} else {
try {
Object innerMap = jsonObject1InnerMap.get(object);
ordered = innerMap instanceof LinkedHashMap;
} catch (IllegalAccessException ignored) {
}
}
}
if (ordered) {
jsonWriter.writeTypeName(TYPE_NAME_JSONObject1O, TYPE_HASH_JSONObject1O);
} else {
jsonWriter.writeTypeName(jsonbTypeInfo, typeNameHash);
}
}
Map map = (Map) object;
JSONWriter.Context context = jsonWriter.context;
jsonWriter.startObject();
Type fieldValueType = this.valueType;
if (fieldType == this.objectType) {
fieldValueType = this.valueType;
} else if (fieldType instanceof ParameterizedType) {
Type[] actualTypeArguments = ((ParameterizedType) fieldType).getActualTypeArguments();
if (actualTypeArguments.length == 2) {
fieldValueType = actualTypeArguments[1];
}
}
long contextFeatures = context.getFeatures();
boolean writeNulls = (contextFeatures & (JSONWriter.Feature.WriteNulls.mask | JSONWriter.Feature.NullAsDefaultValue.mask)) != 0;
boolean fieldBased = (contextFeatures & JSONWriter.Feature.FieldBased.mask) != 0;
ObjectWriterProvider provider = context.provider;
Class itemClass = null;
ObjectWriter itemWriter = null;
boolean contextRefDetect = (contextFeatures & JSONWriter.Feature.ReferenceDetection.mask) != 0;
int i = 0;
for (Iterator it = map.entrySet().iterator(); it.hasNext(); ++i) {
Map.Entry entry = it.next();
Object entryKey = entry.getKey();
Object value = entry.getValue();
if (value == null) {
if (writeNulls) {
if (entryKey instanceof String) {
jsonWriter.writeString((String) entryKey);
} else {
Class entryKeyClass = entryKey.getClass();
boolean keyRefDetect = contextRefDetect
&& !ObjectWriterProvider.isNotReferenceDetect(entryKeyClass);
String refPath = null;
if (keyRefDetect) {
jsonWriter.setPath(i, entry);
refPath = jsonWriter.setPath("key", entryKey);
}
if (refPath != null) {
jsonWriter.writeReference(refPath);
} else {
ObjectWriter keyWriter = provider.getObjectWriter(entryKeyClass, entryKeyClass, fieldBased);
keyWriter.writeJSONB(jsonWriter, entryKey, null, null, 0);
}
if (keyRefDetect) {
jsonWriter.popPath(entry);
jsonWriter.popPath(entryKey);
}
}
jsonWriter.writeNull();
}
continue;
}
if (entryKey instanceof String || (contextFeatures & JSONWriter.Feature.WriteClassName.mask) == 0) {
String key;
if (entryKey instanceof String) {
key = (String) entryKey;
} else {
key = entryKey.toString();
}
if (jsonWriter.symbolTable != null) {
jsonWriter.writeSymbol(key);
if (value instanceof String) {
jsonWriter.writeSymbol((String) value);
continue;
}
} else {
jsonWriter.writeString(key);
}
} else if (entryKey == null) {
jsonWriter.writeNull();
} else {
if (contextRefDetect) {
jsonWriter.config(JSONWriter.Feature.ReferenceDetection, false);
}
Class entryKeyClass = entryKey.getClass();
ObjectWriter keyWriter = provider.getObjectWriter(entryKeyClass, entryKeyClass, fieldBased);
keyWriter.writeJSONB(jsonWriter, entryKey, null, null, 0);
if (contextRefDetect) {
jsonWriter.config(JSONWriter.Feature.ReferenceDetection, true);
}
}
Class valueClass = value.getClass();
if (valueClass == String.class) {
jsonWriter.writeString((String) value);
continue;
}
if (valueClass == Integer.class) {
jsonWriter.writeInt32((Integer) value);
continue;
}
if (valueClass == Long.class) {
jsonWriter.writeInt64((Long) value);
continue;
}
boolean valueRefDetecChanged = false;
boolean valueRefDetect;
if (valueClass == this.valueType) {
valueRefDetect = contextRefDetect && this.valueTypeRefDetect;
} else {
valueRefDetect = contextRefDetect && !ObjectWriterProvider.isNotReferenceDetect(valueClass);
}
if (valueRefDetect) {
if (value == object) {
jsonWriter.writeReference("..");
continue;
}
String refPath;
if (entryKey instanceof String) {
refPath = jsonWriter.setPath((String) entryKey, value);
} else if (ObjectWriterProvider.isPrimitiveOrEnum(entryKey.getClass())) {
refPath = jsonWriter.setPath(entryKey.toString(), value);
} else {
if (map.size() != 1 && !(map instanceof SortedMap) && !(map instanceof LinkedHashMap)) {
refPath = null; // skip
jsonWriter.config(JSONWriter.Feature.ReferenceDetection, false);
valueRefDetecChanged = true;
valueRefDetect = false;
} else {
refPath = jsonWriter.setPath(i, value);
}
}
if (refPath != null) {
jsonWriter.writeReference(refPath);
jsonWriter.popPath(value);
continue;
}
}
ObjectWriter valueWriter;
if (valueClass == this.valueType && this.valueWriter != null) {
valueWriter = this.valueWriter;
} else if (itemClass == valueClass) {
valueWriter = itemWriter;
} else {
if (valueClass == JSONObject.class) {
valueWriter = ObjectWriterImplMap.INSTANCE;
} else if (valueClass == CLASS_JSON_OBJECT_1x) {
valueWriter = ObjectWriterImplMap.INSTANCE_1x;
} else if (valueClass == JSONArray.class) {
valueWriter = ObjectWriterImplList.INSTANCE;
} else if (valueClass == TypeUtils.CLASS_JSON_ARRAY_1x) {
valueWriter = ObjectWriterImplList.INSTANCE;
} else {
valueWriter = provider.getObjectWriter(valueClass, valueClass, fieldBased);
}
if (itemWriter == null) {
itemWriter = valueWriter;
itemClass = valueClass;
}
if (valueClass == this.valueType) {
this.valueWriter = valueWriter;
}
}
valueWriter.writeJSONB(jsonWriter, value, entryKey, fieldValueType, this.features);
if (valueRefDetecChanged) {
jsonWriter.config(JSONWriter.Feature.ReferenceDetection, true);
} else {
if (valueRefDetect) {
jsonWriter.popPath(value);
}
}
}
jsonWriter.endObject();
}
@Override
public boolean writeTypeInfo(JSONWriter jsonWriter) {
if (jsonWriter.utf8) {
jsonWriter.writeNameRaw(typeInfoUTF8);
} else {
jsonWriter.writeNameRaw(typeInfoUTF16);
}
return true;
}
@Override
public void write(JSONWriter jsonWriter, Object object, Object fieldName, Type fieldType, long features) {
if (jsonWriter.jsonb) {
writeJSONB(jsonWriter, object, fieldName, fieldType, features);
return;
}
if (hasFilter(jsonWriter)) {
writeWithFilter(jsonWriter, object, fieldName, fieldType, features);
return;
}
boolean refDetect = jsonWriter.isRefDetect();
jsonWriter.startObject();
if ((fieldType == this.objectType && jsonWriter.isWriteMapTypeInfo(object, objectClass, features))
|| jsonWriter.isWriteTypeInfo(object, fieldType, features)
) {
writeTypeInfo(jsonWriter);
}
Map map = (Map) object;
features |= jsonWriter.getFeatures();
if ((features & JSONWriter.Feature.MapSortField.mask) != 0) {
if (!(map instanceof SortedMap) && map.getClass() != LinkedHashMap.class) {
map = new TreeMap<>(map);
}
}
ObjectWriterProvider provider = jsonWriter.context.provider;
for (Map.Entry entry : (Iterable) map.entrySet()) {
Object value = entry.getValue();
Object key = entry.getKey();
if (value == null) {
if ((features & JSONWriter.Feature.WriteNulls.mask) != 0) {
if (key == null) {
jsonWriter.writeName("null");
} else if (key instanceof String) {
jsonWriter.writeName((String) key);
} else {
if ((features & (WriteNonStringKeyAsString.mask | BrowserCompatible.mask)) != 0) {
jsonWriter.writeName(key.toString());
} else {
if (key instanceof Integer) {
jsonWriter.writeName((Integer) key);
} else if (key instanceof Long) {
jsonWriter.writeName((Long) key);
} else {
jsonWriter.writeNameAny(key);
}
}
}
jsonWriter.writeColon();
jsonWriter.writeNull();
}
continue;
} else if ((features & JSONWriter.Feature.NotWriteEmptyArray.mask) != 0) {
if (value instanceof Collection && ((Collection) value).isEmpty()) {
continue;
}
if (value.getClass().isArray() && Array.getLength(value) == 0) {
continue;
}
}
String strKey = null;
if (keyWriter != null) {
keyWriter.write(jsonWriter, key, null, null, 0);
} else if (key == null) {
jsonWriter.writeName("null");
} else if (key instanceof String) {
jsonWriter.writeName(strKey = (String) key);
} else {
boolean writeAsString = (features & (WriteNonStringKeyAsString.mask | BrowserCompatible.mask)) != 0 && ObjectWriterProvider.isPrimitiveOrEnum(key.getClass());
if (writeAsString && (key instanceof Temporal || key instanceof Date)) {
writeAsString = false;
}
if (writeAsString) {
jsonWriter.writeName(strKey = key.toString());
} else {
if (key instanceof Integer) {
jsonWriter.writeName((Integer) key);
} else if (key instanceof Long) {
long longKey = (Long) key;
jsonWriter.writeName(longKey);
} else {
jsonWriter.writeNameAny(key);
}
}
}
jsonWriter.writeColon();
Class valueClass = value.getClass();
if (valueClass == String.class) {
jsonWriter.writeString((String) value);
continue;
} else if (valueClass == Integer.class) {
jsonWriter.writeInt32((Integer) value);
continue;
} else if (valueClass == Long.class) {
if ((provider.userDefineMask & ObjectWriterProvider.TYPE_INT64_MASK) == 0) {
jsonWriter.writeInt64((Long) value);
} else {
ObjectWriter valueWriter = jsonWriter.getObjectWriter(valueClass);
valueWriter.write(jsonWriter, value, strKey, Long.class, features);
}
continue;
} else if (valueClass == Boolean.class) {
jsonWriter.writeBool((Boolean) value);
continue;
} else if (valueClass == BigDecimal.class) {
if ((provider.userDefineMask & ObjectWriterProvider.TYPE_DECIMAL_MASK) == 0) {
jsonWriter.writeDecimal((BigDecimal) value, features, null);
} else {
ObjectWriter valueWriter = jsonWriter.getObjectWriter(valueClass);
valueWriter.write(jsonWriter, value, key, this.valueType, this.features);
}
continue;
}
boolean isPrimitiveOrEnum;
ObjectWriter valueWriter;
if (valueClass == this.valueType) {
if (this.valueWriter != null) {
valueWriter = this.valueWriter;
} else {
valueWriter = this.valueWriter = format != null
? jsonWriter.getObjectWriter(valueClass, format)
: jsonWriter.getObjectWriter(valueClass);
}
isPrimitiveOrEnum = ObjectWriterProvider.isPrimitiveOrEnum(value.getClass());
} else {
if (valueClass == JSONObject.class) {
valueWriter = ObjectWriterImplMap.INSTANCE;
isPrimitiveOrEnum = false;
} else if (valueClass == CLASS_JSON_OBJECT_1x) {
valueWriter = ObjectWriterImplMap.INSTANCE_1x;
isPrimitiveOrEnum = false;
} else if (valueClass == JSONArray.class) {
valueWriter = ObjectWriterImplList.INSTANCE;
isPrimitiveOrEnum = false;
} else if (valueClass == TypeUtils.CLASS_JSON_ARRAY_1x) {
valueWriter = ObjectWriterImplList.INSTANCE;
isPrimitiveOrEnum = false;
} else {
valueWriter = jsonWriter.getObjectWriter(valueClass);
isPrimitiveOrEnum = ObjectWriterProvider.isPrimitiveOrEnum(value.getClass());
}
}
boolean valueRefDetect = refDetect && strKey != null && !isPrimitiveOrEnum;
if (valueRefDetect) {
if (value == object) {
jsonWriter.writeReference("..");
continue;
}
String refPath = jsonWriter.setPath(strKey, value);
if (refPath != null) {
jsonWriter.writeReference(refPath);
jsonWriter.popPath(value);
continue;
}
}
valueWriter.write(jsonWriter, value, key, this.valueType, this.features);
if (valueRefDetect) {
jsonWriter.popPath(value);
}
}
jsonWriter.endObject();
}
@Override
public void writeWithFilter(JSONWriter jsonWriter, Object object, Object fieldName, Type fieldType, long features) {
if (object == null) {
jsonWriter.writeNull();
return;
}
jsonWriter.startObject();
Map map = (Map) object;
features |= jsonWriter.getFeatures();
if ((features & JSONWriter.Feature.MapSortField.mask) != 0) {
if (!(map instanceof SortedMap) && map.getClass() != LinkedHashMap.class) {
map = new TreeMap<>(map);
}
}
JSONWriter.Context context = jsonWriter.context;
BeforeFilter beforeFilter = context.getBeforeFilter();
if (beforeFilter != null) {
beforeFilter.writeBefore(jsonWriter, object);
}
PropertyPreFilter propertyPreFilter = context.getPropertyPreFilter();
NameFilter nameFilter = context.getNameFilter();
ValueFilter valueFilter = context.getValueFilter();
PropertyFilter propertyFilter = context.getPropertyFilter();
AfterFilter afterFilter = context.getAfterFilter();
boolean writeNulls = context.isEnabled(JSONWriter.Feature.WriteNulls.mask);
for (Map.Entry entry : (Iterable) map.entrySet()) {
Object value = entry.getValue();
if (value == null && !writeNulls) {
continue;
}
Object entryKey = entry.getKey();
String key;
if (entryKey == null) {
key = null;
} else {
key = entryKey.toString();
}
if (propertyPreFilter != null) {
if (!propertyPreFilter.process(jsonWriter, object, key)) {
continue;
}
}
if (nameFilter != null) {
key = nameFilter.process(object, key, value);
}
if (propertyFilter != null) {
if (!propertyFilter.apply(object, key, value)) {
continue;
}
}
if (valueFilter != null) {
value = valueFilter.apply(object, key, value);
}
if (value == null) {
if ((jsonWriter.getFeatures(features) & JSONWriter.Feature.WriteNulls.mask) == 0) {
continue;
}
}
jsonWriter.writeName(key);
jsonWriter.writeColon();
if (value == null) {
jsonWriter.writeNull();
} else {
Class valueType = value.getClass();
ObjectWriter valueWriter = jsonWriter.getObjectWriter(valueType);
valueWriter.write(jsonWriter, value, fieldName, fieldType, this.features);
}
}
if (afterFilter != null) {
afterFilter.writeAfter(jsonWriter, object);
}
jsonWriter.endObject();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy