根据类上的注解信息生成创建表的SQL语句

第一步：创建实体类

定义一个实体类，在字段上使用knife4j的接口文档注解@ApiModelProperty注解指定字段名。

package cn.edu.sgu.www.mhxysy.entity.role;
 
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
 
import java.io.Serializable;
import java.time.LocalDate;
 
/**
 * 黑名单
 * @author heyunlin
 * @version 1.0
 */
@Data
@ApiModel
@TableName("black_list")
public class BlackList implements Serializable {
    private static final long serialVersionUID = 18L;
 
    @TableId(value = "id", type = IdType.AUTO)
    @ApiModelProperty(value = "ID")
    private Integer id;
 
    /**
     * 角色ID
     */
    @ApiModelProperty(value = "角色ID")
    private String roleId;
 
    /**
     * 角色名
     */
    @ApiModelProperty(value = "角色名")
    private String roleName;
 
    /**
     * 拉黑理由
     */
    @ApiModelProperty(value = "拉黑理由")
    private String reason;
 
    /**
     * 拉黑时间
     */
    @ApiModelProperty(value = "拉黑时间")
    @JsonFormat(pattern = "yyyy-MM-dd", timezone = "GMT+8")
    private LocalDate operateTime;
}

第二步：创建一个类保存表的信息

package cn.edu.sgu.www.mhxysy.entity.other;
 
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
 
import java.io.Serializable;
 
/**
 * @author heyunlin
 * @version 1.0
 */
@Data
@ApiModel
public class ClassProperty implements Serializable {
    private static final long serialVersionUID = 18L;
 
    /**
     * 字段名称
     */
    @ApiModelProperty(value = "表名称")
    private String name;
 
    /**
     * 字段类型
     */
    @ApiModelProperty(value = "字段类型")
    private String dataType;
 
    /**
     * 字段注释
     */
    @ApiModelProperty(value = "字段注释")
    private String comment;
}

第三步：Java和数据库的类型映射

 创建Java数据类型和表字段类型的对应关系

package cn.edu.sgu.www.mhxysy.base;
 
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
 
/**
 * Java数据类型-字段类型
 * @author heyunlin
 * @version 1.0
 */
public class FieldTypes {
 
    /**
     * 创建一个map，保存Java数据类型和数据库字段类型的映射关系
     */
    private static final Map<Class<?>, String> CLASS_STRING_MAP = new HashMap<>();
 
    static {
        CLASS_STRING_MAP.put(Integer.class, "int(11) unsigned");
        CLASS_STRING_MAP.put(Date.class, "date");
        CLASS_STRING_MAP.put(LocalDate.class, "date");
        CLASS_STRING_MAP.put(LocalDateTime.class, "datetime");
        CLASS_STRING_MAP.put(String.class, "varchar(255)");
    }
 
    public static String getType(Class<?> propertyType) {
        return CLASS_STRING_MAP.get(propertyType);
    }
 
}

第四步：使用反射生成创建表的SQL语句

package cn.edu.sgu.www.mhxysy;
 
import cn.edu.sgu.www.mhxysy.base.FieldTypes;
import cn.edu.sgu.www.mhxysy.entity.other.ClassProperty;
import com.baomidou.mybatisplus.annotation.TableField;
import io.swagger.annotations.ApiModelProperty;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
 
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
 
/**
 * springboot测试类
 * @author heyunlin
 * @version 1.0
 */
@SpringBootTest
class MhxysyTests {
 
    @Test
    void generateTableCreatementSQL() throws ClassNotFoundException {
        String className = "cn.edu.sgu.www.mhxysy.entity.role.BlackList";
 
        /*
         * 表名
         */
        String tableName = toLower(className);
 
        Class<?> type = Class.forName(className);
 
        // 获取类的所有字段信息
        Field[] fields = type.getDeclaredFields();
        List<ClassProperty> list = new ArrayList<>(fields.length);
 
        for (Field field : fields) {
            ClassProperty classProperty = new ClassProperty();
 
            // 字段名
            if (field.isAnnotationPresent(TableField.class)) {
                TableField tableField = field.getAnnotation(TableField.class);
 
                classProperty.setName(tableField.value());
            } else {
                classProperty.setName(toLower(field.getName()));
            }
 
            // 字段类型
            String fieldType = FieldTypes.getType(field.getType());
 
            classProperty.setDataType(fieldType);
 
            // 字段注释
            if (field.isAnnotationPresent(ApiModelProperty.class)) {
                ApiModelProperty apiModelProperty = field.getAnnotation(ApiModelProperty.class);
 
                classProperty.setComment(apiModelProperty.value());
            }
 
            list.add(classProperty);
        }
 
        // 开始拼接SQL语句
        StringBuilder sb = new StringBuilder();
 
        sb.append("create table ").append(tableName).append("(");
 
        for (ClassProperty classProperty : list) {
            if (!"serialVersionUID".equals(classProperty.getName())) {
                sb.append(classProperty.getName()).append(" ");
                sb.append(classProperty.getDataType()).append(" ");
 
                if ("id".equals(classProperty.getName())) {
                    sb.append("primary key ");
 
                    if (classProperty.getDataType().contains("int")) {
                        sb.append("auto_increment ");
                    }
                }
 
                sb.append("comment '").append(classProperty.getComment()).append("', ");
            }
        }
 
        String sql = sb.substring(0, sb.length() - 2) + ")";
 
        System.out.println(sql);
    }
 
    /**
     * 驼峰命名转下划线命名
     * @param str 待转换的字符串
     * @return String
     */
    public static String toLower(String str) {
        // 小写和大写紧挨一起的地方加上分隔符_，然后全部转为小写
        str = str.replaceAll("([a-z])([A-Z])", "$1_$2");
 
        return str.toLowerCase();
    }
 
}