cn.kduck.core.configuration.DaoConfiguration Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of kduck-core Show documentation
Show all versions of kduck-core Show documentation
The core of the K-Duck development framework encompasses all the featured components of the framework.
package cn.kduck.core.configuration;
import cn.kduck.core.KduckProperties;
import cn.kduck.core.KduckProperties.ShowSqlProperties;
import cn.kduck.core.dao.DefaultDeleteArchiveHandler;
import cn.kduck.core.dao.DeleteArchiveHandler;
import cn.kduck.core.dao.JdbcEntityDao;
import cn.kduck.core.dao.datasource.DataSourceMatcher;
import cn.kduck.core.dao.datasource.DataSourceSwitch;
import cn.kduck.core.dao.datasource.DynamicDataSource;
import cn.kduck.core.dao.datasource.condition.RequestMethodMatcher;
import cn.kduck.core.dao.definition.BeanDefDepository;
import cn.kduck.core.dao.definition.BeanDefSource;
import cn.kduck.core.dao.definition.MemoryBeanDefDepository;
import cn.kduck.core.dao.definition.impl.JdbcBeanDefSource;
import cn.kduck.core.dao.dialect.DatabaseDialect;
import cn.kduck.core.dao.id.IdGenerator;
import cn.kduck.core.dao.id.impl.KduckIdGenerator;
import cn.kduck.core.dao.id.impl.KduckIdGenerator.KduckSnowFlakeProperties;
import cn.kduck.core.dao.sqllog.ShowSqlLogger;
import cn.kduck.core.dao.sqllog.impl.DefaultShowSqlLogger;
import cn.kduck.core.utils.StringUtils;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionOutcome;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.condition.SpringBootCondition;
import org.springframework.context.annotation.*;
import org.springframework.core.env.Environment;
import org.springframework.core.type.AnnotatedTypeMetadata;
import javax.sql.DataSource;
import java.util.*;
import java.util.Map.Entry;
/**
* LiuHG
*/
public class DaoConfiguration {
private final KduckProperties kduckProperties;
public DaoConfiguration(KduckProperties kduckProperties){
this.kduckProperties = kduckProperties;
}
@Bean
@ConditionalOnMissingBean(IdGenerator.class)
public IdGenerator idGenerator(KduckSnowFlakeProperties kduckSnowFlakeProperties){
return new KduckIdGenerator(kduckSnowFlakeProperties.getServerIp(),kduckSnowFlakeProperties.getRegions());
}
@Bean
@ConditionalOnMissingBean(BeanDefDepository.class)
public BeanDefDepository beanDefDepository(@Lazy List beanDefSourceList){
return new MemoryBeanDefDepository(beanDefSourceList);
}
@Bean
@ConditionalOnMissingBean(BeanDefSource.class)
public BeanDefSource beanDefSource(DataSource dataSource){
JdbcBeanDefSource jdbcBeanDefSource = new JdbcBeanDefSource(dataSource,kduckProperties.getDefinition());
return jdbcBeanDefSource;
}
@Bean
@ConditionalOnMissingBean(DeleteArchiveHandler.class)
public DeleteArchiveHandler deleteArchiveHandler(){
return new DefaultDeleteArchiveHandler();
}
@Bean
@ConditionalOnMissingBean(JdbcEntityDao.class)
public JdbcEntityDao jdbcEntityDao(DataSource dataSource,List databaseDialectList,
@Autowired(required = false) ShowSqlLogger sqlLogger,
@Lazy DeleteArchiveHandler deleteArchiveHandler){
JdbcEntityDao jdbcEntityDao = new JdbcEntityDao(dataSource,databaseDialectList);
if(sqlLogger != null){
jdbcEntityDao.setSqlLogger(sqlLogger);
}
ShowSqlProperties showSqlProperties = kduckProperties.getShowSql();
if(showSqlProperties != null){
jdbcEntityDao.setShowSqlMode(showSqlProperties.getMode());
}
//deleteArchiveHandler由于使用了DefaultService进行了表数据持久，会导致循环依赖，因此需要标记为@Lazy懒加载
jdbcEntityDao.setDeleteArchiveHandler(deleteArchiveHandler);
return jdbcEntityDao;
}
@Bean
@ConditionalOnMissingBean(ShowSqlLogger.class)
@ConditionalOnProperty(prefix = "kduck.show-sql",name = "enabled",havingValue = "true")
public ShowSqlLogger showSqlLogger(){
return new DefaultShowSqlLogger(System.out,kduckProperties.getShowSql());
}
// @Bean
//// @Primary
// @ConfigurationProperties("spring.datasource")
// public DataSourceProperties dataSourceProperties() {
// return new DataSourceProperties();
// }
//
// @Bean
// @ConfigurationProperties("spring.datasource.hikari")
// public DataSource mainDataSource(DataSourceProperties properties) {
// return properties.initializeDataSourceBuilder().type(HikariDataSource.class).build();
// }
@Bean
@Conditional(SpringDataSourceUnavailableCondition.class)
@Primary
public DataSource dataSource(KduckProperties properties) {
DynamicDataSource dataSource = new DynamicDataSource();
Map dataSourceMap = new HashMap<>();
DataSource defaultDataSource = null;
Map dsPropMap = properties.getDataSource();
if(dsPropMap.isEmpty()) {
throw new RuntimeException("多数据源置失败，请至少配置1个数据源参数");
}
String[] lookupKeys = new String[dsPropMap.size()];
Iterator keyNames = dsPropMap.keySet().iterator();
int index = 0 ;
while(keyNames.hasNext()){
Object dsKey = keyNames.next();
Map propMap = (Map) dsPropMap.get(dsKey);
lookupKeys[index] = dsKey.toString();
index++;
Object url = propMap.remove("url");
if(url != null) {
propMap.put("jdbcUrl",url);
}
Properties dsProperties = convertProperties(propMap);
Map matchMap = (Map) dsProperties.remove("match");
if(matchMap != null){
String matchMethod = (String)matchMap.get("requestMethod");
String matcherClass = (String)matchMap.get("matcherClass");
if(matchMethod != null){
DataSourceSwitch.addSwitchMatcher(new RequestMethodMatcher(matchMethod.split("[,;]")),dsKey.toString());
}
if(matcherClass != null){
String[] matcherClasses = matcherClass.split("[,;]");
for (String mclsName : matcherClasses) {
Class mcls = null;
try {
mcls = (Class) Class.forName(mclsName);
} catch (ClassNotFoundException e) {
throw new RuntimeException("动态数据源匹配器类不存在：" + mclsName +
"，请检查动态数据源的matcher-class参数配置的正确性",e);
}
if(!DataSourceMatcher.class.isAssignableFrom(mcls)){
throw new RuntimeException("动态数据源匹配器类不正确：" + mclsName +
"，该类不是DataSourceMatcher的接口实现类");
}
DataSourceSwitch.addSwitchMatcher(BeanUtils.instantiateClass(mcls),dsKey.toString());
}
}
}
HikariDataSource hikariDataSource = new HikariDataSource(new HikariConfig(dsProperties));
if(defaultDataSource == null){
defaultDataSource = hikariDataSource;
dataSourceMap.put(lookupKeys[0],hikariDataSource);
}
dataSourceMap.put(dsKey,hikariDataSource);
}
DataSourceSwitch.setLookupKeys(lookupKeys);
dataSource.setTargetDataSources(dataSourceMap);
dataSource.setDefaultTargetDataSource(defaultDataSource);
return dataSource;
}
private Properties convertProperties(Map propMap) {
Properties dsProperties = new Properties();
Set entrySet = propMap.entrySet();
for (Entry entry : entrySet) {
String name = entry.getKey().toString();
Object value = entry.getValue();
if(value instanceof Map){
value = new LinkedHashMap(convertProperties((Map) value));
}
String[] keySplit = name.split("-");
if(keySplit.length > 1){
StringBuilder nameBuilder = new StringBuilder();
for (int i = 0; i < keySplit.length; i++) {
if(i == 0){
nameBuilder.append(keySplit[i]);
}else{
nameBuilder.append(StringUtils.upperFirstChar(keySplit[i]));
}
}
dsProperties.put(nameBuilder.toString(), value);
}else{
dsProperties.put(name, value);
}
}
return dsProperties;
}
public static class SpringDataSourceUnavailableCondition extends SpringBootCondition {
@Override
public ConditionOutcome getMatchOutcome(ConditionContext context, AnnotatedTypeMetadata metadata) {
Environment environment = context.getEnvironment();
String url = environment.getProperty("spring.datasource.url");
String jndi = environment.getProperty("spring.datasource.jndi-name");
boolean match = org.springframework.util.StringUtils.hasText(url) || org.springframework.util.StringUtils.hasText(jndi);
if(match){
return new ConditionOutcome(false,"当前使用了SpringDataSource");
}
return new ConditionOutcome(true,"当前使用了K-Duck DataSource（多数据源支持）");
}
}
}