com.qa.framework.library.base.DynamicCompile Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of smart-api-framework Show documentation
Show all versions of smart-api-framework Show documentation
Support web service api automaton test based on testng and httpclient
package com.qa.framework.library.base;
import javax.tools.*;
import javax.tools.JavaCompiler.CompilationTask;
import javax.tools.JavaFileObject.Kind;
import java.io.IOException;
import java.lang.reflect.Method;
import java.net.URI;
import java.util.Arrays;
import java.util.Collections;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* Java字节码的操纵之动态编译
*/
public class DynamicCompile {
/**
* 源代码
*/
private String source;
/**
* 类名称
*/
private String className;
/**
* 编译输出路径
*/
private String outPath = ".";
/**
* 提取包名称
*/
private Pattern packPattern = Pattern.compile("^package\\s+([a-z0-9.]+);");
/**
* 提取类名称
*/
private Pattern classNamePattern = Pattern.compile("class\\s+([^{]+)");
/**
* Constructor
*
* @param source 源代码
* @param outPath 动态编译文件的输出路径
*/
public DynamicCompile(String source, String outPath) {
this.outPath = outPath;
this.setSource(source);
}
/**
* Eval string object.
*
* @param stringToEval the string to eval
* @return the object
* @throws Exception the exception
*/
public static Object evalString(String stringToEval) throws Exception {
String methodName = "Compiler";
DynamicCompile dynamicComiler = new DynamicCompile("package com.qa.util; class AutoCompilerString {public static Object Compiler(){return \"" + stringToEval + "\";}}", "./target/classes");
return dynamicComiler.Invoke(methodName);
}
/**
* Eval calculate object.
*
* @param stringToEval the string to eval
* @return the object
* @throws Exception the exception
*/
public static Object evalCalculate(String stringToEval) throws Exception {
String methodName = "Compiler";
DynamicCompile dynamicComiler = new DynamicCompile("package com.qa.util; class AutoCompilerCalculate {public static Object Compiler(){return " + stringToEval + ";}}", "./target/classes");
return dynamicComiler.Invoke(methodName);
}
/**
* 测试入口
*
* @param args 参数列表
* @throws Exception 抛出异常
*/
public static void main(String[] args) throws Exception {
String stringToEval = "(4+3)*8/2";
String stringToEval2 = "test";
}
/**
* 编译
*
* @return 编译结果 true/false
* @throws Exception 抛出异常信息
*/
private boolean doCompile() throws Exception {
return new InnerCompiler(new URI(className), Kind.SOURCE, this.source).compile();
}
/**
* 调用
*
* @param methodName 方法名称
* @return result 调用结果
*/
@SuppressWarnings("unchecked")
public Object doInvoke(String methodName) {
ClassLoader classLoader = InnerCompiler.class.getClassLoader();
try {
Class classDef = classLoader.loadClass(className);
Method method = classDef.getMethod(methodName);
return method.invoke(null);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 自动调用
*
* @param methodName 方法名称
* @return resultObj 调用结果
* @throws Exception 忽略所有异常（建议调整）
*/
public Object Invoke(String methodName) throws Exception {
Object result = null;
if (this.doCompile()) {
return doInvoke(methodName);
}
return null;
}
/**
* 设定参与编译的源代码
*
* @param source compiled source
*/
public void setSource(String source) {
String tmpName = analyseClassName(source);
this.className = tmpName.trim();
this.source = source;
}
/**
* 解析类名称
*
* @param source 源字符串
* @return className 类名称/空字符串
*/
private String analyseClassName(String source) {
String tmpName = "";
Matcher matcher = packPattern.matcher(source);
if (matcher.find()) {
tmpName = (matcher.group(1)).concat(".");
}
matcher = classNamePattern.matcher(source);
if (matcher.find()) {
tmpName = tmpName.concat(matcher.group(1));
}
return tmpName;
}
/**
* 指定类名称
*
* @param className 类名称
*/
public void setClassName(String className) {
this.className = className;
}
/**
* 设定输出路径
*
* @param outPath 输出路径
*/
public void setOutPath(String outPath) {
this.outPath = outPath;
}
/*
* (non-Javadoc)
* @see java.lang.Object#toString()
*/
@Override
public String toString() {
return "DynamicCompile [className=" +
className +
", source=" +
source +
"]";
}
/**
* 负责自动编译
*
* @author WangYanCheng
* @version 2011 -2-17
*/
class InnerCompiler extends SimpleJavaFileObject {
/**
* content
*/
private String content;
/**
* Contructor
*
* @param uri uri
* @param kind kind
* @param content content
*/
public InnerCompiler(URI uri, Kind kind, String content) {
this(uri, kind);
this.content = content;
}
/**
* Constructor
*
* @param uri 编译源文件路径
* @param kind 文件类型
*/
protected InnerCompiler(URI uri, Kind kind) {
super(uri, kind);
}
/*
* (non-Javadoc)
* @see javax.tools.SimpleJavaFileObject#getCharContent(boolean)
*/
@Override
public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
return this.content;
}
/**
* 编译
*
* @return result 成功编译标记{true|false}
*/
public boolean compile() {
boolean result = false;
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
Iterable fileObject = Collections.singletonList(this);
Iterable options = Arrays.asList("-d", outPath);
CompilationTask task = compiler.getTask(null, fileManager, null, options, null, fileObject);
result = task.call();
return result;
}
}
}