cn.willingxyz.restdoc.spring.SpringMethodResolver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of RestDocSpring Show documentation

There is a newer version: 0.2.1.4

package cn.willingxyz.restdoc.spring;

import cn.willingxyz.restdoc.core.parse.IMethodResolver;
import lombok.var;
import org.springframework.core.annotation.AnnotatedElementUtils;
import org.springframework.web.bind.annotation.*;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.List;

public class SpringMethodResolver implements IMethodResolver {
    private static List> _classes = Arrays.asList(
            RequestMapping.class,
            GetMapping.class,
            PostMapping.class,
            PutMapping.class,
            DeleteMapping.class,
            PatchMapping.class
    );

    @Override
    public boolean isSupport(Method method) {
        if (method.isSynthetic() || method.isBridge())
            return false;

        // 如果方法和类上都没有ResponseBody，返回false
        if (!AnnotatedElementUtils.hasAnnotation(method, ResponseBody.class) &&
            !AnnotatedElementUtils.hasAnnotation(method.getDeclaringClass(), ResponseBody.class))
        {
                return false;
        }
        var annotations = method.getAnnotations();
        for (var annotation : annotations) {
            var annotationType = annotation.annotationType();

            if (_classes.contains(annotationType))
                return true;
        }
        return false;
    }
}