springboot学习笔记(九)mongo多数据源配置

末蓝、 2022-05-25 06:42 284阅读 0赞

项目当中有时候一个业务需要用到多个数据源，但是又不是微服务的形式，要在一个项目当中配置多个mongo数据源，如何配置呢？

1.springboot 资源文件（两个mongo地址）：

spring.data.mongodb.primary.uri=mongodb://192.9.8.204:27017/reslib_server
    spring.data.mongodb.secondary.uri=mongodb://192.9.8.204:27017/cloudfsdb

2.写配置类

package com.tyky.educloud.portal.config;
    
    import org.springframework.boot.context.properties.ConfigurationProperties;
    
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    import org.springframework.data.mongodb.MongoDbFactory;
    import org.springframework.data.mongodb.core.MongoTemplate;
    import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
    import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
    
    import com.mongodb.MongoClientURI;
    
    
    @Configuration  //Configuration class  
    @ConfigurationProperties(prefix = "spring.data.mongodb.primary") 
    @EnableMongoRepositories(basePackages="com.tyky.educloud.portal.repository",mongoTemplateRef="mongoTemplate")
    public class PrimaryMongoConfig {
      private String uri;     
      
      @Primary      
      @Bean(name="mongoTemplate")
      public MongoTemplate getMongoTemplate() throws Exception {          
          return new MongoTemplate(mongoDbFactory());      
      }
      
      @Primary
      @Bean(name="mongoDbFactory")
      public MongoDbFactory mongoDbFactory() throws Exception {
          MongoClientURI mongoClientURI = new MongoClientURI(uri);      
          return new SimpleMongoDbFactory(mongoClientURI);
      }
    
      public String getUri() {
        return uri;
      }
    
      public void setUri(String uri) {
        this.uri = uri;
      }
    
    
    }

其中

@EnableMongoRepositories(basePackages="com.tyky.educloud.portal.repository",mongoTemplateRef="mongoTemplate")

指定到你这个数据源对应的仓储类。

第二个数据源配置类：

