cn.schoolwow.quickapi.handler.entity.AbstractEntityHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of QuickAPI Show documentation
There is a newer version: 3.1.1
package cn.schoolwow.quickapi.handler.entity;

import cn.schoolwow.quickapi.domain.APIEntity;
import cn.schoolwow.quickapi.domain.APIField;
import cn.schoolwow.quickapi.util.JavaDocReader;
import cn.schoolwow.quickapi.util.PackageUtil;
import cn.schoolwow.quickapi.util.QuickAPIConfig;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.sun.javadoc.ClassDoc;
import com.sun.javadoc.FieldDoc;
import com.sun.javadoc.Tag;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class AbstractEntityHandler implements EntityHandler{
    public static Map apiEntityMap = getEntityMap();
    /**获取实体类映射*/
    private static Map getEntityMap(){
        List classList = PackageUtil.scanPackage(QuickAPIConfig.entityPackageNameList.toArray(new String[0]));
        classList.addAll(QuickAPIConfig.entityClassList);
        EntityHandlerMapping[] entityHandlerMappings =EntityHandlerMapping.values();
        AbstractEntityHandler entityHandler = null;
        for(EntityHandlerMapping entityHandlerMapping:entityHandlerMappings){
            try {
                Class.forName(entityHandlerMapping.className);
                entityHandler = (AbstractEntityHandler) entityHandlerMapping._class.newInstance();
                break;
            } catch (ClassNotFoundException e) {
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        if(null==entityHandler){
            entityHandler = new AbstractEntityHandler();
        }
        return entityHandler.handleEntity(classList);
    }

    private Map handleEntity(List classList) {
        Map apiEntityMap = new HashMap<>();
        for(Class _class:classList) {
            APIEntity apiEntity = new APIEntity();
            //处理类名
            {
                apiEntity.className = _class.getName();
                apiEntity.simpleName = _class.getSimpleName();
                handleClass(_class,apiEntity);
            }
            //处理字段
            List fieldList = new ArrayList<>();
            Class tempClass =_class;
            while(null!=tempClass){
                Field[] fields = tempClass.getDeclaredFields();
                //排除静态变量和常量
                Field.setAccessible(fields,true);
                for(Field field:fields){
                    if(Modifier.isFinal(field.getModifiers())||Modifier.isStatic(field.getModifiers())){
                        continue;
                    }
                    fieldList.add(field);
                }
                tempClass = tempClass.getSuperclass();
                if("java.lang.Object".equals(tempClass)){
                    break;
                }
            }
            Field[] fields = fieldList.toArray(new Field[0]);
            APIField[] apiFields = new APIField[fields.length];
            {
                Field.setAccessible(fields, true);
                for (int i = 0; i < fields.length; i++) {
                    APIField apiField = new APIField();
                    apiField.name = fields[i].getName();
                    apiField.className = fields[i].getType().getName();
                    //处理泛型
                    Type type = fields[i].getGenericType();
                    if(type instanceof ParameterizedType){
                        ParameterizedType pType = (ParameterizedType)type;
                        Type genericType = pType.getActualTypeArguments()[0];
                        apiField.className += "<"+genericType.getTypeName()+">";
                    }
                    handleField(fields[i],apiField);
                    apiFields[i] = apiField;
                }
            }
            apiEntity.apiFields = apiFields;
            try {
                apiEntity.instance = JSON.toJSONString(_class.newInstance(), SerializerFeature.WriteMapNullValue);
            } catch (InstantiationException e) {
                e.printStackTrace();
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
            apiEntityMap.put(_class.getName(), apiEntity);
        }
        //提取注释部分
        {
            ClassDoc[] classDocs = JavaDocReader.getEntityJavaDoc();
            for (APIEntity apiEntity : apiEntityMap.values()) {
                for (ClassDoc classDoc : classDocs) {
                    if (apiEntity.className.equals(classDoc.qualifiedName())) {
                        if(apiEntity.description==null){
                            apiEntity.description = classDoc.commentText();
                        }
                        Tag[] authorTags = classDoc.tags("author");
                        if(null!=authorTags&&authorTags.length>0){
                            apiEntity.author = authorTags[0].text();
                        }
                        Tag[] sinceTags = classDoc.tags("since");
                        if(null!=sinceTags&&sinceTags.length>0){
                            apiEntity.since = sinceTags[0].text();
                        }
                        for (APIField apiField : apiEntity.apiFields) {
                            for (FieldDoc fieldDoc : classDoc.fields()) {
                                if (apiField.name.equals(fieldDoc.name())&&apiField.description==null) {
                                    apiField.description = fieldDoc.getRawCommentText();
                                    break;
                                }
                            }
                        }
                        break;
                    }
                }
            }
        }
        return apiEntityMap;
    }

    /**处理实体类*/
    public void handleClass(Class _class,APIEntity apiEntity){

    }

    /**处理实体属性*/
    public void handleField(Field field, APIField apiField){

    }
}