优化代码

java110

2021-10-17 e9e9d2d2e0f4d57af0b1db4387131a4c86b86230

 java110-generator/src/main/java/com/java110/code/TableToJson.java

@@ -7,36 +7,40 @@
public class TableToJson {

    //show create table c_orders  用这个语句获取
    public static final String createTableSql = "CREATE TABLE `c_orders` (\n" +
            "  `o_id` varchar(30) NOT NULL COMMENT '订单ID',\n" +
            "  `app_id` varchar(30) NOT NULL COMMENT '应用ID',\n" +
            "  `ext_transaction_id` varchar(36) DEFAULT NULL,\n" +
            "  `user_id` varchar(30) NOT NULL COMMENT '用户ID',\n" +
            "  `request_time` varchar(16) NOT NULL COMMENT '外部系统请求时间',\n" +
    public static final String createTableSql = "CREATE TABLE `owner_car_open_user` (\n" +
            "  `open_user_id` varchar(30) NOT NULL COMMENT 'app用户ID',\n" +
            "  `open_id` varchar(128) NOT NULL COMMENT '微信或支付宝开放ID',\n" +
            "  `car_num` varchar(32) NOT NULL COMMENT '车牌号',\n" +
            "  `open_type` varchar(12) NOT NULL DEFAULT 'WECHAT' COMMENT '微信WECHAT 支付宝 ALI',\n" +
            "  `nickname` varchar(255) DEFAULT NULL COMMENT '昵称',\n" +
            "  `headimgurl` varchar(255) DEFAULT NULL COMMENT '微信头像',\n" +
            "  `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
            "  `order_type_cd` varchar(4) NOT NULL COMMENT '订单类型，参考c_order_type表',\n" +
            "  `finish_time` date DEFAULT NULL COMMENT '订单完成时间',\n" +
            "  `remark` varchar(200) DEFAULT NULL COMMENT '备注',\n" +
            "  `status_cd` varchar(2) NOT NULL COMMENT '数据状态，详细参考c_status表',\n" +
            "  UNIQUE KEY `o_id` (`o_id`) USING BTREE,\n" +
            "  UNIQUE KEY `o_id_2` (`o_id`) USING BTREE\n" +
            "  `status_cd` varchar(2) NOT NULL DEFAULT '0' COMMENT '数据状态，详细参考c_status表，0, 在用 1失效',\n" +
            "  PRIMARY KEY (`open_user_id`)\n" +
            ")";

    public static void main(String[] args) {
        String desc = "车辆微信关系";
        String id = "openUserId";
        String name = "ownerCarOpenUser";
        String shareName = "user"; //生成到那个服务下
        String shareColumn = "open_user_id";
        String shareParam = "openUserId";

        //业务名称 desc 业务编码名称生成后类名 name 主键 id  需要放到那个服务 shareName
        String newSql = createTableSql.substring(createTableSql.indexOf("(") + 1, createTableSql.lastIndexOf(")"));
        String tableName = createTableSql.substring(createTableSql.indexOf("TABLE") + 5, createTableSql.indexOf("("));
        tableName = tableName.replaceAll("`", "").trim();
        newSql = newSql.replaceAll("\n", "");
        String[] rowSqls = newSql.split(",");
        String[] rowSqls = newSql.split("',");
        JSONObject param = new JSONObject();
        param.put("autoMove", true);
        param.put("desc", "");
        param.put("id", "");
        param.put("name", "");
        param.put("shareColumn", "");
        param.put("shareName", "");
        param.put("shareParam", "");
        param.put("desc", desc);
        param.put("id", id);
        param.put("name", name);
        param.put("shareColumn", shareColumn);
        param.put("shareName", shareName);
        param.put("shareParam", shareParam);
        param.put("tableName", tableName);
        JSONObject paramColumn = new JSONObject();
        JSONArray requireds = new JSONArray();
@@ -50,13 +54,27 @@
            if ("UNIQUE".equals(key)) {
                continue;
            }
            if ("KEY".equals(key)) {
                continue;
            }
            if ("b_id".equals(key)) {
                continue;
            }
            if ("create_time".equals(key)) {
                continue;
            }
            if(rowSql.toLowerCase().contains("not null")){
                required.put("code",StringUtil.lineToHump(key));
                String comment = rowSql.substring(rowSql.indexOf("COMMENT '")+9,rowSql.lastIndexOf("'"));
                required.put("msg",comment+"不能为空");
            if (rowSql.toLowerCase().contains("not null")) {
                required.put("code", StringUtil.lineToHump(key));
                String comment = rowSql.contains("COMMENT") ? rowSql.substring(rowSql.indexOf("COMMENT '") + 9) : StringUtil.lineToHump(key);
                comment = comment.trim();
                if(comment.contains("，")){
                    comment = comment.split("，")[0];
                }
                if(comment.contains(" ")){
                    comment = comment.split(" ")[0];
                }

                required.put("msg", comment + "不能为空");
                requireds.add(required);
            }
            paramColumn.put(StringUtil.lineToHump(key), key);
@@ -64,7 +82,6 @@
        param.put("param", paramColumn);
        param.put("required", requireds);
        System.out.println(param.toJSONString());

    }

}