Example #1
0
  /**
   * 生成到文件
   *
   * @param tpl
   * @param model
   * @param replaceFile
   * @return
   */
  public static String generateToFile(
      GenTemplate tpl, Map<String, Object> model, boolean isReplaceFile) {
    // 获取生成文件
    String fileName =
        Global.getProjectPath()
            + File.separator
            + StringUtils.replaceEach(
                FreeMarkers.renderString(tpl.getFilePath() + "/", model),
                new String[] {"//", "/", "."},
                new String[] {File.separator, File.separator, File.separator})
            + FreeMarkers.renderString(tpl.getFileName(), model);
    logger.debug(" fileName === " + fileName);

    // 获取生成文件内容
    String content = FreeMarkers.renderString(StringUtils.trimToEmpty(tpl.getContent()), model);
    logger.debug(" content === \r\n" + content);

    // 如果选择替换文件,则删除原文件
    if (isReplaceFile) {
      FileUtils.deleteFile(fileName);
    }

    // 创建并写入文件
    if (FileUtils.createFile(fileName)) {
      FileUtils.writeToFile(fileName, content, true);
      logger.debug(" file create === " + fileName);
      return "生成成功:" + fileName + "<br/>";
    } else {
      logger.debug(" file extents === " + fileName);
      return "文件已存在:" + fileName + "<br/>";
    }
  }
Example #2
0
 /**
  * 根据分类获取模板列表
  *
  * @param config
  * @param genScheme
  * @param isChildTable 是否是子表
  * @return
  */
 public static List<GenTemplate> getTemplateList(
     GenConfig config, String category, boolean isChildTable) {
   List<GenTemplate> templateList = Lists.newArrayList();
   if (config != null && config.getCategoryList() != null && category != null) {
     for (GenCategory e : config.getCategoryList()) {
       if (category.equals(e.getValue())) {
         List<String> list = null;
         if (!isChildTable) {
           list = e.getTemplate();
         } else {
           list = e.getChildTableTemplate();
         }
         if (list != null) {
           for (String s : list) {
             if (StringUtils.startsWith(s, GenCategory.CATEGORY_REF)) {
               templateList.addAll(
                   getTemplateList(
                       config, StringUtils.replace(s, GenCategory.CATEGORY_REF, ""), false));
             } else {
               GenTemplate template = fileToObject(s, GenTemplate.class);
               if (template != null) {
                 templateList.add(template);
               }
             }
           }
         }
         break;
       }
     }
   }
   return templateList;
 }
Example #3
0
  /**
   * 获取模板路径
   *
   * @return
   */
  public static String getTemplatePath() {
    try {
      File file = new DefaultResourceLoader().getResource("").getFile();
      if (file != null) {
        return file.getAbsolutePath()
            + File.separator
            + StringUtils.replaceEach(
                GenUtils.class.getName(),
                new String[] {"util." + GenUtils.class.getSimpleName(), "."},
                new String[] {"template", File.separator});
      }
    } catch (Exception e) {
      logger.error("{}", e);
    }

    return "";
  }