package com.tyky.educloud.portal.config;
    
    import java.util.Collections;
    import java.util.HashSet;
    import java.util.Set;
    
    import org.springframework.beans.factory.config.BeanDefinition;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.core.convert.converter.Converter;
    import org.springframework.core.type.filter.AnnotationTypeFilter;
    import org.springframework.data.annotation.Persistent;
    import org.springframework.data.authentication.UserCredentials;
    import org.springframework.data.mapping.context.MappingContextIsNewStrategyFactory;
    import org.springframework.data.mapping.model.CamelCaseAbbreviatingFieldNamingStrategy;
    import org.springframework.data.mapping.model.FieldNamingStrategy;
    import org.springframework.data.mapping.model.PropertyNameFieldNamingStrategy;
    import org.springframework.data.mongodb.MongoDbFactory;
    import org.springframework.data.mongodb.config.AbstractMongoConfiguration;
    import org.springframework.data.mongodb.core.MongoTemplate;
    import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
    import org.springframework.data.mongodb.core.convert.CustomConversions;
    import org.springframework.data.mongodb.core.convert.DbRefResolver;
    import org.springframework.data.mongodb.core.convert.DefaultDbRefResolver;
    import org.springframework.data.mongodb.core.convert.MappingMongoConverter;
    import org.springframework.data.mongodb.core.mapping.Document;
    import org.springframework.data.mongodb.core.mapping.MongoMappingContext;
    import org.springframework.data.mongodb.gridfs.GridFsTemplate;
    import org.springframework.data.support.CachingIsNewStrategyFactory;
    import org.springframework.data.support.IsNewStrategyFactory;
    import org.springframework.util.ClassUtils;
    import org.springframework.util.StringUtils;
    
    import com.mongodb.MongoClient;
    import com.mongodb.MongoClientURI;
    
    
    @Configuration  //Configuration class  
    @ConfigurationProperties(prefix = "spring.data.mongodb.secondary") 
    public class SecondartyMongoConfig {
      private String uri;   
      
      @Bean(name="cloudfsTemplate")
      public MongoTemplate getMongoTemplate() throws Exception {          
          return new MongoTemplate(mongoDbFactory());      
      }
      
      @Bean(name="secondaryMongoDbFactory")
      public MongoDbFactory mongoDbFactory() throws Exception {
          MongoClientURI mongoClientURI = new MongoClientURI(uri);      
          return new SimpleMongoDbFactory(mongoClientURI);
      }
      
      @Bean(name="cloudfsgrid")
      public GridFsTemplate getGridFsTemplate() throws Exception {          
          return new GridFsTemplate(mongoDbFactory(), mappingMongoConverter());   
      }
     
    
      /**
       * Return the base package to scan for mapped {@link Document}s. Will return the package name of the configuration
       * class' (the concrete class, not this one here) by default. So if you have a {@code com.acme.AppConfig} extending
       * {@link AbstractMongoConfiguration} the base package will be considered {@code com.acme} unless the method is
       * overriden to implement alternate behaviour.
       * 
       * @return the base package to scan for mapped {@link Document} classes or {@literal null} to not enable scanning for
       *         entities.
       */
      protected String getMappingBasePackage() {
    
          Package mappingBasePackage = getClass().getPackage();
          return mappingBasePackage == null ? null : mappingBasePackage.getName();
      }
    
      /**
       * Return {@link UserCredentials} to be used when connecting to the MongoDB instance or {@literal null} if none shall
       * be used.
       * 
       * @return
       * @deprecated since 1.7. {@link MongoClient} should hold authentication data within
       *             {@link MongoClient#getCredentialsList()}
       */
      @Deprecated
      protected UserCredentials getUserCredentials() {
          return null;
      }
    
      /**
       * Creates a {@link MongoMappingContext} equipped with entity classes scanned from the mapping base package.
       * 
       * @see #getMappingBasePackage()
       * @return
       * @throws ClassNotFoundException
       */
      @Bean
      public MongoMappingContext mongoMappingContext() throws ClassNotFoundException {
    
          MongoMappingContext mappingContext = new MongoMappingContext();
          mappingContext.setInitialEntitySet(getInitialEntitySet());
          mappingContext.setSimpleTypeHolder(customConversions().getSimpleTypeHolder());
          mappingContext.setFieldNamingStrategy(fieldNamingStrategy());
    
          return mappingContext;
      }
    
      /**
       * Returns a {@link MappingContextIsNewStrategyFactory} wrapped into a {@link CachingIsNewStrategyFactory}.
       * 
       * @return
       * @throws ClassNotFoundException
       */
      @Bean
      public IsNewStrategyFactory isNewStrategyFactory() throws ClassNotFoundException {
          return new CachingIsNewStrategyFactory(new MappingContextIsNewStrategyFactory(mongoMappingContext()));
      }
    
      /**
       * Register custom {@link Converter}s in a {@link CustomConversions} object if required. These
       * {@link CustomConversions} will be registered with the {@link #mappingMongoConverter()} and
       * {@link #mongoMappingContext()}. Returns an empty {@link CustomConversions} instance by default.
       * 
       * @return must not be {@literal null}.
       */
      @Bean
      public CustomConversions customConversions() {
          return new CustomConversions(Collections.emptyList());
      }
    
      /**
       * Creates a {@link MappingMongoConverter} using the configured {@link #mongoDbFactory()} and
       * {@link #mongoMappingContext()}. Will get {@link #customConversions()} applied.
       * 
       * @see #customConversions()
       * @see #mongoMappingContext()
       * @see #mongoDbFactory()
       * @return
       * @throws Exception
       */
      @Bean
      public MappingMongoConverter mappingMongoConverter() throws Exception {
    
          DbRefResolver dbRefResolver = new DefaultDbRefResolver(mongoDbFactory());
          MappingMongoConverter converter = new MappingMongoConverter(dbRefResolver, mongoMappingContext());
          converter.setCustomConversions(customConversions());
    
          return converter;
      }
    
      /**
       * Scans the mapping base package for classes annotated with {@link Document}.
       * 
       * @see #getMappingBasePackage()
       * @return
       * @throws ClassNotFoundException
       */
      protected Set<Class<?>> getInitialEntitySet() throws ClassNotFoundException {
    
          String basePackage = getMappingBasePackage();
          Set<Class<?>> initialEntitySet = new HashSet<Class<?>>();
    
          if (StringUtils.hasText(basePackage)) {
              ClassPathScanningCandidateComponentProvider componentProvider = new ClassPathScanningCandidateComponentProvider(
                      false);
              componentProvider.addIncludeFilter(new AnnotationTypeFilter(Document.class));
              componentProvider.addIncludeFilter(new AnnotationTypeFilter(Persistent.class));
    
              for (BeanDefinition candidate : componentProvider.findCandidateComponents(basePackage)) {
                  initialEntitySet.add(ClassUtils.forName(candidate.getBeanClassName(),
                          AbstractMongoConfiguration.class.getClassLoader()));
              }
          }
    
          return initialEntitySet;
      }
    
      /**
       * Configures whether to abbreviate field names for domain objects by configuring a
       * {@link CamelCaseAbbreviatingFieldNamingStrategy} on the {@link MongoMappingContext} instance created. For advanced
       * customization needs, consider overriding {@link #mappingMongoConverter()}.
       * 
       * @return
       */
      protected boolean abbreviateFieldNames() {
          return false;
      }
    
      /**
       * Configures a {@link FieldNamingStrategy} on the {@link MongoMappingContext} instance created.
       * 
       * @return
       * @since 1.5
       */
      protected FieldNamingStrategy fieldNamingStrategy() {
          return abbreviateFieldNames() ? new CamelCaseAbbreviatingFieldNamingStrategy()
                  : PropertyNameFieldNamingStrategy.INSTANCE;
      }
    
      public String getUri() {
        return uri;
      }
    
      public void setUri(String uri) {
        this.uri = uri;
      }
      
    }

3.业务当中的使用：

@Resource
    	@Qualifier("mongoTemplate")
    	private MongoOperations mongoOps;
    
    	@Resource
    	@Qualifier("cloudfsgrid")
    	private GridFsOperations gridFsOps;