add relation by testing run

Author: 孙健

src/main/java/org/nlpcn/jcoder/util/Testing.java

@@ -4,6 +4,7 @@
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.lang.reflect.Field;
+import java.lang.reflect.Method;
 import java.nio.file.FileVisitResult;
 import java.nio.file.Files;
 import java.nio.file.Path;
@@ -22,6 +23,9 @@
 import org.nlpcn.commons.lang.util.FileFinder;
 import org.nlpcn.commons.lang.util.IOUtil;
 import org.nlpcn.commons.lang.util.StringUtil;
+import org.nlpcn.commons.lang.util.tuples.KeyValue;
+import org.nlpcn.jcoder.run.annotation.DefaultExecute;
+import org.nlpcn.jcoder.run.annotation.Execute;
 import org.nutz.http.Http;
 import org.nutz.ioc.Ioc;
 import org.nutz.ioc.impl.NutIoc;
@@ -52,12 +56,14 @@ public class Testing {
 	 * @return class c instance
 	 * @throws Exception
 	 */
-	public static <T> T instance(Class<T> c, String iocPath) throws Exception {
+	public static <T> T instance(Class<T> c, String iocPath, Class<?>... relation) throws Exception {
 		Ioc ioc = new NutIoc(new JsonLoader(iocPath));
 
 		StaticValue.setSystemIoc(ioc);
 		StaticValue.setUserIoc(ioc);
 
+		relation(ioc, relation);
+
 		Mirror<?> mirror = Mirror.me(c);
 		T obj = c.newInstance();
 
@@ -75,7 +81,7 @@ public static <T> T instance(Class<T> c, String iocPath) throws Exception {
 		return obj;
 	}
 
-	public static <T> T instance(Class<T> c) throws Exception {
+	public static <T> T instance(Class<T> c, Class<?>... relation) throws Exception {
 
 		File find = new File("src/test/resources/ioc.js");
 
@@ -87,13 +93,64 @@ public static <T> T instance(Class<T> c) throws Exception {
 		}
 
 		if (find != null && find.exists()) {
-			return instance(c, find.getAbsolutePath());
+			return instance(c, find.getAbsolutePath(), relation);
 		} else {
 			throw new FileNotFoundException("ioc.js not found in your classpath ");
 		}
 
 	}
 
+	private static void relation(Ioc ioc, Class<?>... clas) throws InstantiationException, IllegalAccessException {
+		if (clas == null || clas.length == 0) {
+			return;
+		}
+
+		for (Class<?> c : clas) {
+			Mirror<?> mirror = Mirror.me(c);
+			Object obj = c.newInstance();
+
+			for (Field field : mirror.getFields()) {
+				Inject inject = field.getAnnotation(Inject.class);
+				if (inject != null) {
+					if (field.getType().equals(org.slf4j.Logger.class)) {
+						mirror.setValue(obj, field, LoggerFactory.getLogger(c));
+					} else {
+						mirror.setValue(obj, field, ioc.get(field.getType(), StringUtil.isBlank(inject.value()) ? field.getName() : inject.value()));
+					}
+				}
+			}
+
+			for (Method method : mirror.getMethods()) {
+				if (method.getAnnotation(Execute.class) == null && method.getAnnotation(DefaultExecute.class) == null) {
+					continue;
+				}
+				String key = c.getSimpleName() + "/" + method.getName();
+				LOG.info("add relation " + key);
+				TestingFilter.methods.put(key, KeyValue.with(method, obj));
+			}
+		}
+
+	}
+
+	/**
+	 * 释放关联类
+	 * 
+	 * @param clas
+	 * @throws IllegalAccessException
+	 * @throws InstantiationException
+	 */
+	public static void unRelation(Class<?>... clas) {
+		for (Class<?> c : clas) {
+			Mirror<?> mirror = Mirror.me(c);
+			for (Method method : mirror.getMethods()) {
+				if (method.getAnnotation(Execute.class) == null && method.getAnnotation(DefaultExecute.class) == null) {
+					continue;
+				}
+				TestingFilter.methods.remove(c.getSimpleName() + "/" + method.getName());
+			}
+		}
+	}
+
 	/**
 	 * 对比本地代码和线上代码的不同
 	 * 
@@ -178,7 +235,7 @@ public static void startServer(int port, String iocPath, String jcoderHome, Stri
 		server.join();
 
 	}
-	
+
 	/**
 	 * test local api by server
 	 * 
@@ -190,6 +247,5 @@ public static void startServer(int port, String iocPath, String jcoderHome, Stri
 	public static void startServer(String iocPath, String[] packages) throws Exception {
 		startServer(8080, iocPath, null, packages);
 	}
-	
-	
+
 }

src/main/java/org/nlpcn/jcoder/util/TestingFilter.java

@@ -30,7 +30,6 @@
 import org.nutz.mvc.ActionInfo;
 import org.nutz.mvc.Mvcs;
 import org.nutz.mvc.NutFilter;
-import org.nutz.mvc.impl.processor.EncodingProcessor;
 import org.nutz.resource.Scans;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -39,7 +38,7 @@ public class TestingFilter extends NutFilter {
 
 	private static final Logger LOG = LoggerFactory.getLogger(TestingFilter.class);
 
-	public static Map<String, KeyValue<Method, Object>> methods = null;
+	public static Map<String, KeyValue<Method, Object>> methods = new HashMap<>();
 
 	public static void init(String... packages) throws IOException {
 		Map<String, KeyValue<Method, Object>> tempMethods = new HashMap<>();

src/test/java/cn/com/infcn/api/test/TaskDemo.java

@@ -1,12 +1,13 @@
 package cn.com.infcn.api.test;
 
-import org.apache.log4j.Logger;
 import org.nlpcn.jcoder.filter.TokenFilter;
 import org.nlpcn.jcoder.run.annotation.DefaultExecute;
 import org.nlpcn.jcoder.run.annotation.Single;
+import org.nlpcn.jcoder.util.Testing;
 import org.nutz.ioc.loader.annotation.Inject;
 import org.nutz.mvc.annotation.By;
 import org.nutz.mvc.annotation.Filters;
+import org.slf4j.Logger;
 
 @Single(true)
 public class TaskDemo {
@@ -24,6 +25,8 @@ public class TaskDemo {
 	@DefaultExecute
 	@Filters(@By(type = TokenFilter.class, args = { "false" }))
 	public Object execute(String name, Integer age, Character sex) {
+		log.info("12312312");
 		return "name:" + name +" age:"+age+" sex:"+sex;
 	}
+	
 }