/**
* 版权声明：中图一购网络科技有限公司 版权所有 违者必究 2012
* 日 期：12-6-4
*/
package com.rop.config;
import com.rop.RopException;
import com.rop.ThreadFerry;
import com.rop.impl.AnnotationServletServiceRouterFactoryBean;
import com.rop.impl.DefaultServiceAccessController;
import com.rop.security.DefaultInvokeTimesController;
import com.rop.security.DefaultSecurityManager;
import com.rop.security.FileBaseAppSecretManager;
import com.rop.session.DefaultSessionManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.RuntimeBeanReference;
import org.springframework.beans.factory.parsing.BeanComponentDefinition;
import org.springframework.beans.factory.parsing.CompositeComponentDefinition;
import org.springframework.beans.factory.support.RootBeanDefinition;
import org.springframework.beans.factory.xml.BeanDefinitionParser;
import org.springframework.beans.factory.xml.ParserContext;
import org.springframework.format.support.FormattingConversionServiceFactoryBean;
import org.springframework.util.ClassUtils;
import org.springframework.util.StringUtils;
import org.w3c.dom.Element;
/**
* <pre>
* 功能说明：
* </pre>
*
* @author 陈雄华
* @version 1.0
*/
public class AnnotationDrivenBeanDefinitionParser implements BeanDefinitionParser {
public static final int DEFAULT_CORE_POOL_SIZE = 200;
public static final int DEFAULT_MAX_POOL_SIZE = 500;
public static final int DEFAULT_KEEP_ALIVE_SECONDS = 5 * 60;
public static final int DEFAULT_QUENE_CAPACITY = 20;
protected final Logger logger = LoggerFactory.getLogger(getClass());
@Override
public BeanDefinition parse(Element element, ParserContext parserContext) {
Object source = parserContext.extractSource(element);
CompositeComponentDefinition compDefinition = new CompositeComponentDefinition(element.getTagName(), source);
parserContext.pushContainingComponent(compDefinition);
//注册ServiceRouter Bean
RootBeanDefinition serviceRouterDef = new RootBeanDefinition(AnnotationServletServiceRouterFactoryBean.class);
String serviceRouterName = parserContext.getReaderContext().registerWithGeneratedName(serviceRouterDef);
parserContext.registerComponent(new BeanComponentDefinition(serviceRouterDef, serviceRouterName));
//设置formattingConversionService
RuntimeBeanReference conversionServiceRbf = getConversionService(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("formattingConversionService", conversionServiceRbf);
//会话管理器
RuntimeBeanReference sessionManager = getSessionManager(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("sessionManager", sessionManager);
//密钥管理器
RuntimeBeanReference appSecretManager = getAppSecretManager(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("appSecretManager", appSecretManager);
//服务访问控制器
RuntimeBeanReference serviceAccessController = getServiceAccessController(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("serviceAccessController", serviceAccessController);
//访问次数/频度控制器
RuntimeBeanReference invokeTimesController = getInvokeTimesController(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("invokeTimesController", invokeTimesController);
//设置TaskExecutor
RuntimeBeanReference taskExecutorBeanReference = getExecutor(element, source, parserContext);
serviceRouterDef.getPropertyValues().add("threadPoolExecutor", taskExecutorBeanReference);
//设置signEnable
String signEnable = element.getAttribute("sign-enable");
if (StringUtils.hasText(signEnable)) {
serviceRouterDef.getPropertyValues().addPropertyValue("signEnable", Boolean.valueOf(signEnable));
}
//设置threadFerryClass
String threadFerryClassName = element.getAttribute("thread-ferry-class");
if (StringUtils.hasText(threadFerryClassName)) {
try {
Class<?> threadFerryClass = ClassUtils.forName(threadFerryClassName, getClass().getClassLoader());
if (!ClassUtils.isAssignable(ThreadFerry.class, threadFerryClass)) {
throw new RopException(threadFerryClassName + "没有实现" + ThreadFerry.class.getName() + "接口");
}
serviceRouterDef.getPropertyValues().addPropertyValue("threadFerryClass", threadFerryClass);
} catch (ClassNotFoundException e) {
throw new RopException(e);
}
}
//设置signEnable
String extErrorBasename = element.getAttribute("ext-error-base-name");
if (StringUtils.hasText(extErrorBasename)) {
serviceRouterDef.getPropertyValues().addPropertyValue("extErrorBasename", extErrorBasename);
}
String serviceTimeoutSeconds = element.getAttribute("service-timeout-seconds");
if (StringUtils.hasText(serviceTimeoutSeconds)) {
serviceRouterDef.getPropertyValues().addPropertyValue("serviceTimeoutSeconds", Integer.parseInt(serviceTimeoutSeconds));
}
String uploadFileMaxSize = element.getAttribute("upload-file-max-size");
if (StringUtils.hasText(uploadFileMaxSize)) {
serviceRouterDef.getPropertyValues().addPropertyValue("uploadFileMaxSize", Integer.parseInt(uploadFileMaxSize));
}
String uploadFileTypes = element.getAttribute("upload-file-types");
if (StringUtils.hasText(uploadFileTypes)) {
serviceRouterDef.getPropertyValues().addPropertyValue("uploadFileTypes", uploadFileTypes);
}
parserContext.popAndRegisterContainingComponent();
return null;
}
private RuntimeBeanReference getInvokeTimesController(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference invokeTimesControllerRef;
if (element.hasAttribute("invoke-times-controller")) {
invokeTimesControllerRef = new RuntimeBeanReference(element.getAttribute("invoke-times-controller"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的服务调用次数/频度控制器:" + invokeTimesControllerRef.getBeanName());
}
} else {
RootBeanDefinition invokeTimesControllerDef = new RootBeanDefinition(DefaultInvokeTimesController.class);
invokeTimesControllerDef.setSource(source);
invokeTimesControllerDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String invokeTimesControllerName = parserContext.getReaderContext().registerWithGeneratedName(invokeTimesControllerDef);
parserContext.registerComponent(new BeanComponentDefinition(invokeTimesControllerDef, invokeTimesControllerName));
invokeTimesControllerRef = new RuntimeBeanReference(invokeTimesControllerName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的服务调用次数/频度控制器:" + DefaultInvokeTimesController.class.getName());
}
}
return invokeTimesControllerRef;
}
private RuntimeBeanReference getServiceAccessController(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference serviceAccessControllerRef;
if (element.hasAttribute("service-access-controller")) {
serviceAccessControllerRef = new RuntimeBeanReference(element.getAttribute("service-access-controller"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的服务访问控制器:" + serviceAccessControllerRef.getBeanName());
}
} else {
RootBeanDefinition serviceAccessControllerDef = new RootBeanDefinition(DefaultServiceAccessController.class);
serviceAccessControllerDef.setSource(source);
serviceAccessControllerDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String serviceAccessControllerName = parserContext.getReaderContext().registerWithGeneratedName(serviceAccessControllerDef);
parserContext.registerComponent(new BeanComponentDefinition(serviceAccessControllerDef, serviceAccessControllerName));
serviceAccessControllerRef = new RuntimeBeanReference(serviceAccessControllerName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的服务访问控制器:" + DefaultServiceAccessController.class.getName());
}
}
return serviceAccessControllerRef;
}
private RuntimeBeanReference getAppSecretManager(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference appSecretManagerRef;
if (element.hasAttribute("app-secret-manager")) {
appSecretManagerRef = new RuntimeBeanReference(element.getAttribute("app-secret-manager"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的密钥管理器:" + appSecretManagerRef.getBeanName());
}
} else {
RootBeanDefinition appSecretManagerDef = new RootBeanDefinition(FileBaseAppSecretManager.class);
appSecretManagerDef.setSource(source);
appSecretManagerDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String appSecretManagerName = parserContext.getReaderContext().registerWithGeneratedName(appSecretManagerDef);
parserContext.registerComponent(new BeanComponentDefinition(appSecretManagerDef, appSecretManagerName));
appSecretManagerRef = new RuntimeBeanReference(appSecretManagerName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的密钥管理器:" + FileBaseAppSecretManager.class.getName());
}
}
return appSecretManagerRef;
}
private RuntimeBeanReference getSessionManager(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference sessionManagerRef;
if (element.hasAttribute("session-manager")) {
sessionManagerRef = new RuntimeBeanReference(element.getAttribute("session-manager"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的SessionManager:" + sessionManagerRef.getBeanName());
}
} else {
RootBeanDefinition sessionManagerDef = new RootBeanDefinition(DefaultSessionManager.class);
sessionManagerDef.setSource(source);
sessionManagerDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String sessionManagerName = parserContext.getReaderContext().registerWithGeneratedName(sessionManagerDef);
parserContext.registerComponent(new BeanComponentDefinition(sessionManagerDef, sessionManagerName));
sessionManagerRef = new RuntimeBeanReference(sessionManagerName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的会话管理器:" + DefaultSessionManager.class.getName());
}
}
return sessionManagerRef;
}
private RuntimeBeanReference getConversionService(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference conversionServiceRbf;
if (element.hasAttribute("formatting-conversion-service")) {
conversionServiceRbf = new RuntimeBeanReference(element.getAttribute("formatting-conversion-service"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的FormattingConversionService:" + conversionServiceRbf.getBeanName());
}
} else {
RootBeanDefinition conversionDef = new RootBeanDefinition(FormattingConversionServiceFactoryBean.class);
conversionDef.setSource(source);
conversionDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String conversionName = parserContext.getReaderContext().registerWithGeneratedName(conversionDef);
parserContext.registerComponent(new BeanComponentDefinition(conversionDef, conversionName));
conversionServiceRbf = new RuntimeBeanReference(conversionName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的FormattingConversionService:" +
FormattingConversionServiceFactoryBean.class.getName());
}
}
return conversionServiceRbf;
}
private RuntimeBeanReference getRopValidator(Element element, Object source, ParserContext parserContext) {
RuntimeBeanReference ropValidatorRbf;
if (element.hasAttribute("rop-validator")) {
ropValidatorRbf = new RuntimeBeanReference(element.getAttribute("rop-validator"));
if (logger.isInfoEnabled()) {
logger.info("Rop装配一个自定义的RopValidator:" + ropValidatorRbf.getBeanName());
}
} else {
RootBeanDefinition conversionDef = new RootBeanDefinition(DefaultSecurityManager.class);
conversionDef.setSource(source);
conversionDef.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
String conversionName = parserContext.getReaderContext().registerWithGeneratedName(conversionDef);
parserContext.registerComponent(new BeanComponentDefinition(conversionDef, conversionName));
ropValidatorRbf = new RuntimeBeanReference(conversionName);
if (logger.isInfoEnabled()) {
logger.info("使用默认的RopValidator:" + DefaultSecurityManager.class.getName());
}
}
return ropValidatorRbf;
}
/**
* 创建异步执行器
*
* @param element
* @param source
* @param parserContext
* @return
*/
private RuntimeBeanReference getExecutor(Element element, Object source, ParserContext parserContext) {
RootBeanDefinition taskExecutorDef =
new RootBeanDefinition(org.springframework.scheduling.concurrent.ThreadPoolExecutorFactoryBean.class);
String taskExecutorName = parserContext.getReaderContext().registerWithGeneratedName(taskExecutorDef);
String corePoolSize = element.getAttribute("core-pool-size");
if (StringUtils.hasText(corePoolSize)) {
taskExecutorDef.getPropertyValues().addPropertyValue("corePoolSize", corePoolSize);
}else{
taskExecutorDef.getPropertyValues().addPropertyValue("corePoolSize", DEFAULT_CORE_POOL_SIZE);
}
String maxPoolSize = element.getAttribute("max-pool-size");
if (StringUtils.hasText(maxPoolSize)) {
taskExecutorDef.getPropertyValues().addPropertyValue("maxPoolSize", maxPoolSize);
}else{
taskExecutorDef.getPropertyValues().addPropertyValue("maxPoolSize", DEFAULT_MAX_POOL_SIZE);
}
String keepAliveSeconds = element.getAttribute("keep-alive-seconds");
if (StringUtils.hasText(keepAliveSeconds)) {
taskExecutorDef.getPropertyValues().addPropertyValue("keepAliveSeconds", keepAliveSeconds);
}else{
taskExecutorDef.getPropertyValues().addPropertyValue("keepAliveSeconds", DEFAULT_KEEP_ALIVE_SECONDS);
}
String queueCapacity = element.getAttribute("queue-capacity");
if (StringUtils.hasText(queueCapacity)) {
taskExecutorDef.getPropertyValues().addPropertyValue("queueCapacity", queueCapacity);
}else{
taskExecutorDef.getPropertyValues().addPropertyValue("queueCapacity", DEFAULT_QUENE_CAPACITY);
}
parserContext.registerComponent(new BeanComponentDefinition(taskExecutorDef, taskExecutorName));
RuntimeBeanReference taskExecutorBeanReference = new RuntimeBeanReference(taskExecutorName);
return taskExecutorBeanReference;
}
}