修改mybatis-generator-1.3.2源码实现自定义代码生成详解（三）

前两篇对mybatis-generator-core源码进行了修改，在一定程度上做到了方便使用。这一篇先将写个测试程序看看生成的结果，然后用maven assembly将它打包生成一个zip文件。
    新建一个类StartUp，找到org.mybatis.generator.api.ShellRunner这个类，它就是生成文件的入口类，拷贝106-117行代码，再稍微作下修改。
    
    
    generatorConfig.xml在resources目录下面，它里面要定义输出文件的目录，如果不指定具体的路径，默认是在当前文件夹下面生成。
    

    但是呢，这里有个比较蛋疼的地方，mybatis-generator-core默认是不给你生成文件夹的，不信？请看org.mybatis.generator.internal.DefaultShellCallback的第50行，当指定的目录不存在时，就抛出ShellException异常，所以输出文件也就不可能有了。这里我们注释掉51行代码，添加如下代码：
    
    
    现在再来执行StartUp看看，发现在与src、target同级目录下生成了mybatis3目录，里面有相应的pojo与xml，打开之后看看效果，与我们改动代码后预想的结果一致。那么接下来就看看，如何将它打包输出成下面这样的效果。
    

    首先我们新建一个src/main/scripts/run.bat文件，内容如下：
    java -jar mybatis-generator-1.3.2.jar -configfile generatorConfig.xml –overwrite
    pause

    其实就是一个java命令行，里面加的参数都是ShellRunner需要的，加上pause是方便查看输出信息。
    然后修改pom文件，既然最终输出有源文件JAR和编译后的class JAR，所以我们需要增加maven-source-plugin和maven-jar-plugin这两个插件，另外还用到了maven-assembly-plugin插件。这些plugin具体的用法，我就不讲了，我只说下要注意的地方。
    

    注意红色框的内容，表示我将打包的源代码只会是原来的，增加的com.bolo.StartUp是不会包含进去的，而且这也没必要添加进去。
    

    编译之后的打包同样只包含原来的代码，并且一定要在这里指定入口类。classpathPrefix是类目录，为空就是生成在当前目录下。如果写成<classpathPrefix>lib/</classpathPrefix>这样，则表示生成的mybatis-generator-1.3.2.jar会在lib目录下面。
    

    上图descriptor里面的内容表明了assembly.xml的位置，它可以做很多自定义设置。
    

    这里定义的id默认是加到生成的文件名后面的，但我们在上图中设置了<appendAssemblyId>false</appendAssemblyId>，所以，这个all-in-one就不会加到生成的文件名后面了，如果想要这个名字，将false改成true即可。<format>zip</format>表示将生成一个以zip为后缀的压缩包。
    

    我们最终是想把几个文件都放在一个目录下面，所以outputDirectory就不需要再定义其它的名字。通过前面的输出知道，最后只会包含generatorConfig.xml和run.bat这两个文件，其它的配置文件并不是我们需要的，因此这里需要排除它们。
    另外还有两个fileSet是指定源代码的JAR和编译class的JAR，这两个设置和前面类似。
    
    有点补充说明：此例都是以MySQL数据库为例，所以运行是需要mysql的数据库驱动的，但我又不想在输出里面包含这个jar，因此就在generatorConfig.xml文件里面进行了配置，不清楚的可以看看。
    最后的最后，这些我们都做完了，那么就可以打包输出了，使用mvn package或是在Run As -> Maven Build -> Goals里面填写package，点击 Run。
    源码下载： mybatis-generator-1.3.2.zip
    本文为菠萝大象原创，如要转载请注明出处。http://www.blogjava.net/bolo