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

package com.java110.code;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.java110.utils.util.StringUtil;

public class TableToJson {

    //show create table c_orders  用这个语句获取
    public static final String createTableSql = "CREATE TABLE `building_room` (\n" +
            "  `room_id` varchar(30) NOT NULL COMMENT '房屋ID',\n" +
            "  `b_id` varchar(30) NOT NULL COMMENT '业务Id',\n" +
            "  `room_num` varchar(12) NOT NULL COMMENT '房屋编号',\n" +
            "  `unit_id` varchar(30) NOT NULL COMMENT '单元ID',\n" +
            "  `layer` int(11) NOT NULL COMMENT '层数',\n" +
            "  `section` int(11) DEFAULT NULL COMMENT '室',\n" +
            "  `apartment` varchar(20) NOT NULL COMMENT '户型',\n" +
            "  `built_up_area` decimal(6,2) NOT NULL COMMENT '建筑面积',\n" +
            "  `fee_coefficient` decimal(12,2) NOT NULL DEFAULT '1.00' COMMENT '算费系数',\n" +
            "  `user_id` varchar(30) NOT NULL COMMENT '用户ID',\n" +
            "  `remark` varchar(200) DEFAULT NULL COMMENT '备注',\n" +
            "  `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
            "  `status_cd` varchar(2) NOT NULL DEFAULT '0' COMMENT '数据状态，详细参考c_status表，S 保存，0, 在用 1失效',\n" +
            "  `state` varchar(4) NOT NULL COMMENT '房屋状态，如房屋出售等，请查看state 表',\n" +
            "  `community_id` varchar(30) DEFAULT NULL COMMENT '小区ID',\n" +
            "  `room_type` varchar(12) NOT NULL DEFAULT '1010301' COMMENT '房屋类型',\n" +
            "  `room_sub_type` varchar(12) NOT NULL DEFAULT '110' COMMENT '房屋类型 110 住宅房屋，119 办公室 120 宿舍',\n" +
            "  `room_area` decimal(6,2) NOT NULL COMMENT '室内面积',\n" +
            "  `room_rent` decimal(10,2) NOT NULL DEFAULT '0.00' COMMENT '租金',\n" +
            "  UNIQUE KEY `room_id` (`room_id`) USING BTREE,\n" +
            "  UNIQUE KEY `idx_room_id` (`room_id`) USING BTREE,\n" +
            "  KEY `idx_room_b_id` (`b_id`) USING BTREE,\n" +
            "  KEY `i_br_unit_id` (`unit_id`)\n" +
            ")";

    public static void main(String[] args) {

        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(",\\n");
        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("tableName", tableName);
        JSONObject paramColumn = new JSONObject();
        JSONArray requireds = new JSONArray();
        JSONObject required = null;
        String key = "";
        for (String rowSql : rowSqls) {
            required = new JSONObject();
            key = rowSql.trim();
            key = key.substring(0, key.indexOf(" "));
            key = key.replaceAll("`", "");
            if ("UNIQUE".equals(key)) {
                continue;
            }
            if ("KEY".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+"不能为空");
                requireds.add(required);
            }
            paramColumn.put(StringUtil.lineToHump(key), key);
        }
        param.put("param", paramColumn);
        param.put("required", requireds);
        System.out.println(param.toJSONString());

    }

}