多数据源模式JPA整合sharding-jdbc实现数据脱敏

< dependency >

     < groupId >org.apache.shardingsphere</ groupId >

     < artifactId >sharding-jdbc-spring-boot-starter</ artifactId >

     < version >${sharding.jdbc.version}</ version >

</ dependency >

#数据库源配置

spring.shardingsphere.datasource.name = ds

spring.shardingsphere.datasource.ds.type = com.zaxxer.hikari.HikariDataSource

spring.shardingsphere.datasource.ds.driver- class -name = com.mysql.jdbc.Driver

spring.shardingsphere.datasource.ds.jdbc-url = jdbc:mysql: //xxx?autoReconnect=true&useUnicode=true&characterEncoding=utf-8

spring.shardingsphere.datasource.ds.username = root

spring.shardingsphere.datasource.ds.password = xxx

spring.shardingsphere.encrypt.encryptors.encryptor_aes.type = aes

spring.shardingsphere.encrypt.encryptors.encryptor_aes.props.aes.key.value = 123456

spring.shardingsphere.encrypt.tables.account.columns.password.plainColumn = password

spring.shardingsphere.encrypt.tables.account.columns.password.cipherColumn = password_encrypt

spring.shardingsphere.encrypt.tables.account.columns.password.encryptor = encryptor_aes

spring.shardingsphere.props.sql.show = true

spring.shardingsphere.props.query.with.cipher.column = true

@SpringBootApplication (exclude = SpringBootConfiguration. class )

@Configuration

@EnableConfigurationProperties (JpaProperties. class )

public class DataSourceConfiguration{

     private final JpaProperties jpaProperties;

     private final Environment environment;

     public DataSourceConfiguration(JpaProperties jpaProperties, Environment environment) {

         this .jpaProperties = jpaProperties;

         this .environment = environment;

     }

     @Primary

     @Bean

     public DataSource dataSource(){

         String prefix = "spring.shardingsphere.datasource." ;

         String each = getDataSourceNames(prefix).get( 0 );

         try {

             return   getDataSource(prefix, each);

         } catch ( final ReflectiveOperationException ex) {

             throw new ShardingSphereException( "Can't find datasource type!" , ex);

         }

     }

     @Primary

     @Bean

     public EntityManagerFactory entityManagerFactory() {

         HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();

         vendorAdapter.setDatabase(Database.MYSQL);

         vendorAdapter.setGenerateDdl( true );

         vendorAdapter.setShowSql( true );

         LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();

         factory.setJpaVendorAdapter(vendorAdapter);

         factory.setPersistenceUnitName( "default" );

         factory.setPackagesToScan(Constants.BASE_PACKAGES);

         factory.setDataSource(dataSource());

         factory.setJpaPropertyMap(jpaProperties.getProperties());

         factory.afterPropertiesSet();

         return factory.getObject();

     }

     @Bean

     @Primary

     public EntityManager entityManager(EntityManagerFactory entityManagerFactory){

         return SharedEntityManagerCreator.createSharedEntityManager(entityManagerFactory);

     }

     @Primary

     @Bean

     public PlatformTransactionManager transactionManager(EntityManagerFactory entityManagerFactory){

         JpaTransactionManager txManager = new JpaTransactionManager();

         txManager.setEntityManagerFactory(entityManagerFactory);

         return txManager;

     }

     private ListgetDataSourceNames( final String prefix) {

         StandardEnvironment standardEnv = (StandardEnvironment) environment;

         standardEnv.setIgnoreUnresolvableNestedPlaceholders( true );

         return null == standardEnv.getProperty(prefix + "name" )

                 ? new InlineExpressionParser(standardEnv.getProperty(prefix + "names" )).splitAndEvaluate() : Collections.singletonList(standardEnv.getProperty(prefix + "name" ));

     }

     @SuppressWarnings ( "unchecked" )

     private DataSource getDataSource( final String prefix, final String dataSourceName) throws ReflectiveOperationException {

         Map dataSourceProps = PropertyUtil.handle(environment, prefix + dataSourceName.trim(), Map. class );

         Preconditions.checkState(!dataSourceProps.isEmpty(), "Wrong datasource properties!" );

         DataSource result = DataSourceUtil.getDataSource(dataSourceProps.get( "type" ).toString(), dataSourceProps);

         DataSourcePropertiesSetterHolder.getDataSourcePropertiesSetterByType(dataSourceProps.get( "type" ).toString()).ifPresent(

                 dataSourcePropertiesSetter -> dataSourcePropertiesSetter.propertiesSet(environment, prefix, dataSourceName, result));

         return result;

     }

}

/**

  * @author: kl @kailing.pub

  * @date: 2020/5/18

  */

@Configuration

@EnableConfigurationProperties ({JpaProperties. class ,SpringBootEncryptRuleConfigurationProperties. class , SpringBootPropertiesConfigurationProperties. class })

@AutoConfigureAfter ({DataSourceAutoConfiguration. class , DataSourceConfiguration. class })

public class EncryptDataSourceConfiguration {

     private final SpringBootPropertiesConfigurationProperties props;

     private final SpringBootEncryptRuleConfigurationProperties encryptRule;

     private final JpaProperties jpaProperties;

     private final DataSource dataSource;

     public EncryptDataSourceConfiguration(SpringBootPropertiesConfigurationProperties props, SpringBootEncryptRuleConfigurationProperties encryptRule, JpaProperties jpaProperties, DataSource dataSource) {

         this .props = props;

         this .encryptRule = encryptRule;

         this .jpaProperties = jpaProperties;

         this .dataSource = dataSource;

     }

     @Bean

     public DataSource encryptDataSource() throws SQLException {

         return EncryptDataSourceFactory.createDataSource(dataSource, new EncryptRuleConfigurationYamlSwapper().swap(encryptRule), props.getProps());

     }

     @Bean

     public EntityManagerFactory encryptEntityManagerFactory() throws SQLException {

         HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();

         vendorAdapter.setDatabase(Database.MYSQL);

         LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();

         factory.setJpaVendorAdapter(vendorAdapter);

         factory.setPersistenceUnitName( "encryptPersistenceUnit" );

         factory.setPackagesToScan(Constants.BASE_PACKAGES);

         factory.setDataSource(encryptDataSource());

         factory.setJpaPropertyMap(jpaProperties.getProperties());

         factory.afterPropertiesSet();

         return factory.getObject();

     }

     @Bean

     public EntityManager encryptEntityManager() throws SQLException {

         return SharedEntityManagerCreator.createSharedEntityManager(encryptEntityManagerFactory());

     }

     @Bean

     public PlatformTransactionManager encryptTransactionManager() throws SQLException {

         JpaTransactionManager txManager = new JpaTransactionManager();

         txManager.setEntityManagerFactory(encryptEntityManagerFactory());

         return txManager;

     }

}

@Repository

public class AccountRepository extends AbstractJpaRepository {

     public AccountRepository( @Qualifier ( "encryptEntityManager" ) EntityManager em) {

         super (AccountModel. class , em);

     }

     @Override

     @Transactional (transactionManager = "encryptTransactionManager" )

     public   S save(S entity) {

         return super .save(entity);

     }

}