swagger+knife4j记录

1. pom.xml



    4.0.0
    
        org.springframework.boot
        spring-boot-starter-parent
        2.2.2.RELEASE
        
    
    cn.itcast
    knife4j_demo
    1.0-SNAPSHOT
    
        
            org.springframework.boot
            spring-boot-starter-web
        
        
            org.springframework.boot
            spring-boot-configuration-processor
            true
        
        
            com.github.xiaoymin
            knife4j-spring-boot-starter
            2.0.1
        
        
            org.projectlombok
            lombok
        
    

2.application.yml

server:
  port: 8081
niewj:
  swagger:
    enabled: true #是否启用swagger
    base-package: com.niewj.controller
#    docket:
#      user:
#        title: 用户模块
#        base-package: com.niewj.controller.user
#      menu:
#        title: 菜单模块
#        base-package: com.niewj.controller.menu

3. 配置读取属性类SwaggerProperties

package com.niewj.config;

import lombok.*;
import org.springframework.boot.context.properties.ConfigurationProperties;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/*
*配置属性类,用于封装接口文档相关属性,从配置文件读取信息封装成当前对象
*/

@Data
@ConfigurationProperties(prefix = "niewj.swagger")
public class SwaggerProperties {
    private String title = "在线文档"; //标题
    private String group = ""; //自定义组名
    private String description = "在线文档"; //描述
    private String version = "1.0"; //版本
    private Contact contact = new Contact(); //联系人
    private String basePackage = "com.niewj"; //swagger会解析的包路径
    private List basePath = new ArrayList<>(); //swagger会解析的url规则
    private List excludePath = new ArrayList<>();//在basePath基础上需要排除的url规则
    private Map docket = new LinkedHashMap<>(); //分组文档
    public String getGroup() {
        if (group == null || "".equals(group)) {
            return title;
        }
        return group;
    }
    @Data
    public static class DocketInfo {
        private String title = "在线文档"; //标题
        private String group = ""; //自定义组名
        private String description = "在线文档"; //描述
        private String version = "1.0"; //版本
        private Contact contact = new Contact(); //联系人
        private String basePackage = ""; //swagger会解析的包路径
        private List basePath = new ArrayList<>(); //swagger会解析的url规则
        private List excludePath = new ArrayList<>();//在basePath基础上需要排除的url
        public String getGroup() {
            if (group == null || "".equals(group)) {
                return title;
            }
            return group;
        }
    }
    @Data
    public static class Contact {
        private String name = "niewj"; //联系人
        private String url = ""; //联系人url
        private String email = ""; //联系人email
    }
}

4. SwaggerConfig

package com.niewj.config;

import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Contact;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

@Configuration
@ConditionalOnProperty(name = "niewj.swagger.enabled", havingValue = "true", matchIfMissing = true)
@EnableSwagger2
@EnableConfigurationProperties(SwaggerProperties.class)
public class SwaggerConfig implements BeanFactoryAware {
    @Autowired
    SwaggerProperties swaggerProperties;
    private BeanFactory beanFactory;

    @Bean
    @ConditionalOnMissingBean
    public List createRestApi() {
        ConfigurableBeanFactory configurableBeanFactory = (ConfigurableBeanFactory) beanFactory;
        List docketList = new LinkedList<>();
        // 没有分组
        if (swaggerProperties.getDocket().isEmpty()) {
            Docket docket = createDocket(swaggerProperties);
            configurableBeanFactory.registerSingleton(swaggerProperties.getTitle(), docket);
            docketList.add(docket);
            return docketList;
        }
        // 分组创建
        for (String groupName : swaggerProperties.getDocket().keySet()) {
            SwaggerProperties.DocketInfo docketInfo = swaggerProperties.getDocket().get(groupName);
            ApiInfo apiInfo = new ApiInfoBuilder()
                    //页面标题
                    .title(docketInfo.getTitle())
                    //创建人
                    .contact(new Contact(docketInfo.getContact().getName(),
                            docketInfo.getContact().getUrl(),
                            docketInfo.getContact().getEmail()))
                    //版本号
                    .version(docketInfo.getVersion())
                    //描述
                    .description(docketInfo.getDescription())
                    .build();

            // base-path处理
            // 当没有配置任何path的时候,解析/**
            if (docketInfo.getBasePath().isEmpty()) {
                docketInfo.getBasePath().add("/**");
            }
            List> basePath = new ArrayList<>();
            for (String path : docketInfo.getBasePath()) {
                basePath.add(PathSelectors.ant(path));
            }

            // exclude-path处理
            List> excludePath = new ArrayList<>();
            for (String path : docketInfo.getExcludePath()) {
                excludePath.add(PathSelectors.ant(path));
            }

            Docket docket = new Docket(DocumentationType.SWAGGER_2)
                    .apiInfo(apiInfo)
                    .groupName(docketInfo.getGroup())
                    .select()
                    //为当前包路径
                    .apis(RequestHandlerSelectors.basePackage(docketInfo.getBasePackage()))
                    .paths(Predicates.and(Predicates.not(Predicates.or(excludePath)), Predicates.or(basePath)))
                    .build();
            configurableBeanFactory.registerSingleton(groupName, docket);
            docketList.add(docket);
        }
        return docketList;
    }

