建表sql转实体类（含swagger，含lombok）

Love The Way You Lie 2023-10-08 16:14 20阅读 0赞

public static void main(String[] args) throws IOException {
        
            Integer isSwagger = 1;
            Integer isLombok = 1;
            String sqlContent = "CREATE TABLE `report_form_institution_data_source` (\n" +
                "\t`id` BIGINT(20) NOT NULL AUTO_INCREMENT COMMENT '主键',\n" +
                "\t`institution_id` VARCHAR(100) NOT NULL DEFAULT '0' COMMENT '制度id' COLLATE 'utf8_bin',\n" +
                "\t`ldap` VARCHAR(50) NOT NULL DEFAULT 'meidi' COMMENT '员工ldap' COLLATE 'utf8_bin',\n" +
                "\t`encryption_ldap` VARCHAR(200) NOT NULL DEFAULT '0' COMMENT '加密的员工ldap' COLLATE 'utf8_bin',\n" +
                "\t`emp_id` VARCHAR(10) NOT NULL DEFAULT '0' COMMENT '员工empid' COLLATE 'utf8_bin',\n" +
                "\t`implement_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '发布时间',\n" +
                "\t`institution_create_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '制度创建时间',\n" +
                "\t`is_sign` TINYINT(1) NOT NULL DEFAULT '1' COMMENT '1未签收 2签收 3已读',\n" +
                "\t`sign_client_type` TINYINT(1) NOT NULL DEFAULT '0' COMMENT '制度签收客户端类别，h5:1、pc:2',\n" +
                "\t`is_cx` TINYINT(1) NOT NULL DEFAULT '0' COMMENT '是否橙心外包下面的制度 1:是，0:不是',\n" +
                "\t`create_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
                "\t`update_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',\n" +
                "\tPRIMARY KEY (`id`) USING BTREE,\n" +
                "\tUNIQUE INDEX `unique_id` (`ldap`, `institution_id`) USING BTREE\n" +
                ")\n" +
                "COMMENT='报表源数据落地表'\n" +
                "COLLATE='utf8_bin'\n" +
                "ENGINE=InnoDB\n" +
                "AUTO_INCREMENT=7469\n" +
                ";\n";
    
            sqlContent = sqlContent.toLowerCase();
            String tableNames = sqlContent.substring(sqlContent.indexOf("`")+1, sqlContent.indexOf("` ("));
    
            StringBuilder classTop = new StringBuilder("import com.fasterxml.jackson.annotation.JsonFormat;\n" +
                "import java.util.Date;\n" +
                "import javax.persistence.Column;\n" +
                "import javax.persistence.GeneratedValue;\n" +
                "import javax.persistence.GenerationType;\n" +
                "import javax.persistence.Table;\n" +
                "import javax.persistence.Id;\n" +
                "import org.springframework.format.annotation.DateTimeFormat;\n");
            if (1 == isSwagger) {
        
                classTop.insert(0, "import io.swagger.annotations.ApiModel;\n" +
                    "import io.swagger.annotations.ApiModelProperty;\n");
            }
            if (1 == isLombok) {
        
                classTop.append("import lombok.AllArgsConstructor;\n" +
                    "import lombok.Builder;\n" +
                    "import lombok.Data;\n" +
                    "import lombok.NoArgsConstructor;\n" +
                    "@Data\n" +
                    "@Builder\n" +
                    "@AllArgsConstructor\n" +
                    "@NoArgsConstructor\n");
            }
            String stringField = getStringField(sqlContent, 1);
            classTop.append("@Table(name=\""+ tableNames+"\")\n" +
                "public class "+stringField+" {\n");
    
            String fieldNames = sqlContent.substring(sqlContent.indexOf("(")+1, sqlContent.lastIndexOf(")"));
            String[] splitField = fieldNames.split("\n\t");
            StringBuilder fieldName = new StringBuilder();
            for (String field : splitField) {
        
                if (StringUtils.isNotEmpty(field) && field.substring(0,1).equals("`")) {
        
                    if (!field.contains("collate")) {
        
                        fieldName.append("    @ApiModelProperty(\""+ field.substring(field.indexOf("comment '")+9,
                            field.lastIndexOf("',")) + "\")\n");
                    }else {
        
                        fieldName.append("    @ApiModelProperty(\""+ field.substring(field.indexOf("comment '")+9,
                            field.lastIndexOf("' collate")) + "\")\n");
                    }
    
                    fieldName.append("    @Column(name = \"" + field.substring(field.indexOf("`") + 1,
                        field.indexOf("`",field.indexOf("`")+1)) + "\")\n");
                    String[] split = field.split(Constants.BLANK);
                    if (split[1].contains("bigint")) {
        
                        fieldName.append("    private Long " + getStringField(field, 0) + ";\n");
                    } else if (split[1].contains("int")) {
        
                        fieldName.append("    private Integer " + getStringField(field, 0) + ";\n");
                    } else if (split[1].contains("char")) {
        
                        fieldName.append("    private String " + getStringField(field, 0) + ";\n");
                    } else if (split[1].contains("date") || split[1].contains("time")) {
        
                        fieldName.append("    @DateTimeFormat(pattern = \"yyyy-MM-dd hh:mm:ss\")\n" +
                            "    @JsonFormat(timezone = \"GMT+8\", pattern = \"yyyy-MM-dd HH:mm:ss\")\n");
                        fieldName.append("    private Date " + getStringField(field, 0) + ";\n");
                    }
                    fieldName.append("\n");
                }
            }
            System.out.println(classTop.append(fieldName).append("}"));
    
        }
    
        public static String getStringField(String str, Integer isUpperCase) {
        
            String tableNames = str.substring(str.indexOf("`")+1, str.indexOf("`",str.indexOf("`")+1));
            String[] split = tableNames.split(Constants.UNDERLINE);
            StringBuilder tableName = new StringBuilder();
            int index = 0;
            for (String s : split) {
        
                if (isUpperCase == 0 && index == 0) {
        
                    tableName.append(s);
                }else {
        
                    tableName.append(s.substring(0, 1).toUpperCase() + "" + s.substring(1));
                }
                index++;
            }
            return tableName.toString();
        }