优化代码

java110

2021-12-02 d91329bda26053d9189f345f023da4bd200d3dae

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

@@ -2,41 +2,55 @@

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.java110.core.client.RestTemplate;
import com.java110.db.dao.IQueryServiceDAO;
import com.java110.utils.factory.ApplicationContextFactory;
import com.java110.utils.util.StringUtil;
import org.springframework.beans.BeansException;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.web.client.RestTemplateBuilder;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.http.converter.StringHttpMessageConverter;

import java.nio.charset.Charset;

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 `oa_workflow_data_file` (\n" +
            "  `file_id` varchar(30) NOT NULL PRIMARY KEY COMMENT '主键ID',\n" +
            "  `id` varchar(30) not null COMMENT '对应表的ID',\n" +
            "  `file_name` varchar(512) DEFAULT NULL COMMENT '文件路径',\n" +
            "  `real_file_name` varchar(512) DEFAULT NULL COMMENT '真实文件名',\n" +
            "  `store_id` varchar(30) NOT NULL COMMENT '商户ID',\n" +
            "  `create_user_id` varchar(30) NOT NULL COMMENT '创建人ID',\n" +
            "  `create_user_name` varchar(64) NOT 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表，S 保存，0, 在用 1失效'\n" +
            ")";

    public static void main(String[] args) {

        String desc = "OA附件";
        String id = "fileId";
        String name = "workflowDataFile";
        String shareName = "oa"; //生成到那个服务下
        String shareColumn = "store_id";
        String shareParam = "storeId";
        //业务名称 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 +64,30 @@
            if ("UNIQUE".equals(key)) {
                continue;
            }
            if ("KEY".equals(key)) {
                continue;
            }
            if ("PRIMARY".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 +95,5 @@
        param.put("param", paramColumn);
        param.put("required", requireds);
        System.out.println(param.toJSONString());

    }

}