com.alipay.rdf.file.util.BeanMapWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rdf-file-core Show documentation
RDF FILE CORE
There is a newer version: 2.2.11
package com.alipay.rdf.file.util;

import java.beans.BeanInfo;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import com.alipay.rdf.file.exception.RdfErrorEnum;
import com.alipay.rdf.file.exception.RdfFileException;

/**
 * Copyright (C) 2013-2018 Ant Financial Services Group
 * 
 * @author hongwei.quhw
 * @version $Id: BeanMapWrapper.java, v 0.1 2017年4月7日 下午2:56:25 hongwei.quhw Exp $
 */
public class BeanMapWrapper {
    private static final Map> CLASSINFO_CACHE = new ConcurrentHashMap>();
    private Map                           beanInfoCache   = null;
    private final Object                                              bean;
    private boolean                                                   isMap;

    public BeanMapWrapper(Object bean) {
        this.bean = bean;
        isMap = Map.class.isAssignableFrom(bean.getClass());
        if (!isMap) {
            initBeanInfo(bean.getClass());
        }
    }

    public BeanMapWrapper(Class clazz) {
        try {
            bean = clazz.newInstance();

            isMap = Map.class.isAssignableFrom(clazz);
            if (!isMap) {
                initBeanInfo(clazz);
            }
        } catch (Exception e) {
            throw new RdfFileException("ref-file# class=" + clazz.getName() + " 对象实例化错误", e,
                RdfErrorEnum.INSTANTIATION_ERROR);
        }
    }

    private void initBeanInfo(Class clazz) {
        try {

            Map beanInfoCache = CLASSINFO_CACHE.get(clazz.getName());

            if (null == beanInfoCache) {
                beanInfoCache = new HashMap();
                BeanInfo beanInfo = Introspector.getBeanInfo(clazz);
                PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
                for (PropertyDescriptor pd : propertyDescriptors) {
                    beanInfoCache.put(pd.getName(), pd);
                }
                CLASSINFO_CACHE.put(clazz.getName(), beanInfoCache);
            }

            this.beanInfoCache = beanInfoCache;
        } catch (Exception e) {
            throw new RdfFileException(e, RdfErrorEnum.INSTANTIATION_ERROR);
        }
    }

    public void setProperties(Map values) {
        for (String key : values.keySet()) {
            setProperty(key, values.get(key));
        }
    }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    public void setProperty(String propertyName, Object value) {
        if (isMap) {
            ((Map) bean).put(propertyName, value);
        } else {
            PropertyDescriptor pd = beanInfoCache.get(propertyName);
            if (null == pd || null == pd.getWriteMethod()) {
                throw new RdfFileException(
                    bean.getClass().getName() + "没有" + propertyName + "属性对应的写方法",
                    RdfErrorEnum.TYPE_GET_PROPERTY_ERROR);
            }

            try {
                pd.getWriteMethod().invoke(bean, new Object[] { value });
            } catch (Exception e) {
                throw new RdfFileException(e, RdfErrorEnum.TYPE_CONVERTION_ERROR);
            }
        }
    }

    @SuppressWarnings("rawtypes")
    public Object getProperty(String propertyName) {
        if (isMap) {
            return ((Map) bean).get(propertyName);
        } else {
            PropertyDescriptor pd = beanInfoCache.get(propertyName);
            if (null == pd || null == pd.getReadMethod()) {
                throw new RdfFileException(
                    bean.getClass().getName() + "没有" + propertyName + "属性对应的读方法",
                    RdfErrorEnum.TYPE_GET_PROPERTY_ERROR);
            }

            try {
                return pd.getReadMethod().invoke(bean, new Object[0]);
            } catch (Exception e) {
                throw new RdfFileException(e, RdfErrorEnum.TYPE_CONVERTION_ERROR);
            }
        }
    }

    public Object getBean() {
        return bean;
    }

    @Override
    public String toString() {
        return bean.toString();
    }
}