swagger扩展为按代码定义顺序展示接口和字段

1. 存在的问题

我们在项目中使用swagger2.0时,发现如下问题:

  • 我们想要按字段排序必须显示声明排序字段@ApiModelProperty(position=0)
  • swagger对注解@ApiOperation不支持position属性了,即不能按照指定接口顺序排序。(ps:官方也不提供解决,说设计如此,通过接口地址来排序。链接

2.解决方法

查询官网
使用对应的扩展插件(ModelPropertyBuilderPlugin、OperationBuilderPlugin)

2.1使用ModelPropertyBuilderPlugin按代码顺序展示请求和响应字段

可以参考ModelPropertyBuilderPlugin的实现类ApiModelPropertyPropertyBuilder来实现自己的插件,
大概实现思路为:获取到字段上有ApiModelProperty注解的字段所在的类,然后获取所有字段数组,看这个字段在数组的哪个位置,就设置position属性。
实现代码如下:

/**
 * 通过编写插件实现字段按类变量定义顺序排序,丰富模型属性
 */
@Component
public class CustomApiModelPropertyPositionBuilder implements ModelPropertyBuilderPlugin {
    private Log log = LogFactory.getLog(getClass());


    @Override
    public boolean supports(DocumentationType delimiter) {
        return SwaggerPluginSupport.pluginDoesApply(delimiter);
    }


    @Override
    public void apply(ModelPropertyContext context) {
        Optional<BeanPropertyDefinition> beanPropertyDefinitionOpt = context.getBeanPropertyDefinition();
        Optional<ApiModelProperty> annotation = Optional.absent();
        if (context.getAnnotatedElement().isPresent()) {
            annotation = annotation.or(findApiModePropertyAnnotation(context.getAnnotatedElement().get()));
        }
        if (context.getBeanPropertyDefinition().isPresent()) {
            annotation = annotation.or(findPropertyAnnotation(context.getBeanPropertyDefinition().get(), ApiModelProperty.class));
        }
        //必须是有ApiModelProperty注解的字段,并且beanPropertyDefinitionOpt带有注解
        if (annotation.isPresent() && beanPropertyDefinitionOpt.isPresent()) {
            BeanPropertyDefinition beanPropertyDefinition = beanPropertyDefinitionOpt.get();
            
            //获取到注解字段
            AnnotatedField field = beanPropertyDefinition.getField();
            //获取到字段所在的类
            Class<?> clazz = field.getDeclaringClass();
            //获取类中所有字段
            Field[] declaredFields = clazz.getDeclaredFields();
            Field declaredField;//获取当前字段的Field

            try {
                declaredField = clazz.getDeclaredField(field.getName());
            } catch (NoSuchFieldException | SecurityException e) {
                log.error("", e);
                return;
            }

            //获取当前字段在数组中的位置。然后设置position属性
            int indexOf = ArrayUtils.indexOf(declaredFields, declaredField);
            if (indexOf != -1) {
                context.getBuilder().position(indexOf);
            }
        }
    }
}

2.2使用OperationBuilderPlugin按代码顺序展示请求接口

可以参考OperationBuilderPlugin的实现类OperationHiddenReader来实现自己的插件,
大概实现思路为:获取到接口上有ApiOperation注解所在的类,然后获取到该接口方法在类中的多少行,用行号来进行接口排序。
实现代码如下:

/**
 * 通过编写插件实现接口按定义顺序排序
 */
@Component
public class CustomOperationBuilderPlugin implements OperationBuilderPlugin {
    private static final Logger log = LoggerFactory.getLogger(CustomOperationBuilderPlugin.class);

    @Override
    public boolean supports(DocumentationType documentationType) {
        return SwaggerPluginSupport.pluginDoesApply(documentationType);
    }


    @Override
    public void apply(OperationContext context) {
        //1.没有ApiOperation注解的直接返回
        Optional<ApiOperation> apiOperation = context.findAnnotation(ApiOperation.class);
        if (!apiOperation.isPresent()) {
            return;
        }

        //2.获取当前方法的位置,然后设置进position中
        try {
            Class<? extends OperationContext> operationContextClass = context.getClass();
            Field requestContextFiled = operationContextClass.getDeclaredField("requestContext");
            requestContextFiled.setAccessible(true);
            RequestMappingContext requestContext = (RequestMappingContext) requestContextFiled.get(context);

            Class<? extends RequestMappingContext> requestContextClass = requestContext.getClass();
            Field handler = requestContextClass.getDeclaredField("handler");
            handler.setAccessible(true);
            RequestHandler requestHandler = (RequestHandler) handler.get(requestContext);

            //得到当前handler对应的Controller
            Class<?> aClass = requestHandler.declaringClass();
            //获取所有方法
//            Method[] declaredMethods = aClass.getDeclaredMethods();
            //获取当前api对应哪个方法
            Method nowMethod = requestHandler.getHandlerMethod().getMethod();

            //等到当前方法在所有方法中的位置(TO:位置变成了编译后的位置,需要找到编译前的位置)
//            int indexOf = ArrayUtils.indexOf(declaredMethods, nowMethod);
            //使用javasisit获取到对应方法在原始类的多少行
            int indexOf = getMethodOriginalLine(aClass, nowMethod);
            if (indexOf != -1) {
                //swagger-ui高版本不支持position排序了,解决方法是引入knife4j-spring-ui
                //ps:找不到swagger-ui前端页面源代码加载位置,所以搞不定了~~~。有能力的小伙伴可以直接去修改swagger-ui的前端源代码
                context.operationBuilder().position(indexOf);
                //添加扩展参数:x-order(支持knife4j-spring-ui)
                context.operationBuilder().extensions(Lists.newArrayList(new StringVendorExtension("x-order", indexOf + "")));
            }

        } catch (Exception e) {
            log.warn("加载swagger中方法api={},设置顺序出错。", context.getName(), e);
        }
    }

    /**
     * 获取方法在类中的原始开始行数
     *
     * @param clazz     原始类
     * @param nowMethod 需要查找的哪个方法
     * @return
     */
    private int getMethodOriginalLine(Class clazz, Method nowMethod) throws Exception {
        ClassPool pool = ClassPool.getDefault();
        //部署到tomcat需要添加下面两行代码,否则javassist.NotFoundException
        /**The default ClassPool returned by a static method ClassPool.getDefault() searches the same path that the underlying JVM (Java virtual machine) has. If a program is running on a web application server such as JBoss and Tomcat, the ClassPoolobject may not be able to find user classes since such a web application server uses multiple class loaders as well as the system class loader. In that case, an additional class path must be registered to the ClassPool.
**/
        ClassClassPath classPath = new ClassClassPath(this.getClass());
        pool.insertClassPath(classPath);

        String className = clazz.getName();
        CtClass cc = pool.get(className);
        Class<?>[] parameterTypes = nowMethod.getParameterTypes();
        String[] objects = Arrays.stream(parameterTypes).map(Class::getName).collect(Collectors.toList()).toArray(new String[]{});
        return cc.getDeclaredMethod(nowMethod.getName(), pool.get(objects)).getMethodInfo().getLineNumber(0);
    }
}

3.注意事项

原来swagger-ui.html地址也会保留
访问地址: http://ip:port/doc.html
使用到的包:

用于获取接口在源代码中的行数:
<groupId>org.javassist</groupId>
<artifactId>javassist</artifactId>
<version>3.21.0-GA</version>
用于前端页面展示:
<groupId>com.github.xiaoymin</groupId>
<artifactId>knife4j-spring-ui</artifactId>
<version>2.0.8</version>

4.参考资料

  1. https://github.com/springfox/springfox/issues/732
  2. http://springfox.github.io/springfox/docs/current/#plugins-available-for-extensibility
  3. https://www.xiaominfo.com/2019/05/29/springfox-13/
  4. https://gitee.com/xiaoym/knife4j
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,117评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,963评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,897评论 0 240
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,805评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,208评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,535评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,797评论 2 311
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,493评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,215评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,477评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,988评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,325评论 2 252
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,971评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,807评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,544评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,455评论 2 266

推荐阅读更多精彩内容