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

固定类型的软件写多了，里面总是有一些复制粘贴改变类名改变量的基础文件，相似程度非常高。作为一名程序员，坚持不多写一行重复代码的精神，写了一个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>

 

还有很多需要扩展完善的地方，比如数据库的自动支持、各类型文档的生成。

项目地址 https://github.com/buaawp/codegen