详情见: https://redkale.org * diff --git a/src/main/java/org/redkale/inject/ResourceAnnotationLoader.java b/src/main/java/org/redkale/inject/ResourceAnnotationLoader.java index 522d3f48a..f770c0bbf 100644 --- a/src/main/java/org/redkale/inject/ResourceAnnotationLoader.java +++ b/src/main/java/org/redkale/inject/ResourceAnnotationLoader.java @@ -9,6 +9,52 @@ import java.lang.reflect.Field; /** * 自定义注入加载器 * + *
+ *
+ *
+ * @Documented
+ * @Target({FIELD})
+ * @Retention(RUNTIME)
+ * public @interface CustomConf {
+ * String path();
+ * }
+ *
+ *
+ * public class CustomConfProvider implements ResourceAnnotationLoader<CustomConf> {
+ *
+ * @Override
+ * public void load(
+ * ResourceFactory factory,
+ * String srcResourceName,
+ * Object srcObj,
+ * CustomConf annotation,
+ * Field field,
+ * Object attachment) {
+ * try {
+ * field.set(srcObj, new File(annotation.path()));
+ * } catch (Exception e) {
+ * e.printStackTrace();
+ * }
+ * System.out.println("对象是 src =" + srcObj + ", path=" + annotation.path());
+ * }
+ *
+ * @Override
+ * public Class<CustomConf> annotationType() {
+ * return CustomConf.class;
+ * }
+ * }
+ *
+ *
+ * public class InjectBean {
+ *
+ * @CustomConf(path = "conf/test.xml")
+ * public File conf;
+ * }
+ *
+ *
+ *
+ *
+ *
* 详情见: https://redkale.org
*
* @since 2.8.0
diff --git a/src/main/java/org/redkale/inject/ResourceFactory.java b/src/main/java/org/redkale/inject/ResourceFactory.java
index 3b467358c..ffc4f060b 100644
--- a/src/main/java/org/redkale/inject/ResourceFactory.java
+++ b/src/main/java/org/redkale/inject/ResourceFactory.java
@@ -106,17 +106,23 @@ public final class ResourceFactory {
return result;
}
- /** 清空当前已注入资源的缓存 */
+ /**
+ * 清空当前已注入资源的缓存
+ */
public void release() {
this.entryStore.clear();
}
- /** inject时的锁 */
+ /**
+ * inject时的锁
+ */
public void lock() {
lock.lock();
}
- /** inject时的锁 */
+ /**
+ * inject时的锁
+ */
public void unlock() {
lock.unlock();
}
@@ -279,10 +285,7 @@ public final class ResourceFactory {
* @return 旧资源对象
*/
public A register(final boolean autoSync, final A rs) {
- if (rs == null) {
- return null;
- }
- return register(autoSync, "", rs);
+ return rs == null ? null : register(autoSync, "", rs);
}
/**
@@ -623,6 +626,7 @@ public final class ResourceFactory {
* 注册Configuration配置类
*
* @param configuareClass 标记Configuration的类
+ * @return 方法数
*
*/
public int registerConfiguration(final Class configuareClass) {
@@ -871,20 +875,62 @@ public final class ResourceFactory {
return parent == null ? null : parent.findSuperTypeLoader(ft, field);
}
+ /**
+ * 找指定类型对应的资源对象
+ *
+ * @param 泛型
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
public A find(Class clazz) {
return find("", clazz);
}
- public A find(String name, Type clazz) {
- ResourceEntry re = findEntry(name, clazz);
- return re == null ? null : (A) re.value;
- }
-
+ /**
+ * 找指定类型和资源名对应的资源对象
+ *
+ * @param 泛型
+ * @param name 资源名
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
public A find(String name, Class clazz) {
ResourceEntry re = findEntry(name, clazz);
return re == null ? null : re.value;
}
+ /**
+ * 找指定类型对应的资源对象
+ *
+ * @param 泛型
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
+ public A find(Type clazz) {
+ return find("", clazz);
+ }
+
+ /**
+ * 找指定类型和资源名对应的资源对象
+ *
+ * @param 泛型
+ * @param name 资源名
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
+ public A find(String name, Type clazz) {
+ ResourceEntry re = findEntry(name, clazz);
+ return re == null ? null : (A) re.value;
+ }
+
+ /**
+ * 找指定类型或子类型和资源名对应的资源对象
+ *
+ * @param 泛型
+ * @param name 资源名
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
public A findChild(final String name, final Class clazz) {
A rs = find(name, clazz);
if (rs != null) {
@@ -905,6 +951,13 @@ public final class ResourceFactory {
return null;
}
+ /**
+ * 找指定类型和资源名对应的资源对象
+ * @param 泛型
+ * @param name 资源名
+ * @param clazz 资源类型
+ * @return 资源对象
+ */
private ResourceEntry findEntry(String name, Type clazz) {
Map 详情见: https://redkale.org
*
* @author zhangjx
diff --git a/src/test/java/org/redkale/test/inject/ResourceTypeTest.java b/src/test/java/org/redkale/test/inject/ResourceTypeTest.java
new file mode 100644
index 000000000..33280dec4
--- /dev/null
+++ b/src/test/java/org/redkale/test/inject/ResourceTypeTest.java
@@ -0,0 +1,68 @@
+/*
+
+*/
+
+package org.redkale.test.inject;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.Type;
+import org.junit.jupiter.api.Test;
+import org.redkale.annotation.Resource;
+import org.redkale.inject.ResourceFactory;
+import org.redkale.inject.ResourceTypeLoader;
+import org.redkale.source.DataMemorySource;
+import org.redkale.source.DataSource;
+
+/**
+ *
+ * @author zhangjx
+ */
+public class ResourceTypeTest {
+
+ public static void main(String[] args) throws Throwable {
+ ResourceTypeTest test = new ResourceTypeTest();
+ test.run();
+ }
+
+ @Test
+ public void run() throws Exception {
+ ResourceFactory factory = ResourceFactory.create();
+ factory.register(new DataSourceProvider());
+ InjectBean bean = new InjectBean();
+ factory.inject(bean);
+ }
+
+ public static class DataSourceProvider implements ResourceTypeLoader {
+
+ @Override
+ public Object load(
+ ResourceFactory factory,
+ String srcResourceName,
+ Object srcObj,
+ String resourceName,
+ Field field,
+ Object attachment) {
+ DataSource source = new DataMemorySource(resourceName);
+ factory.register(resourceName, DataSource.class, source);
+ if (field != null) {
+ try {
+ field.set(srcObj, source);
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+ return source;
+ }
+
+ @Override
+ public Type resourceType() {
+ return DataSource.class;
+ }
+ }
+
+ public static class InjectBean {
+
+ @Resource(name = "platf")
+ public DataSource source;
+ }
+}
+ *
+ *
*
+ *
+ * public class CustomConfProvider implements ResourceAnnotationLoader<CustomConf> {
+ *
+ * @Override
+ * public Object load(
+ * ResourceFactory factory,
+ * String srcResourceName,
+ * Object srcObj,
+ * CustomConf annotation,
+ * Field field,
+ * Object attachment) {
+ * DataSource source = new DataMemorySource(resourceName);
+ * factory.register(resourceName, DataSource.class, source);
+ * if (field != null) {
+ * try {
+ * field.set(srcObj, source);
+ * } catch (Exception e) {
+ * e.printStackTrace();
+ * }
+ * }
+ * return source;
+ * }
+ *
+ * @Override
+ * public Type resourceType() {
+ * return DataSource.class;
+ * }
+ * }
+ *
+ *
+ * public class InjectBean {
+ *
+ * @Resource(name = "platf")
+ * public DataSource source;
+ * }
+ *
+ *
+ * ResourceFactory factory = ResourceFactory.create();
+ * factory.register(new DataSourceProvider());
+ * InjectBean bean = new InjectBean();
+ * factory.inject(bean);
+ *
+ *
+ *
+ *