基于Spring Boot+Vue框架的学生综合素质测评系统设计源码

package com.example.utils.generator; import cn.hutool.core.io.FileUtil; import cn.hutool.core.util.StrUtil; import cn.hutool.db.Db; import cn.hutool.db.Entity; import com.alibaba.druid.pool.DruidDataSource; import org.apache.commons.compress.utils.Lists; import java.sql.SQLException; import java.util.HashMap; import java.util.List; import java.util.Map; /** * mybatis代码生成器 * * @date 2021-3-10 */ public class MybatisCodeGenerator { private static final DruidDataSource ds = new DruidDataSource(); private static final String schemaName = "x-admin"; // 数据库名称，必填 private static final String[] table = {"book", ""}; // 必填，第一个是表名，第二个是实体类的名字 private static final String modelName = "图书"; // 必填 //=========================================以上内容必填===================================================// static { // 必填 ds.setUrl("jdbc:mysql://localhost:3306/" + schemaName + "?useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&useSSL=false&serverTimezone=GMT%2b8"); ds.setUsername("root"); ds.setPassword("123456"); } private static final String BaseFilePath = System.getProperty("user.dir") + "/src/main/java/com/example/"; private static final String space4 = " "; private static final String space6 = space4 + " "; private static final String space8 = space4 + space4; private static final String space12 = space4 + space8; public static void main(String[] args) throws Exception { if (StrUtil.isBlank(table[0])) { System.err.println("请完善配置"); return; } String entityName = getEntityName(); // 创建entity createEntity(table[0], entityName); createMapper(entityName); createService(entityName); createController(entityName, table[0]); createXml(entityName); // html createVueHtml(entityName, table[0]); } /** * 获取数据库对象 * * @param tableName * @return * @throws SQLException */ static List<TableColumn> getTableColumns(String tableName) throws SQLException { String sql = "SELECT table_name,column_name,data_type, column_comment FROM information_schema.COLUMNS WHERE table_schema = ? and table_name = ?"; List<Entity> user = Db.use(ds).query(sql, schemaName, tableName); List<TableColumn> columnList = Lists.newArrayList(); for (Entity entity : user) { TableColumn tableColumn = new TableColumn(); tableColumn.setTableName(entity.getStr("table_name")); tableColumn.setColumnName(entity.getStr("column_name")); tableColumn.setDataType(convertDataType(entity.getStr("data_type"))); tableColumn.setColumnComment(entity.getStr("column_comment")); columnList.add(tableColumn); } return columnList; } /** * 生成entity * * @throws SQLException */ static void createEntity(String tableName, String entityName) throws SQLException { List<TableColumn> columnList = getTableColumns(tableName); StringBuilder entityHeadBuild = StrUtil.builder() .append("package com.example.entity;\n\n") .append("import lombok.Data;\n") .append("import com.baomidou.mybatisplus.annotation.TableName;\n") .append("import com.baomidou.mybatisplus.annotation.IdType;\n") .append("import com.baomidou.mybatisplus.extension.activerecord.Model;\n") .append("import com.baomidou.mybatisplus.annotation.TableId;\n\n"); StringBuilder entityBodyBuild = StrUtil.builder() .append("@Data\n") .append("@TableName(\"").append(tableName).append("\")\n") .append("public class ").append(entityName).append(" extends Model<").append(entityName).append("> {\n") .append(space4).append("/**\n") .append(space6).append("*").append(" 主键\n") .append(space6).append("*/\n") .append(space4).append("@TableId(value = \"id\", type = IdType.AUTO)\n") .append(space4).append("private Long id;\n\n"); for (TableColumn tableColumn : columnList) { String columnName = tableColumn.getColumnName(); if (!"id".equals(columnName)) { // 注释 if (StrUtil.isNotBlank(tableColumn.getColumnComment())) { entityBodyBuild .append(space4).append("/**\n") .append(space6).append("* ").append(tableColumn.getColumnComment()).append(" \n") .append(space6).append("*/\n"); } entityBodyBuild.append(space4).append("private ").append(tableColumn.getDataType()).append(" ").append(StrUtil.toCamelCase(columnName)).append(";\n\n"); } } // 查看是否需要额外导入包 boolean dateExists = columnList.stream().anyMatch(tableColumn -> tableColumn.getDataType().equals("Date")); if (dateExists) { entityHeadBuild.append("import java.util.Date;\n"); } boolean decimalExists = columnList.stream().anyMatch(tableColumn -> tableColumn.getDataType().equals("BigDecimal")); if (decimalExists) { entityHeadBuild.append("import java.math.BigDecimal;\n"); } entityHeadBuild.append("\n"); entityBodyBuild.append("}"); FileUtil.writeString(entityHeadBuild.append(entityBodyBuild).toString(), BaseFilePath + "/entity/" + entityName + ".java", "UTF-8"); System.out.println(entityName + "Entity生成成功！"); } /** * 生成mapper */ static void createMapper(String entityName) { Map<String, Object> map = new HashMap<>(); map.put("entityName", entityName); String format = StrUtil.format(FileUtil.readUtf8String(BaseFilePath + "/utils/generator/template/mapper.template"), map); FileUtil.writeString(format, BaseFilePath + "/mapper/" + entityName + "Mapper" + ".java", "UTF-8"); System.out.println(entityName + "Mapper生成成功！"); } /** * 生成service */ static void createService(String entityName) { String lowerName = entityName.substring(0, 1).toLowerCase() + entityName.substring(1); Map<String, Object> map = new HashMap<>(); map.put("entityName", entityName); map.put("lowerName", lowerName); String format = StrUtil.format(FileUtil.readUtf8String(BaseFilePath + "/utils/generator/template/service.template"), map); FileUtil.writeString(format, BaseFilePath + "/service/" + entityName + "Service" + ".java", "UTF-8"); System.out.println(entityName + "Service生成成功！"); } /** * 生成controller * * @param entityName */ static void createController(String entityName, String tableName) throws SQLException { String lowerEntityName = entityName.substring(0, 1).toLowerCase() + entityName.substring(1); StringBuilder exportCode = new StringBuilder(); StringBuilder importCode = new StringBuilder(); List<TableColumn> tableColumns = getTableColumns(tableName); int count = 1; for (TableColumn tableColumn : tableColumns) { exportCode.append(space12 + "row.put(\"" + tableColumn.getColumnComment() + "\", obj.get" + toFirstUpper(tableColumn.getColumnName()) + "());\n"); if (tableColumn.getColumnName().equals("id")) { continue; } String value; if("Integer".equals(tableColumn.getDataType())) { value = "Integer.valueOf((String) row.get(" + count++ + "))"; } else if("Double".equals(tableColumn.getDataType()))