Sam的程式筆記

buildscript {
    ext {
        springBootVersion = '1.4.3.RELEASE'
    }
    repositories {
        mavenCentral()
    }
    dependencies {
        classpath("org.springframework.boot:spring-boot-gradle-plugin:${springBootVersion}")
    }
}

apply plugin: 'java'
apply plugin: 'eclipse'
apply plugin: 'org.springframework.boot'

jar {
    baseName = 'convert'
    version = '0.0.1-SNAPSHOT'
}

sourceCompatibility = 1.8

repositories {
    mavenCentral()
}

dependencies {
    compile('org.springframework.boot:spring-boot-starter-actuator')
    compile('org.springframework.boot:spring-boot-starter-data-mongodb')
    compileOnly('org.projectlombok:lombok')
    testCompile('org.springframework.boot:spring-boot-starter-test')
}

mongodb:

  primary:

    uri: mongodb://samzhu:123456@ds123456.mlab.com:59217/dbname

  secondary:

    uri: mongodb://samzhu:123456@ds123-a0.mlab.com:21971,ds456-a1.mlab.com:21971/dbname

@Data
@ConfigurationProperties(prefix = "mongodb")
public class MultipleMongoProperties {
    private MongoProperties primary = new MongoProperties();
    private MongoProperties secondary = new MongoProperties();
}

@Configuration
@EnableMongoRepositories(basePackages = "com.giish.convert.repositories.main", mongoTemplateRef = PrimaryMongoConfig.MONGO_TEMPLATE)
public class PrimaryMongoConfig {
    public static final String MONGO_TEMPLATE = "primaryMongoTemplate";
}

@Configuration
@EnableMongoRepositories(basePackages = "com.giish.convert.repositories.secondary", mongoTemplateRef = SecondaryMongoConfig.MONGO_TEMPLATE)
public class SecondaryMongoConfig {
    protected static final String MONGO_TEMPLATE = "secondaryMongoTemplate";
}

import com.mongodb.MongoClientURI;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.mongo.MongoProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
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;

/**
 * Created by samchu on 2017/1/3.
 */

@Configuration
public class MultipleMongoConfig {
    private MultipleMongoProperties mongoProperties;
        
    // @SuppressWarnings 加這個注解讓IDE 不報: Could not autowire

    @Autowired
    public void setMongoProperties(@SuppressWarnings("SpringJavaAutowiringInspection") MultipleMongoProperties mongoProperties) {
        this.mongoProperties = mongoProperties;
    }

    @Primary
    @Bean(name = PrimaryMongoConfig.MONGO_TEMPLATE)
    public MongoTemplate primaryMongoTemplate() throws Exception {
        return new MongoTemplate(primaryFactory(this.mongoProperties.getPrimary()));
    }

    @Bean
    @Qualifier(SecondaryMongoConfig.MONGO_TEMPLATE)
    public MongoTemplate secondaryMongoTemplate() throws Exception {
        return new MongoTemplate(secondaryFactory(this.mongoProperties.getSecondary()));
    }

    @Bean
    @Primary
    public MongoDbFactory primaryFactory(@SuppressWarnings("SpringJavaAutowiringInspection") MongoProperties mongoProperties) throws Exception {
//        這邊是使用另一種方式連接

//        MongoCredential credential = MongoCredential.createCredential("username", "databasename", "password".toCharArray());

//        ServerAddress serverAdress = new ServerAddress(mongo.getHost(), mongo.getPort());

//        new SimpleMongoDbFactory(new MongoClient(serverAdress, Arrays.asList(credential)), mongo.getDatabase());

        return new SimpleMongoDbFactory(new MongoClientURI(mongoProperties.getUri()));
    }

    @Bean
    public MongoDbFactory secondaryFactory(@SuppressWarnings("SpringJavaAutowiringInspection") MongoProperties mongoProperties) throws Exception {
        return new SimpleMongoDbFactory(new MongoClientURI(mongoProperties.getUri()));
    }
}

@EnableConfigurationProperties(MultipleMongoProperties.class)
@SpringBootApplication(exclude = MongoAutoConfiguration.class)
public class GiishConvertApplication {
    public static void main(String[] args) {
        SpringApplication.run(GiishConvertApplication.class, args);
    }
}

@RequiredArgsConstructor
@Component
public class ApplicationLoader implements CommandLineRunner {
    //@Autowired

    @NonNull
    private com.giish.convert.repositories.main.UserRep mainUserRep;
    //@Autowired

    @NonNull
    private com.giish.convert.repositories.secondary.OldUserRep secondaryUserRep;

    @Override
    public void run(String... args) throws Exception {
        // 取得主資料庫

        System.out.println(mainUserRep.count());
        // 取得從資料庫

        System.out.println(secondaryUserRep.count());
    }
}