magicodex commented on issue #9524:
URL:
https://github.com/apache/shardingsphere/issues/9524#issuecomment-1684903475
> Is there any conclusion to this question?
Code Examples:
import cn.hutool.core.util.ReflectUtil;
import
org.apache.shardingsphere.driver.jdbc.core.datasource.ShardingSphereDataSource;
import org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
import
org.apache.shardingsphere.infra.metadata.rule.ShardingSphereRuleMetaData;
import org.apache.shardingsphere.infra.rule.ShardingSphereRule;
import org.apache.shardingsphere.mode.manager.ContextManager;
import org.apache.shardingsphere.mode.metadata.MetaDataContexts;
import org.apache.shardingsphere.singletable.rule.SingleTableRule;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.Collection;
import java.util.Map;
@Component
@ConfigurationProperties(prefix = "spring.shardingsphere.custom",
ignoreUnknownFields = true)
public class ShardingSphereDataSourcePostProcessor implements
BeanPostProcessor {
private String defaultDataSourceName;
public void setDefaultDataSourceName(String defaultDataSourceName) {
this.defaultDataSourceName = defaultDataSourceName;
}
@Override
public Object postProcessAfterInitialization(Object bean, String
beanName) throws BeansException {
if (bean instanceof ShardingSphereDataSource) {
ShardingSphereDataSource dataSource = (ShardingSphereDataSource)
bean;
setDefaultDataSourceName(dataSource, defaultDataSourceName);
}
return bean;
}
private void setDefaultDataSourceName(ShardingSphereDataSource
dataSource, String defaultDataSourceName) {
ContextManager contextManager = dataSource.getContextManager();
MetaDataContexts metaDataContexts =
contextManager.getMetaDataContexts();
Map<String, ShardingSphereMetaData> metaDataMap =
metaDataContexts.getMetaDataMap();
Collection<ShardingSphereMetaData> metaDataCollection =
metaDataMap.values();
for (ShardingSphereMetaData metaData : metaDataCollection) {
ShardingSphereRuleMetaData ruleMetaData =
metaData.getRuleMetaData();
Collection<ShardingSphereRule> rules = ruleMetaData.getRules();
for (ShardingSphereRule rule : rules) {
if (rule instanceof SingleTableRule) {
SingleTableRule singleTableRule = (SingleTableRule) rule;
ReflectUtil.setFieldValue(singleTableRule,
"defaultDataSource", defaultDataSourceName);
}
}
}
}
}
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
