top.doudou.common.aop.collector.LogCollectorExecutor Maven / Gradle / Ivy
package top.doudou.common.aop.collector;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import top.doudou.common.aop.LogData;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Map;
/**
* @Description
* @Author 傻男人 <[email protected]>
* @Date 2020-09-24 14:53
* @Version V1.0
*/
@Component
public class LogCollectorExecutor {
@Resource
private LogCollector collector;
private Map, LogCollector> collectors = new HashMap<>();
private ApplicationContext applicationContext;
public LogCollectorExecutor(@Autowired ApplicationContext applicationContext) {
this.applicationContext = applicationContext;
}
public ApplicationContext getApplicationContext() {
return applicationContext;
}
/**
* AsyncMode log collection 异步模式日志收集
*
* @param clz 日志收集器Class对象
* @param logData 日志数据
*/
@Async
public void asyncExecute(Class clz, LogData logData) {
execute(clz, logData);
}
/**
* 同步模式收集日志
*
* @param clz 日志收集器Class对象
* @param logData 日志数据
*/
public void execute(Class clz, LogData logData) {
getExecuteLogCollector(clz).collect(logData);
}
/**
* Get the specified log collector 获取指定的日志收集器
*
* @param clz 日志收集器Class对象
* @return 获取指定的日志收集器
*/
private LogCollector getExecuteLogCollector(Class clz) {
if (clz != DefaultCollector.class) {
LogCollector c;
try {
c = applicationContext.getBean(clz);
} catch (Exception e) {
c = collectors.get(clz);
if (c == null) {
c = BeanUtils.instantiateClass(clz);
collectors.put(clz, c);
}
}
return c;
} else {
return collector;
}
}
}