io.github.jinghui70.rainbow.dbaccess.fieldmapper.BlobObjectFieldMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rainbow-dbaccess Show documentation
Rainbow DBAccess
The newest version!
package io.github.jinghui70.rainbow.dbaccess.fieldmapper;

import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONUtil;

import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.nio.charset.StandardCharsets;
import java.util.List;

public class BlobObjectFieldMapper extends BlobFieldMapper {

    private final Class fieldClass;

    private Class componentClass;

    public BlobObjectFieldMapper(Class fieldClass, Field field) {
        this.fieldClass = fieldClass;
        if (fieldClass.isArray())
            componentClass = fieldClass.getComponentType();
        else if (field!=null && fieldClass.isAssignableFrom(List.class)) {
            ParameterizedType parameterizedType = (ParameterizedType) field.getGenericType();
            Type[] actualTypeArguments = parameterizedType.getActualTypeArguments();
            componentClass = (Class) actualTypeArguments[0];
        }
    }

    @Override
    protected byte[] getBytes(Object value) {
        String json = JSONUtil.toJsonStr(value);
        return json.getBytes(StandardCharsets.UTF_8);
    }

    @SuppressWarnings("unchecked")
    @Override
    protected T parse(byte[] bytes) {
        String json = new String(bytes, StandardCharsets.UTF_8);
        if (fieldClass.isArray()) {
            JSONArray array = JSONUtil.parseArray(json);
            return (T) array.toArray(componentClass);
        }
        if (fieldClass.isAssignableFrom(List.class)) {
            JSONArray array = JSONUtil.parseArray(json);
            return (T) array.toList(componentClass);
        }
        return JSONUtil.toBean(json, fieldClass);
    }

    public static  BlobObjectFieldMapper of(Class fieldClass) {
        return new BlobObjectFieldMapper<>(fieldClass, null);
    }

    @SuppressWarnings("unchecked")
    public static  BlobObjectFieldMapper> ofList(Class componentClass) {
        BlobObjectFieldMapper result = new BlobObjectFieldMapper<>(List.class, null);
        result.componentClass = componentClass;
        return  (BlobObjectFieldMapper>) result;
    }

    @SuppressWarnings("unchecked")
    public static  BlobObjectFieldMapper ofArray(Class componentClass) {
        BlobObjectFieldMapper result = new BlobObjectFieldMapper<>(Object[].class, null);
        result.componentClass = componentClass;
        return  (BlobObjectFieldMapper) result;
    }
}