    //构建 api文档的详细信息
    private ApiInfo apiInfo(SwaggerProperties swaggerProperties) {
        return new ApiInfoBuilder()
                //页面标题
                .title(swaggerProperties.getTitle())
                //创建人
                .contact(new Contact(swaggerProperties.getContact().getName(),
                        swaggerProperties.getContact().getUrl(),
                        swaggerProperties.getContact().getEmail()))
                //版本号
                .version(swaggerProperties.getVersion())
                //描述
                .description(swaggerProperties.getDescription())
                .build();
    }

    //创建接口文档对象
    private Docket createDocket(SwaggerProperties swaggerProperties) {
        //API 基础信息
        ApiInfo apiInfo = apiInfo(swaggerProperties);

        // base-path处理
        // 当没有配置任何path的时候,解析/**
        if (swaggerProperties.getBasePath().isEmpty()) {
            swaggerProperties.getBasePath().add("/**");
        }
        List> basePath = new ArrayList<>();
        for (String path : swaggerProperties.getBasePath()) {
            basePath.add(PathSelectors.ant(path));
        }

        // exclude-path处理
        List> excludePath = new ArrayList<>();
        for (String path : swaggerProperties.getExcludePath()) {
            excludePath.add(PathSelectors.ant(path));
        }

        return new Docket(DocumentationType.SWAGGER_2)
                .apiInfo(apiInfo)
                .groupName(swaggerProperties.getGroup())
                .select()
                .apis(RequestHandlerSelectors.basePackage(swaggerProperties.getBasePackage()))
                .paths(Predicates.and(Predicates.not(Predicates.or(excludePath)), Predicates.or(basePath)))
                .build();
    }

    @Override
    public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
        this.beanFactory = beanFactory;
    }
}

5. 业务相关类

5.1 返回结果格式封装Result

package com.niewj.common.response;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.ArrayList;

/**
 * 返回结果格式封装
 *
 * @author nieweijun
 * @since 2021/6/29 18:43
 */
@NoArgsConstructor
@AllArgsConstructor
@Data
@ApiModel(description = "返回结果数据封装")
public class Result {

    @ApiModelProperty(value = "返回结果code码")
    private Integer code;
    @ApiModelProperty(value = "返回结果信息描述")
    private String msg;
    @ApiModelProperty(value = "返回数据成功标识")
    private Boolean success;
    @ApiModelProperty(value = "返回数据详情")
    private T data;

    /**
     * 用枚举来构造
     *
     * @param responseEnum 使用枚举值的code和msg填充Result
     * @param data         数据对象
     * @param ifSuccess    是否请求成功
     */
    public Result(ResponseEnum responseEnum, T data, Boolean ifSuccess) {
        this.setCode(responseEnum.getCode());
        this.setMsg(responseEnum.getMsg());
        this.setData(data);
        this.setSuccess(ifSuccess);
    }

    /**
     * success={"code":200,"msg":"成功","data":"xxx","success":true}
     *
     * @param data
     * @return
     */
    public static  Result success(T data) {
        return new Result<>(ResponseEnum.SUCCESS.getCode(), ResponseEnum.SUCCESS.getMsg(), Boolean.TRUE, data);
    }

    /**
     * success={"code":200,"msg":"成功","data": null,"success":true}
     *
     * @return
     */
    public static  Result successWithNull() {
        return new Result<>(ResponseEnum.SUCCESS.getCode(), ResponseEnum.SUCCESS.getMsg(), Boolean.TRUE, null);
    }

