use of com.alipay.sofa.rpc.boot.runtime.param.RpcBindingParam in project sofa-boot by sofastack.
the class ConsumerConfigHelper method getConsumerConfig.
/**
* 获取 ConsumerConfig
*
* @param contract the Contract
* @param binding the RpcBinding
* @return the ConsumerConfig
*/
public ConsumerConfig getConsumerConfig(Contract contract, RpcBinding binding) {
RpcBindingParam param = binding.getRpcBindingParam();
String id = binding.getBeanId();
String interfaceId = contract.getInterfaceType().getName();
String uniqueId = contract.getUniqueId();
Integer timeout = param.getTimeout();
Integer retries = param.getRetries();
String type = param.getType();
Integer addressWaitTime = param.getAddressWaitTime();
Object callbackHandler = param.getCallbackHandler();
String genericInterface = param.getGenericInterface();
String loadBalancer = param.getLoadBalancer();
Boolean lazy = param.getLazy();
Boolean check = param.getCheck();
String mockMode = param.getMockMode();
String serialization = param.getSerialization();
List<Filter> filters = param.getFilters();
List<MethodConfig> methodConfigs = convertToMethodConfig(param.getMethodInfos());
String targetUrl = param.getTargetUrl();
String referenceLimit = sofaBootRpcProperties.getConsumerRepeatedReferenceLimit();
ConsumerConfig consumerConfig = new ConsumerConfig();
if (StringUtils.hasText(appName)) {
consumerConfig.setApplication(new ApplicationConfig().setAppName(appName));
}
if (StringUtils.hasText(id)) {
consumerConfig.setId(id);
}
if (StringUtils.hasText(genericInterface)) {
consumerConfig.setGeneric(true);
consumerConfig.setInterfaceId(genericInterface);
} else if (StringUtils.hasText(interfaceId)) {
consumerConfig.setInterfaceId(interfaceId);
}
if (StringUtils.hasText(uniqueId)) {
consumerConfig.setUniqueId(uniqueId);
}
if (timeout != null) {
consumerConfig.setTimeout(timeout);
}
if (retries != null) {
consumerConfig.setRetries(retries);
}
if (StringUtils.hasText(type)) {
consumerConfig.setInvokeType(type);
}
if (addressWaitTime != null) {
consumerConfig.setAddressWait(addressWaitTime);
}
if (StringUtils.hasText(loadBalancer)) {
consumerConfig.setLoadBalancer(loadBalancer);
}
if (lazy != null) {
consumerConfig.setLazy(lazy);
}
if (check != null) {
consumerConfig.setCheck(check);
}
if (mockMode != null) {
consumerConfig.setMockMode(mockMode);
}
if (callbackHandler != null) {
if (callbackHandler instanceof SofaResponseCallback) {
consumerConfig.setOnReturn((SofaResponseCallback) callbackHandler);
} else {
throw new SofaBootRpcRuntimeException("callback handler must implement SofaResponseCallback [" + callbackHandler + "]");
}
}
if (!CollectionUtils.isEmpty(filters)) {
consumerConfig.setFilterRef(filters);
}
if (!CollectionUtils.isEmpty(methodConfigs)) {
consumerConfig.setMethods(methodConfigs);
}
if (StringUtils.hasText(targetUrl)) {
consumerConfig.setDirectUrl(targetUrl);
consumerConfig.setLazy(true);
consumerConfig.setSubscribe(false);
consumerConfig.setRegister(false);
}
if (StringUtils.hasText(referenceLimit)) {
consumerConfig.setRepeatedReferLimit(Integer.valueOf(referenceLimit));
}
String protocol = binding.getBindingType().getType();
consumerConfig.setBootstrap(protocol);
if (protocol.equals(SofaBootRpcConfigConstants.RPC_PROTOCOL_DUBBO)) {
consumerConfig.setInJVM(false);
}
if (param.getRegistrys() != null && param.getRegistrys().size() > 0) {
List<String> registrys = param.getRegistrys();
for (String registryAlias : registrys) {
RegistryConfig registryConfig = registryConfigContainer.getRegistryConfig(registryAlias);
consumerConfig.setRegistry(registryConfig);
}
} else if (registryConfigContainer.isMeshEnabled(protocol)) {
RegistryConfig registryConfig = registryConfigContainer.getRegistryConfig(SofaBootRpcConfigConstants.REGISTRY_PROTOCOL_MESH);
consumerConfig.setRegistry(registryConfig);
} else {
RegistryConfig registryConfig = registryConfigContainer.getRegistryConfig();
consumerConfig.setRegistry(registryConfig);
}
if (StringUtils.hasText(serialization)) {
consumerConfig.setSerialization(serialization);
}
if (Boolean.TRUE.toString().equals(sofaBootRpcProperties.getHystrixEnable())) {
consumerConfig.setParameter(HystrixConstants.SOFA_HYSTRIX_ENABLED, Boolean.TRUE.toString());
}
// after sofaBootRpcProperties#getHystrixEnable for override global config
if (param.getParameters() != null) {
consumerConfig.setParameters(param.getParameters());
}
return consumerConfig.setProtocol(protocol);
}
use of com.alipay.sofa.rpc.boot.runtime.param.RpcBindingParam in project sofa-boot by sofastack.
the class ConsumerConfigHelper method getMockRef.
public Object getMockRef(RpcBinding binding, ApplicationContext applicationContext) {
RpcBindingParam rpcBindingParam = binding.getRpcBindingParam();
String mockBean = rpcBindingParam.getMockBean();
if (StringUtils.hasText(mockBean)) {
return applicationContext.getBean(mockBean);
} else {
throw new IllegalArgumentException("Mock mode is open, mock bean can't be empty.");
}
}
use of com.alipay.sofa.rpc.boot.runtime.param.RpcBindingParam in project sofa-boot by sofastack.
the class RpcBindingConverter method convert.
/**
* convert xml Element to concrete {@link RpcBinding}
*
* @param element xml Element
* @param bindingConverterContext binding converter context
* @return RpcBinding Object
*/
@Override
public RpcBinding convert(Element element, BindingConverterContext bindingConverterContext) {
RpcBindingParam param = createRpcBindingParam();
// global-attrs,filter,method,route
Element globalAttrsElement = DomUtils.getChildElementByTagName(element, RpcBindingXmlConstants.TAG_GLOBAL_ATTRS);
Element routeElement = DomUtils.getChildElementByTagName(element, RpcBindingXmlConstants.TAG_ROUTE);
List<Element> methodElements = DomUtils.getChildElementsByTagName(element, RpcBindingXmlConstants.TAG_METHOD);
List<Element> parameterElements = DomUtils.getChildElementsByTagName(element, RpcBindingXmlConstants.TAG_PARAMETER);
parseGlobalAttrs(globalAttrsElement, param, bindingConverterContext);
parseFilter(globalAttrsElement, param, bindingConverterContext);
parseMethod(methodElements, param);
parseRoute(routeElement, param);
parseParameter(parameterElements, param);
return convert(param, bindingConverterContext);
}
use of com.alipay.sofa.rpc.boot.runtime.param.RpcBindingParam in project sofa-boot by sofastack.
the class BoltBindingConverter method convert.
@Override
public RpcBinding convert(SofaReference sofaReferenceAnnotation, SofaReferenceBinding sofaReferenceBindingAnnotation, BindingConverterContext bindingConverterContext) {
RpcBindingParam bindingParam = new BoltBindingParam();
convertReferenceAnnotation(bindingParam, sofaReferenceBindingAnnotation, bindingConverterContext);
return new BoltBinding(bindingParam, bindingConverterContext.getApplicationContext(), bindingConverterContext.isInBinding());
}
use of com.alipay.sofa.rpc.boot.runtime.param.RpcBindingParam in project sofa-boot by sofastack.
the class HttpBindingConverter method convert.
@Override
public RpcBinding convert(SofaService sofaServiceAnnotation, SofaServiceBinding sofaServiceBindingAnnotation, BindingConverterContext bindingConverterContext) {
RpcBindingParam bindingParam = new HttpBindingParam();
convertServiceAnnotation(bindingParam, sofaServiceAnnotation, sofaServiceBindingAnnotation, bindingConverterContext);
return new HttpBinding(bindingParam, bindingConverterContext.getApplicationContext(), bindingConverterContext.isInBinding());
}
Aggregations