基于Freemarker的eclipse plugin代码生成器插件开发

最新推荐文章于 2021-09-22 16:59:26 发布

f7anty

最新推荐文章于 2021-09-22 16:59:26 发布

阅读量658

点赞数

分类专栏： freeMarker

freeMarker 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

原文 http://www.cnblogs.com/buaawp/p/4138168.html

固定类型的软件写多了，里面总是有一些复制粘贴改变类名改变量的基础文件，相似程度非常高。作为一名程序员，坚持不多写一行重复代码的精神，写了一个Eclipse的代码生成器插件。插件通过在xml文件中配置的变量信息及模版位置、目标文件位置信息，直接生成目标文件，减少了大量的重复工作。

1.建立一个plug-in with a popup menu工程，引入freemarker.jar，配置popup menu的对应文件扩展名为.coding.xml

2.先写核心的文档生成代码，保证使用main函数可调用。核心的内容是按照freemarker的要求写好模版路径、模板文件名、目标文件路径、目标文件名、文件所使用的字符集、以及包含所有数据的map

 1 public class CodegenUtil
 2 {
 3   public static void genFile(String templatePath, String templateFileName, String targetPath, String targetFileName, String charset, Map paramMap)
 4     throws IOException, TemplateException
 5   {
 6    
 7     File localFile = new File(targetPath, targetFileName);
 8     if (!localFile.exists()) {
 9       if (!localFile.getParentFile().exists())
10         localFile.getParentFile().mkdirs();
11       localFile.createNewFile();
12     }
13     OutputStreamWriter localOutputStreamWriter = new OutputStreamWriter(new FileOutputStream(localFile), charset);
14 
15     Configuration freemarkerConfigration = new Configuration();
16     freemarkerConfigration.setDirectoryForTemplateLoading(new File(templatePath));
17 
18     Template localTemplate = freemarkerConfigration.getTemplate(templateFileName, charset);
19     localTemplate.process(paramMap, localOutputStreamWriter);
20     localOutputStreamWriter.close();
21     
22   }
23 }

3.写Action调用

public void run(IAction action) {		
    
    //读取选定的配置文件	
    IStructuredSelection selection =(IStructuredSelection) this.selection;		

    ConsoleFactory.printToConsole("--------Start Coding--------", true);
    
    for(Object element:selection.toList()){
      File file = (File)element;	
      String fullpath = file.getLocationURI().getPath();
      
      Map<String, String> params;
      List<Map<String,String>>  templateMapList=new ArrayList<Map<String,String>>(); 
      
      try {

        String configfilepath=fullpath.substring(1);
        ConsoleFactory.printToConsole("...load coding config "+configfilepath, true);
        
        params = XmlUtil.getVars(configfilepath);
        templateMapList=XmlUtil.getTemplates(configfilepath);
        
        for(Map<String ,String > templateMap:templateMapList){
          String templateFilePath=templateMap.get(XmlUtil.TEMPLATE_PATH);
          String templateFileName=templateMap.get(XmlUtil.TEMPLATE_NAME);
          String targetFilePath=templateMap.get(XmlUtil.TARGET_PATH);
          String targetFileName=templateMap.get(XmlUtil.TARGET_NAME);
          
          ConsoleFactory.printToConsole("... ... coding ... "+targetFilePath+"\\"+targetFileName, true);

          params.put(XmlUtil.TEMPLATE_PATH, templateFilePath);
          params.put(XmlUtil.TEMPLATE_NAME, templateFileName);
          params.put(XmlUtil.TARGET_PATH, targetFilePath);
          params.put(XmlUtil.TARGET_NAME, targetFileName);					
          
          String charset=params.get(XmlUtil.CHARSET);	
          
          CodegenUtil.genFile(templateFilePath,templateFileName,targetFilePath,targetFileName,charset,params);		

        }			
        
      } catch (Exception e) {
        e.printStackTrace();
      }		
    }

    ConsoleFactory.printToConsole("--------Finish Coding--------", true);
    
  }

4.使用System.out.print所打印的东西在插件的运行环境下是不显示的，ConsoleFactory是我写的一个控制台输出类，调用了Eclipse的控制台，主要用来在使用时给出相关的代码生成提示。

public class ConsoleFactory implements IConsoleFactory {

  private static MessageConsole console=new MessageConsole("",null);
  static boolean exists=false;
  
  @Override
  public void openConsole() {
    showConsole();
  }
  
  private static void showConsole(){
    if(console!=null){
      IConsoleManager manager=ConsolePlugin.getDefault().getConsoleManager();
      IConsole[] existing = manager.getConsoles();
      
      exists=false;
      for(int i=0;i<existing.length;i++){
        if(console==existing[i]){
          exists=true;
        }
      }
      if(!exists){
        manager.addConsoles(new IConsole[]{console});
      }
      
    }
  }
  
  public static void closeConsole(){
    IConsoleManager manager = ConsolePlugin.getDefault().getConsoleManager();
    if(console!=null){
      manager.removeConsoles(new IConsole[]{console});			
    }
  }
  
  public static MessageConsole getConsole(){
    showConsole();
    return console;
  }
  
  public static void printToConsole(String message , boolean activate){
    MessageConsoleStream printer = ConsoleFactory.getConsole().newMessageStream();
    printer.setActivateOnWrite(activate);
    printer.println(message);
  }

}

5.主要内容完毕。使用时先配好xml文件，如下示例。在Eclipse中选中.coding.xml文件，右键菜单 [代码生成器]-->[生成代码]

<?xml version="1.0" encoding="utf-8" ?>
<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
  <variables>
    <variable name="developer" value="PennPeng" />
    <variable name="charset" value="utf-8" />		
    <variable name="class" value="CodeGen" />
    <variable name="name" value="姓名" />
    <variable name="age" value="年龄" />
  </variables>
  
  <templates>
    <template>			
      <variable name="templatepath" value="E:\CodeGenTest\template" />
      <variable name="templatename" value="a.ftl" />
      <variable name="targetpath" value="E:\CodeGenTest" />		
      <variable name="targetname" value="CodeGen.java" />
    </template>
    <template>			
      <variable name="templatepath" value="E:\CodeGenTest\template" />
      <variable name="templatename" value="b.ftl" />
      <variable name="targetpath" value="E:\CodeGenTest" />		
      <variable name="targetname" value="CodeGen2.java" />
    </template>
  </templates>
</config>