代码生成(1) : 根据接口文档生成实体类属性

ExcelUtil.java : https://blog.csdn.net/Lxinccode/article/details/97279040

说明 : xls文件第一行可以不需要,字段名、字段类型、字段含义和字段说明位置可以配置

创建xls文件,内容如下

字段名字段类型长度字段含义字段说明必选
nameString10姓名人的姓名必须
ageInteger2年龄人的年龄必须

生成的代码如下

@ApiModelProperty("字段含义,说明:字段说明")
private null 字段名;

@ApiModelProperty("姓名,说明:人的姓名")
private String name;

@ApiModelProperty("年龄,说明:人的年龄")
private Integer age;

工具代码如下 


import cn.nordrassil.util.ExcelUtil;
import cn.nordrassil.util.StringUtil;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * @Date: 2020/9/16 15:08
 * @Description: 根据文档生成实体类
 */
public class GenerateAttributeServer {

    private final static Map<String, String> map = new HashMap<String, String>() {{
        put("JSONString", "String");
        put("String", "String");
        put("VarChar2", "String");
        put("Varchar2", "String");
        put("varchar", "String");
        put("int", "Integer");
        put("Int", "Integer");
        put("Integer", "Integer");
        put("Double", "Double");
        put("Long", "Long");
    }};

    private final static Integer ZI_DUAN_MING = 0;
    private final static Integer ZI_DUAN_LEI_XING = 1;
    private final static Integer ZI_DUAN_HAN_YI = 3;
    private final static Integer ZI_DUAN_SHUO_MING = 4;

    public static void main(String[] args) {
        test1();
    }

    private static void test1() {
        List<String> obj = new LinkedList<>();
        List<List<String>> lists = ExcelUtil.readXls("/Users/leyili/Desktop/test1.xls");
        StringBuilder stringBuilder = new StringBuilder();
        for (List<String> list : lists) {
            if (list.size() == 0)
                break;
            int i = 0;
            for (String l : list) {
                if (l.equals(""))
                    i++;
            }
            if (i == list.size())
                continue;
            String descrbe = list.get(ZI_DUAN_HAN_YI);
            if (list.size() > ZI_DUAN_SHUO_MING && !list.get(ZI_DUAN_SHUO_MING).equals("")){
                descrbe += ",说明:"+list.get(ZI_DUAN_SHUO_MING);
            }
            descrbe = descrbe.replaceAll("\"", "");
            descrbe = descrbe.replaceAll("\n",",");
            stringBuilder.append("@ApiModelProperty(\"" + descrbe + "\")\n");
            stringBuilder.append("private ");
            if (list.get(ZI_DUAN_LEI_XING).indexOf("JSONString_") != -1) {
                String type = list.get(ZI_DUAN_LEI_XING).split("_")[1];
                switch (type) {
                    case "S":
                        stringBuilder.append("String");
                        break;
                    case "I":
                        stringBuilder.append("Integer");
                        break;
                    case "LI":
                        stringBuilder.append("List<Integer>");
                        break;
                    case "LLI":
                        stringBuilder.append("List<List<Integer>>");
                        break;
                    default:
                        break;
                }
            } else if (list.get(ZI_DUAN_LEI_XING).equals("JsonArray")) {
                String o = StringUtil.oneToUp(list.get(ZI_DUAN_MING));
                stringBuilder.append("List<"+o+"> ");
                obj.add(o);
            } else if (list.get(ZI_DUAN_LEI_XING).equals("JSONObject") || list.get(ZI_DUAN_LEI_XING).equals("Json") || list.get(ZI_DUAN_LEI_XING).equals("JsonObject")) {
                String o =oneToUp(list.get(ZI_DUAN_MING));
                stringBuilder.append(o);
                obj.add(o);
            } else if (list.get(ZI_DUAN_LEI_XING).indexOf("JSONString_") != -1) {
                String type = list.get(ZI_DUAN_LEI_XING).split("_")[1];
                switch (type) {
                    case "LS":
                        stringBuilder.append("List<String>");
                        break;
                    case "LI":
                        stringBuilder.append("List<Integer>");
                    default:
                        break;
                }
            } else {
                stringBuilder.append(map.get(list.get(ZI_DUAN_LEI_XING)));
            }
            stringBuilder.append(" ").append(list.get(ZI_DUAN_MING)).append(";\n\n");
        }
        if (obj.size() >0){
            for (String s : obj) {
                stringBuilder.append("@Setter\n").append("@Getter\npublic static class ").append(s).append("{\n");
                stringBuilder.append("\n}\n");
            }
        }
        System.err.println(stringBuilder);
    }


    /**
     * 首字母转大写
     * @param str
     * @return
     */
    public static String oneToUp(String str){
        return str.substring(0,1).toUpperCase()+str.substring(1);
    }

}

@ApiModelProperty 是 swagger,maven坐标如下

<springfox-swagger.version>2.9.2</springfox-swagger.version>
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger-ui</artifactId>
            <version>${springfox-swagger.version}</version>
        </dependency>
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger2</artifactId>
            <version>${springfox-swagger.version}</version>
            <exclusions>
                <exclusion>
                    <artifactId>mapstruct</artifactId>
                    <groupId>org.mapstruct</groupId>
                </exclusion>
            </exclusions>
        </dependency>

END。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值