天天看點
傳回

利用Freemarker實作表到功能界面的一鍵生成

      利用Freemarker産生項目所需的常用檔案，可以根據一個表産生對應的ibtais的java及xml檔案，及前台的JSP檔案，這樣可以大大的提高普通程式編寫人員的開發效率，及從表到界面及CURD功能利用這個工程可以直接生成好，下面提供下最核心的檔案産生類，代碼如下： 

     
  1.  package com.hundsun.pms.generator.util;        
    2.           
  2.  import com.hundsun.pms.generator.api.Constants;        
    3.       
  3.  import com.hundsun.pms.generator.model.page.Page;        
    4.       
  4.  import freemarker.template.Configuration;        
    5.       
  5.  import freemarker.template.DefaultObjectWrapper;        
    6.       
  6.  import freemarker.template.Template;        
    7.       
  7.  import freemarker.template.TemplateException;        
    8.       
  8.  import org.apache.commons.lang.StringUtils;        
    9.       
  9.  import org.springframework.util.ResourceUtils;        
    10.           
  10.  import java.io.*;        
    11.       
  11.  import java.net.URL;        
    12.       
  12.  import java.util.Locale;        
    13.       
  13.  import java.util.Map;        
    14.           
  14.  /**        
    15.       
  15.   * Created by IntelliJ IDEA.        
    16.       
  16.   * User: pangj        
    17.       
  17.   * Date: 2012-3-6        
    18.       
  18.   * Time: 10:05:26        
    19.       
  19.   * 檔案操作工具類        
    20.       
  20.   */        
    21.       
  21.  public class FreemarkerUtil {        
    22.           
  22.      public static String PROJECT_ROOT = "projectRoot";        
    23.           
  23.      public static String PACKAGE_NAME = "packageName";        
    24.           
  24.      public static String CLASS_SIMPLE_NAME = "classSimpleName";        
    25.           
  25.      public static String MODULES_PACKAGE = "modulesPackage";        
    26.           
  26.      public static String METHOD_NAME = "methodName";        
    27.           
  27.      //模闆根目錄        
    28.       
  28.      private static final String templateRootWithClasspath = "classpath*:com/hundsun/pms/generator/resource/template";        
    29.           
  29.      private static final String templateRoot = "com/hundsun/pms/generator/resource/template";        
    30.       
  30.      //工程根目錄        
    31.       
  31.      private static String projectRoot;        
    32.       
  32.      //包名        
    33.       
  33.      private static String packageName;        
    34.       
  34.      //服務所在的包路徑        
    35.       
  35.      private static String modulesPackage;        
    36.           
  36.      private static String classSimpleName;        
    37.           
  37.      private static String templateName;        
    38.           
  38.      private static String methodName;        
    39.           
  39.      public static final String FILE_BO_NAME_SUFFIX = "Bo.java";        
    40.           
  40.      public static final String FILE_BOIMPL_NAME_SUFFIX = "BoImpl.java";        
    41.           
  41.      public static final String FILE_DAO_NAME_SUFFIX = "Dao.java";        
    42.           
  42.      public static final String FILE_DAOIMPL_NAME_SUFFIX = "DaoImpl.java";        
    43.           
  43.      public static final String FILE_DTO_NAME_SUFFIX = ".java";        
    44.           
  44.      public static final String FILE_SQL_MAP_SUFFIX = ".xml";        
    45.           
  45.      public static final String FILE_SERVICE_SUFFIX = ".service";        
    46.           
  46.      public static final String FILE_COMPONENT_SUFFIX = ".component";        
    47.           
  47.      public static final String FILE_MODULE_NAME = "module.xml";        
    48.           
  48.      public static final String FILE_PAGE_SUFFIX = ".jsp";        
    49.               
  49.      /**        
    50.       
  50.       * 資料綁定至模闆        
    51.       
  51.       *        
    52.       
  52.       * @param tpName 模闆名        
    53.       
  53.       * @param param  參數        
    54.       
  54.       * @return        
    55.       
  55.       */        
    56.       
  56.      public static void process(String tpName, Map<String, Object> param) {        
    57.       
  57.          init(tpName, param);        
    58.       
  58.          Writer out = null;        
    59.       
  59.          try {        
    60.       
  60.              Template temp = getTemplate(templateName);        
    61.       
  61.              String fileDir = getFileDir();        
    62.       
  62.              File dir = new File(fileDir);        
    63.       
  63.              if (!dir.exists()) {        
    64.       
  64.                  if (!dir.mkdirs()) {        
    65.       
  65.                      MsgUtil.print("建立目錄：" + fileDir + "失敗");        
    66.       
  66.                      return;        
    67.       
  67.                  }        
    68.       
  68.              }        
    69.       
  69.              File f = new File(fileDir + getFileName());        
    70.       
  70.              out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(f), "utf-8"));        
    71.       
  71.              try {        
    72.       
  72.                  temp.setEncoding("utf-8");        
    73.       
  73.                  temp.process(param, out);        
    74.       
  74.                  MsgUtil.print("檔案：" + getFileDir() + getFileName() + "生成完成！");        
    75.       
  75.              } catch (TemplateException e) {        
    76.       
  76.                  MsgUtil.print("檔案：" + getFileDir() + getFileName() + "生成失敗，請檢查原因！");        
    77.       
  77.                  MsgUtil.print(e.getMessage());        
    78.       
  78.              }        
    79.       
  79.              out.flush();        
    80.       
  80.              out.close();        
    81.       
  81.          } catch (IOException e) {        
    82.       
  82.              e.printStackTrace();        
    83.       
  83.              MsgUtil.print(e.getMessage());        
    84.       
  84.          }        
    85.       
  85.      }        
    86.           
  86.      /**        
    87.       
  87.       * 生成頁面        
    88.       
  88.       *        
    89.       
  89.       * @param tpName 模闆名        
    90.       
  90.       * @param jsp    jsp頁面全路徑        
    91.       
  91.       * @param page   頁面對象        
    92.       
  92.       */        
    93.       
  93.      public static void process(String tpName, String jsp, Page page) {        
    94.       
  94.          Writer writer = null;        
    95.       
  95.          try {        
    96.       
  96.              Template tmp = FreemarkerUtil.getTemplate(tpName);        
    97.       
  97.              String dir = jsp.substring(0, jsp.lastIndexOf("\\"));        
    98.       
  98.              File fdir = new File(dir);        
    99.       
  99.              if (!fdir.exists()) {        
    100.       
  100.                  if (!fdir.mkdirs()) {        
    101.       
  101.                      MsgUtil.print("建立目錄：" + fdir + "失敗");        
    102.       
  102.                      return;        
    103.       
  103.                  }        
    104.       
  104.              }        
    105.       
  105.              File file = new File(jsp);        
    106.       
  106.              writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));        
    107.       
  107.              tmp.setEncoding("utf-8");        
    108.       
  108.              tmp.process(page, writer);        
    109.       
  109.              writer.flush();        
    110.       
  110.              writer.close();        
    111.       
  111.              MsgUtil.print("頁面：" + jsp + " 生成成功!");        
    112.       
  112.          } catch (TemplateException e) {        
    113.       
  113.              e.printStackTrace();        
    114.       
  114.              MsgUtil.print(e.getMessage());        
    115.       
  115.          } catch (IOException e) {        
    116.       
  116.              e.printStackTrace();        
    117.       
  117.              MsgUtil.print(e.getMessage());        
    118.       
  118.          }        
    119.       
  119.      }        
    120.           
  120.      public static Template getTemplate(String templateName) {        
    121.       
  121.          Configuration cfg = new Configuration();        
    122.       
  122.          Template temp = null;        
    123.       
  123.          URL path = FreemarkerUtil.class.getClassLoader().getSystemResource("");        
    124.       
  124.          File tmpRootFile = null;        
    125.       
  125.          if (path == null) {        
    126.       
  126.              tmpRootFile = getResourceURL("resource/template");        
    127.       
  127.          } else {        
    128.       
  128.              String tmpRoot = path + templateRoot;        
    129.       
  129.              tmpRoot = tmpRoot.replaceAll("file:/", "");        
    130.       
  130.              tmpRootFile = new File(tmpRoot);        
    131.       
  131.          }        
    132.       
  132.          if (tmpRootFile == null) {        
    133.       
  133.              throw new RuntimeException("無法取得模闆根路徑！");        
    134.       
  134.          }        
    135.       
  135.          try {        
    136.       
  136.              cfg.setDefaultEncoding("utf-8");        
    137.       
  137.              cfg.setOutputEncoding("utf-8");        
    138.       
  138.              cfg.setDirectoryForTemplateLoading(tmpRootFile);        
    139.       
  139.              /*cfg.setDirectoryForTemplateLoading(getResourceURL());*/        
    140.       
  140.              cfg.setObjectWrapper(new DefaultObjectWrapper());        
    141.       
  141.              temp = cfg.getTemplate(templateName);        
    142.       
  142.          } catch (IOException e) {        
    143.       
  143.              e.printStackTrace();        
    144.       
  144.              MsgUtil.print(e.getMessage());        
    145.       
  145.          }        
    146.       
  146.          return temp;        
    147.       
  147.      }        
    148.           
  148.      public static File getResourceURL(String templatePath) {        
    149.       
  149.          try {        
    150.       
  150.              URL url = ResourceUtils.getURL(templateRootWithClasspath);        
    151.       
  151.              String path = url.getPath();        
    152.       
  152.              path = path.replace(templateRootWithClasspath, templatePath);        
    153.       
  153.              MsgUtil.print("File path is:"+path);        
    154.       
  154.              path = StringUtils.replace(path, "%20", " ");        
    155.       
  155.              return new File(path);        
    156.       
  156.          } catch (FileNotFoundException e) {        
    157.       
  157.              throw new RuntimeException("不能擷取路徑：" + templateRootWithClasspath);        
    158.       
  158.          }        
    159.       
  159.      }        
    160.           
  160.      private static void init(String tpName, Map<String, Object> param) {        
    161.       
  161.          projectRoot = (String) param.get(PROJECT_ROOT);        
    162.       
  162.          packageName = (String) param.get(PACKAGE_NAME);        
    163.       
  163.          classSimpleName = (String) param.get(CLASS_SIMPLE_NAME);        
    164.       
  164.          modulesPackage = (String) param.get(MODULES_PACKAGE);        
    165.       
  165.          methodName = (String) param.get(METHOD_NAME);        
    166.       
  166.          templateName = tpName;        
    167.       
  167.      }        
    168.           
  168.      private static String getFileName() {        
    169.       
  169.          if (Constants.TEMPLATE_BO.equals(templateName)) {        
    170.       
  170.              return classSimpleName + FILE_BO_NAME_SUFFIX;        
    171.       
  171.          } else if (Constants.TEMPLATE_BOIMPL.equals(templateName)) {        
    172.       
  172.              return classSimpleName + FILE_BOIMPL_NAME_SUFFIX;        
    173.       
  173.          } else if (Constants.TEMPLATE_DAO.equals(templateName)) {        
    174.       
  174.              return classSimpleName + FILE_DAO_NAME_SUFFIX;        
    175.       
  175.          } else if (Constants.TEMPLATE_DAOIMPL.equals(templateName)) {        
    176.       
  176.              return classSimpleName + FILE_DAOIMPL_NAME_SUFFIX;        
    177.       
  177.          } else if (Constants.TEMPLATE_DTO.equals(templateName)) {        
    178.       
  178.              return classSimpleName + FILE_DTO_NAME_SUFFIX;        
    179.       
  179.          } else if (Constants.TEMPLATE_SQLMAP.equals(templateName)) {        
    180.       
  180.              return classSimpleName + FILE_SQL_MAP_SUFFIX;        
    181.       
  181.          } else if (Constants.TEMPLATE_SERVICE.equals(templateName)) {        
    182.       
  182.              return methodName + FILE_SERVICE_SUFFIX;        
    183.       
  183.          } else if (Constants.TEMPLATE_COMPONENT.equals(templateName)) {        
    184.       
  184.              return NamingRuleConvert.firstLetterToLowerCase(classSimpleName) + FILE_COMPONENT_SUFFIX;        
    185.       
  185.          } else if (Constants.TEMPLATE_CS.equals(templateName) || Constants.TEMPLATE_DS.equals(templateName) ||        
    186.       
  186.                  Constants.TEMPLATE_MODULE.equals(templateName)) {        
    187.       
  187.              return FILE_MODULE_NAME;        
    188.       
  188.          }        
    189.       
  189.          return null;        
    190.       
  190.      }        
    191.           
  191.      private static String getFileDir() {        
    192.       
  192.          String path = null;        
    193.       
  193.          if (packageName != null)        
    194.       
  194.              path = packageName.replaceAll("\\.", "/");        
    195.       
  195.          String servicePath = null;        
    196.       
  196.          if (!StringUtils.isEmpty(modulesPackage)) {        
    197.       
  197.              servicePath = modulesPackage.replaceAll("\\.", "/");        
    198.       
  198.          }        
    199.       
  199.          if (Constants.TEMPLATE_BO.equals(templateName)) {        
    200.       
  200.              return projectRoot + Constants.SRC_DIR + path + "/bo/";        
    201.       
  201.          } else if (Constants.TEMPLATE_BOIMPL.equals(templateName)) {        
    202.       
  202.              return projectRoot + Constants.SRC_DIR + path + "/bo/impl/";        
    203.       
  203.          } else if (Constants.TEMPLATE_DAO.equals(templateName)) {        
    204.       
  204.              return projectRoot + Constants.SRC_DIR + path + "/dao/";        
    205.       
  205.          } else if (Constants.TEMPLATE_DAOIMPL.equals(templateName)) {        
    206.       
  206.              return projectRoot + Constants.SRC_DIR + path + "/dao/impl/";        
    207.       
  207.          } else if (Constants.TEMPLATE_DTO.equals(templateName)) {        
    208.       
  208.              return projectRoot + Constants.SRC_DIR + path + "/dto/";        
    209.       
  209.          } else if (Constants.TEMPLATE_SQLMAP.equals(templateName)) {        
    210.       
  210.              return projectRoot + Constants.SRC_DIR + path + "/dao/maps/";        
    211.       
  211.          } else if (Constants.TEMPLATE_SERVICE.equals(templateName) || (Constants.TEMPLATE_CS.equals(templateName))) {        
    212.       
  212.              return projectRoot + "/modules/" + servicePath + "/cs/";        
    213.       
  213.          } else if (Constants.TEMPLATE_COMPONENT.equals(templateName) || Constants.TEMPLATE_DS.equals(templateName)) {        
    214.       
  214.              return projectRoot + "/modules/" + servicePath + "/ds/";        
    215.       
  215.          } else if (Constants.TEMPLATE_MODULE.equals(templateName)) {        
    216.       
  216.              return projectRoot + "/modules/" + servicePath + "/";        
    217.       
  217.          }        
    218.       
  218.          return null;        
    219.       
  219.      }        
    220.       
  220.  }        
    221.
開發 表 工程 執行檔案 程式編寫
上一篇: Gird自動選中特定的行
下一篇: 驚天大謀略 二

繼續閱讀