    /**
     * success={"code":200,"msg":"成功","data": [],"success":true}
     *
     * @return
     */
    public static  Result successWithEmptyList() {
        return (Result) Result.success(new ArrayList());
    }

    /**
     * success=除了返回true指定, 其他的都由参数指定
     *
     * @param code
     * @param msg
     * @param data
     * @param 
     * @return
     */
    public static  Result success(Integer code, String msg, T data) {
        return new Result<>(code, msg, Boolean.TRUE, data);
    }

    /**
     * success等价于new Result(ResponseEnum, T, true)
     *
     * @param responseEnum
     * @param data
     * @param 
     * @return
     */
    public static  Result success(ResponseEnum responseEnum, T data) {
        return new Result<>(responseEnum, data, Boolean.TRUE);
    }

    /**
     * fail-除了返回false, 其他的参数指定
     * 例如: {"code":xxx,"msg":"xxx", "data": "指定的错误", "success":false}
     *
     * @param data
     * @param 
     * @return
     */
    public static  Result fail(Integer code, String msg, T data) {
        return new Result<>(code, msg, Boolean.FALSE, data);
    }

    /**
     * fail-除了返回false, 其他的参数指定-由枚举类指定
     * 例如: {"code":xxx,"msg":"xxx", "data": "指定的错误", "success":false}
     *
     * @param responseEnum
     * @param 
     * @return
     */
    public static  Result fail(ResponseEnum responseEnum, T data) {
        return new Result<>(responseEnum.getCode(), responseEnum.getMsg(), Boolean.FALSE, data);
    }

}

5.2 响应枚举

package com.niewj.common.response;

/**
 * 响应枚举:code和message的封装(注意: code声明为int方便比较; Integer的话需要equals!)
 *
 * @author nieweijun
 * @since 2021/6/29 18:51
 */
public enum ResponseEnum {

    SUCCESS(200, "成功"),
    REPEAT(201, "重复操作"),
    BAD_REQUEST(400, "请求参数不合规"),
    ERROR(500, "服务器异常"),
    RPC_ERROR(501, "RPC服务器异常"),

    //=========================业务编码=========================
    NO_LOGIN(1001, "未登录"),
    RISK(1002, "系统繁忙"),
    RECODE_REPEAT(1003, "有重复数据"),
    DATA_LACK(1004, "请求数据缺失或超时"),
    NO_PERMISSION(1005, "用户无权限"),

    // ========================RPC服务问题========================
    RPC_INVOKE_ERROR(2001, "依赖服务发生异常"),
    RPC_DATA_INVALID(2002, "依赖服务数据无效"),
    RPC_INVOKE_TIMEOUT(2003, "依赖服务请求超时"),
    UNDER_REVIEW(2004, "预计10分钟内审核完毕,请稍后再重新进入小程序"),

    RPC_DISASSEMBLY_FAIL(2008, "提交旧机拆解失败"),

    OTHER(-1, "其他");

    // 声明为简单类型是为了方便用==比较;
    // 200 == Integer(200) ->true
    // 200 == 200 -> true
    // Integer(200) == Integer(200) -> false
    private int code;
    private String msg;

    ResponseEnum(int code, String msg) {
        this.code = code;
        this.msg = msg;
    }

    public String getMsg() {
        return msg;
    }

    public int getCode() {
        return code;
    }

    public static ResponseEnum getByCode(int code) {
        for (ResponseEnum responseCode : ResponseEnum.values()) {
            if (responseCode.getCode() == code) {
                return responseCode;
            }
        }
        return ERROR;
    }

}

5.3 model-1: Menu

package com.niewj.model;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@AllArgsConstructor
@NoArgsConstructor
@Data
@ApiModel(description = "菜单实体")
public class Menu {
    @ApiModelProperty(value = "主键")
    private int id;
    @ApiModelProperty(value = "菜单名称")
    private String name;
}

5.3 model-2: User

package com.niewj.model;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@AllArgsConstructor
@NoArgsConstructor
@Data
@ApiModel(description = "用户实体")
public class User {
    @ApiModelProperty(value = "主键")
    private int id;
    @ApiModelProperty(value = "姓名")
    private String name;
    @ApiModelProperty(value = "年龄")
    private int age;
    @ApiModelProperty(value = "地址")
    private String address;
}