Example #4
0
  /**
   * 初始化列属性字段
   *
   * @param genTable
   */
  public static void initColumnField(GenTable genTable) {
    for (GenTableColumn column : genTable.getColumnList()) {

      // 如果是不是新增列,则跳过。
      if (StringUtils.isNotBlank(column.getId())) {
        continue;
      }

      // 设置字段说明
      if (StringUtils.isBlank(column.getComments())) {
        column.setComments(column.getName());
      }

      // 设置java类型
      if (StringUtils.startsWithIgnoreCase(column.getJdbcType(), "CHAR")
          || StringUtils.startsWithIgnoreCase(column.getJdbcType(), "VARCHAR")
          || StringUtils.startsWithIgnoreCase(column.getJdbcType(), "NARCHAR")) {
        column.setJavaType("String");
      } else if (StringUtils.startsWithIgnoreCase(column.getJdbcType(), "DATETIME")
          || StringUtils.startsWithIgnoreCase(column.getJdbcType(), "DATE")
          || StringUtils.startsWithIgnoreCase(column.getJdbcType(), "TIMESTAMP")) {
        column.setJavaType("java.util.Date");
        column.setShowType("dateselect");
      } else if (StringUtils.startsWithIgnoreCase(column.getJdbcType(), "BIGINT")
          || StringUtils.startsWithIgnoreCase(column.getJdbcType(), "NUMBER")) {
        // 如果是浮点型
        String[] ss =
            StringUtils.split(StringUtils.substringBetween(column.getJdbcType(), "(", ")"), ",");
        if (ss != null && ss.length == 2 && Integer.parseInt(ss[1]) > 0) {
          column.setJavaType("Double");
        }
        // 如果是整形
        else if (ss != null && ss.length == 1 && Integer.parseInt(ss[0]) <= 10) {
          column.setJavaType("Integer");
        }
        // 长整形
        else {
          column.setJavaType("Long");
        }
      }

      // 设置java字段名
      column.setJavaField(StringUtils.toCamelCase(column.getName()));

      // 是否是主键
      column.setIsPk(genTable.getPkList().contains(column.getName()) ? "1" : "0");

      // 插入字段
      column.setIsInsert("1");

      // 编辑字段
      if (!StringUtils.equalsIgnoreCase(column.getName(), "id")
          && !StringUtils.equalsIgnoreCase(column.getName(), "create_by")
          && !StringUtils.equalsIgnoreCase(column.getName(), "create_date")
          && !StringUtils.equalsIgnoreCase(column.getName(), "del_flag")) {
        column.setIsEdit("1");
      }

      // 列表字段
      if (StringUtils.equalsIgnoreCase(column.getName(), "name")
          || StringUtils.equalsIgnoreCase(column.getName(), "title")
          || StringUtils.equalsIgnoreCase(column.getName(), "remarks")
          || StringUtils.equalsIgnoreCase(column.getName(), "update_date")) {
        column.setIsList("1");
      }

      // 查询字段
      if (StringUtils.equalsIgnoreCase(column.getName(), "name")
          || StringUtils.equalsIgnoreCase(column.getName(), "title")) {
        column.setIsQuery("1");
      }

      // 查询字段类型
      if (StringUtils.equalsIgnoreCase(column.getName(), "name")
          || StringUtils.equalsIgnoreCase(column.getName(), "title")) {
        column.setQueryType("like");
      }

      // 设置特定类型和字段名

      // 用户
      if (StringUtils.startsWithIgnoreCase(column.getName(), "user_id")) {
        column.setJavaType(User.class.getName());
        column.setJavaField(column.getJavaField().replaceAll("Id", ".id|name"));
        column.setShowType("userselect");
      }
      // 部门
      else if (StringUtils.startsWithIgnoreCase(column.getName(), "office_id")) {
        column.setJavaType(Office.class.getName());
        column.setJavaField(column.getJavaField().replaceAll("Id", ".id|name"));
        column.setShowType("officeselect");
      }
      // 创建者、更新者
      else if (StringUtils.startsWithIgnoreCase(column.getName(), "create_by")
          || StringUtils.startsWithIgnoreCase(column.getName(), "update_by")) {
        column.setJavaType(User.class.getName());
        column.setJavaField(column.getJavaField() + ".id");
      }
      // 创建时间、更新时间
      else if (StringUtils.startsWithIgnoreCase(column.getName(), "create_date")
          || StringUtils.startsWithIgnoreCase(column.getName(), "update_date")) {
        column.setShowType("dateselect");
      }
      // 备注、内容
      else if (StringUtils.equalsIgnoreCase(column.getName(), "remarks")
          || StringUtils.equalsIgnoreCase(column.getName(), "content")) {
        column.setShowType("textarea");
      }
      // 父级ID
      else if (StringUtils.equalsIgnoreCase(column.getName(), "parent_id")) {
        column.setJavaType("This");
        column.setJavaField("parent.id|name");
        column.setShowType("treeselect");
      }
      // 所有父级ID
      else if (StringUtils.equalsIgnoreCase(column.getName(), "parent_ids")) {
        column.setQueryType("like");
      }
      // 删除标记
      else if (StringUtils.equalsIgnoreCase(column.getName(), "del_flag")) {
        column.setShowType("radiobox");
        column.setDictType("del_flag");
      }
    }
  }
Example #5
0
  /**
   * 获取数据模型
   *
   * @param genScheme
   * @param genTable
   * @return
   */
  public static Map<String, Object> getDataModel(GenScheme genScheme) {
    Map<String, Object> model = Maps.newHashMap();

    model.put("packageName", StringUtils.lowerCase(genScheme.getPackageName()));
    model.put(
        "lastPackageName", StringUtils.substringAfterLast((String) model.get("packageName"), "."));
    model.put("moduleName", StringUtils.lowerCase(genScheme.getModuleName()));
    model.put("subModuleName", StringUtils.lowerCase(genScheme.getSubModuleName()));
    model.put("className", StringUtils.uncapitalize(genScheme.getGenTable().getClassName()));
    model.put("ClassName", StringUtils.capitalize(genScheme.getGenTable().getClassName()));

    model.put("functionName", genScheme.getFunctionName());
    model.put("functionNameSimple", genScheme.getFunctionNameSimple());
    model.put(
        "functionAuthor",
        StringUtils.isNotBlank(genScheme.getFunctionAuthor())
            ? genScheme.getFunctionAuthor()
            : UserUtils.getUser().getName());
    model.put("functionVersion", DateUtils.getDate());

    model.put(
        "urlPrefix",
        model.get("moduleName")
            + (StringUtils.isNotBlank(genScheme.getSubModuleName())
                ? "/" + StringUtils.lowerCase(genScheme.getSubModuleName())
                : "")
            + "/"
            + model.get("className"));
    model.put(
        "viewPrefix", // StringUtils.substringAfterLast(model.get("packageName"),".")+"/"+
        model.get("urlPrefix"));
    model.put(
        "permissionPrefix",
        model.get("moduleName")
            + (StringUtils.isNotBlank(genScheme.getSubModuleName())
                ? ":" + StringUtils.lowerCase(genScheme.getSubModuleName())
                : "")
            + ":"
            + model.get("className"));

    model.put("dbType", Global.getConfig("jdbc.type"));

    model.put("table", genScheme.getGenTable());

    return model;
  }