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

来源：互联网发布：淘宝上买黄牛票犯法吗编辑：程序博客网时间：2024/06/06 08:37

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

0 0