5.4 MenuController

package com.niewj.controller;

import com.niewj.common.response.Result;
import com.niewj.model.Menu;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.List;

@RestController
@RequestMapping("/menu")
@Api(tags = "菜单控制器")
@Slf4j
public class MenuController {

    @GetMapping("/getMenus")
    @ApiOperation(value = "查询所有菜单", notes = "查询所有菜单信息")
    public Result> getMenus() {
        log.info("getMenus success!");

        List list = new ArrayList<>();
        list.add(new Menu(100, "menu100"));
        list.add(new Menu(101, "menu101"));
        list.add(new Menu(102, "menu102"));
        return Result.success(list);
    }

    @PostMapping("/save")
    @ApiOperation(value = "新增菜单", notes = "新增菜单信息")
    public Result save(@RequestBody Menu menu) {
        log.info("save success!");

        return Result.success(Boolean.TRUE);
    }

    @PutMapping("/update")
    @ApiOperation(value = "修改菜单", notes = "修改菜单信息")
    public Result update(@RequestBody Menu menu) {
        log.info("update success!");

        return Result.success(Boolean.TRUE);
    }

    @DeleteMapping("/delete")
    @ApiOperation(value = "删除菜单", notes = "删除菜单信息")
    public Result delete(int id) {
        log.info("update success!");

        return Result.success(Boolean.TRUE);
    }

    @ApiImplicitParams({
            @ApiImplicitParam(name = "pageNum", value = "页码", required = true, type = "Integer"),
            @ApiImplicitParam(name = "pageSize", value = "每页条数", required = true, type = "Integer"),
    })
    @ApiOperation(value = "分页查询菜单信息")
    @GetMapping(value = "page/{pageNum}/{pageSize}")
    public String findByPage(@PathVariable Integer pageNum,
                             @PathVariable Integer pageSize) {

        List list = new ArrayList<>();
        list.add(new Menu(100, "menu100"));
        list.add(new Menu(101, "menu101"));
        list.add(new Menu(102, "menu102"));

        return "OK";
    }
}

5.5 UserController

package com.niewj.controller;

import com.niewj.common.response.Result;
import com.niewj.model.User;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.List;

@RestController
@RequestMapping("/user")
@Api(tags = "用户控制器")
public class UserController {
    @GetMapping("/getUsers")
    @ApiOperation(value = "查询所有用户", notes = "查询所有用户信息")
    public Result> getAllUsers() {
        List list = new ArrayList<>();
        list.add(new User(100, "小0", 20, "北京"));
        list.add(new User(101, "小1", 30, "北京"));
        list.add(new User(102, "小2", 23, "西安"));
        return Result.success(list);
    }

    @PostMapping("/save")
    @ApiOperation(value = "新增用户", notes = "新增用户信息")
    public Result save(@RequestBody User user) {
        return Result.success(Boolean.TRUE);
    }

    @PutMapping("/update")
    @ApiOperation(value = "修改用户", notes = "修改用户信息")
    public Result update(@RequestBody User user) {
        return Result.success(Boolean.TRUE);
    }

    @DeleteMapping("/delete")
    @ApiOperation(value = "删除用户", notes = "删除用户信息")
    public Result delete(int id) {
        return Result.success(Boolean.TRUE);
    }

    @ApiImplicitParams({
            @ApiImplicitParam(name = "pageNum", value = "页码", required = true, type = "Integer"),
            @ApiImplicitParam(name = "pageSize", value = "每页条数", required = true, type = "Integer"),
    })
    @ApiOperation(value = "分页查询用户信息")
    @GetMapping(value = "page/{pageNum}/{pageSize}")
    public Result> findByPage(@PathVariable Integer pageNum,
                                         @PathVariable Integer pageSize) {
        List list = new ArrayList<>();
        list.add(new User(100, "小0", 20, "北京"));
        list.add(new User(101, "小1", 30, "北京"));
        list.add(new User(102, "小2", 23, "西安"));
        return Result.success(list);
    }
}

5.6 启动类

package com.niewj;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

@SpringBootApplication
public class DemoApplication {

    public static void main(String[] args) {
        SpringApplication.run(DemoApplication.class, args);
    }

}

5.7 启动后访问

访问地址: http://localhost:8081/doc.html
即可

你可能感兴趣的:(swagger)