JavaCompiler 编译JAVA文件并指定输出路径

新手勿喷！！

前几年有个很牛的同事自己写了个java的编译插件，要比ant快很多。今天自己尝试着写了一个，使用的还是javatool里的JavaCompiler进行编译操作，虽然并没有什么乱用，不想再做深入的研究，但可以给想自己写编译的人提供一下参考。

大体说一下，代码里是使用JavaFileManager去加载需要编译的java文件，编译操作时给ComplitionTask指定javac执行的参数option，使用JavaFileManager也可以指定一些参数，不过不推荐，毕竟option是针对操作不是针对文件。option是个Iterator类型，开始我加入的是这样的："-d d:"，但是一执行就报错了，说不支持，就去各种反编译出来看，原来带参数的操作是不在一起的，要分别加入Iterator，这么设计也是有点醉了。

不多说，代码列出来看看，有点内存问题，不过懒得改了。

CompileUitl.java  可以直接运行

package com.common.util;

import java.io.File;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;

public class CompileUtil {

	private static String KEY_JAVA_FILE = "javaFile";
	private static String KEY_OTHER_FILE = "otherFile";

	/**
	 * compile the class file under the given path:src,and copy the others to the dest path.
	 * the hidden file is not supported
	 * 
	 * @param src the path your class file exist
	 * @param dest  the output path
	 * @param options javac compile option,you can refer to @{com.sun.tools.javac.main.OptionName},if the option has values, just add it to the iterator separately
	 * @param charSet your file encoding,like "UTF-8"
	 * @throws Exception
	 */
	public static void compileAndCopyFiles(String src, String dest, List<String> options,String charSet) throws Exception {
		System.out.println("start compileAndCopyFiles.....");
		long start = System.currentTimeMillis();
		FileUtil.mkDirs(dest);
		Map<String, List<File>> allFiles = getFiles(src);
		List<File> files = allFiles.get(KEY_JAVA_FILE);
		List<File> otherFiles = allFiles.get(KEY_OTHER_FILE);

		JavaCompiler cmp = ToolProvider.getSystemJavaCompiler();
		StandardJavaFileManager fileManager = cmp.getStandardFileManager(null, null, Charset.forName(charSet));

		Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(files);

		cmp.getTask(null, fileManager, null, options, null, compilationUnits).call();
		fileManager.close();

		copyOtherFiles(otherFiles, src, dest);
		System.out.println("end compileAndCopyFiles.Files amount:"+allFiles.size()+".Total spend time : "+(System.currentTimeMillis()-start)/1000+" s");
	}

	private static void copyOtherFiles(List<File> files, String src, String dest) throws Exception{
		for (File file : files) {
			String fileAbsolutePath = file.getAbsolutePath();
			String destPath = dest + fileAbsolutePath.substring(src.length(), fileAbsolutePath.length());
			FileUtil.copy(fileAbsolutePath, destPath);
		}
	}

	private static Map<String, List<File>> getFiles(String src) {
		List<File> files = FileUtil.getAllFiles(src);
		String suffix = ".java";
		Map<String, List<File>> map = new HashMap<String, List<File>>();
		List<File> javaFile = new ArrayList<File>();
		List<File> otherFile = new ArrayList<File>();
		for (File file : files) {
			String fileName = file.getName();
			if (fileName.endsWith(suffix)) {
				javaFile.add(file);
			} else {
				otherFile.add(file);
			}
		}
		map.put(KEY_JAVA_FILE, javaFile);
		map.put(KEY_OTHER_FILE, otherFile);
		return map;
	}
	
	public static void main(String ars[]) throws Exception{
		System.out.println(System.getProperty("user.dir"));
		String src=System.getProperty("user.dir") + File.separator +"src";
		String dest=System.getProperty("user.dir") + File.separator +"classes"+File.separator+"me";
		List<String> options = new ArrayList<String>();
		//options.add("-verbose");
		options.add("-d");
		options.add(dest);
		CompileUtil.compileAndCopyFiles(src, dest, options, "GBK");
				
	}

}

FileUtil.java  辅助类

package com.common.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;

public class FileUtil {

	public static List<File> getAllFiles(String path) {
		List<File> files = new ArrayList<File>();
		File file = new File(path);
		if (file.exists() && !file.isHidden()) {
			if (file.isDirectory()) {
				File[] fs = file.listFiles();
				for (File f : fs) {
					if (!file.isHidden()) {
						if (f.isDirectory()) {
							files.addAll(getAllFiles(path + File.separator
									+ f.getName()));
						}
						if (f.isFile()) {
							files.add(f);
						}
					}
				}

			} else if (file.isFile()) {
				files.add(file);
			}
		}
		return files;
	}

	public static void copy(String from, String to) throws Exception {
		InputStream in = new FileInputStream(from);
		OutputStream out = new FileOutputStream(to);

		byte[] buff = new byte[1024];
		int len = 0;
		while ((len = in.read(buff)) != -1) {
			out.write(buff, 0, len);
		}
		in.close();
		out.close();
	}

	public static void mkDirs(String path) {
		File destFile = new File(path);
		if (!destFile.exists()) {
			destFile.mkdirs();
		}
	}

}