1
|
package eu.dnetlib.uoamonitorservice.configuration.mongo;
|
2
|
|
3
|
import com.mongodb.MongoClient;
|
4
|
import com.mongodb.MongoCredential;
|
5
|
import com.mongodb.ServerAddress;
|
6
|
import eu.dnetlib.uoamonitorservice.configuration.properties.MongoConfig;
|
7
|
import org.springframework.beans.factory.annotation.Autowired;
|
8
|
import org.springframework.context.annotation.Bean;
|
9
|
import org.springframework.context.annotation.Configuration;
|
10
|
import org.springframework.context.annotation.Primary;
|
11
|
import org.springframework.data.mongodb.MongoDbFactory;
|
12
|
import org.springframework.data.mongodb.core.MongoTemplate;
|
13
|
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
|
14
|
import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
|
15
|
|
16
|
import java.util.Collections;
|
17
|
|
18
|
@Configuration
|
19
|
@EnableMongoRepositories(basePackages = {"eu.dnetlib.uoamonitorservice.dao"})
|
20
|
public class MongoConnection {
|
21
|
|
22
|
@Autowired
|
23
|
private MongoConfig mongoConfig;
|
24
|
|
25
|
@Bean
|
26
|
@Primary
|
27
|
public MongoDbFactory mongoDbFactory() {
|
28
|
return new SimpleMongoDbFactory(getMongoClient(), mongoConfig.getDatabase());
|
29
|
}
|
30
|
|
31
|
@Bean(name = "mongoTemplate")
|
32
|
@Primary
|
33
|
public MongoTemplate getMongoTemplate() {
|
34
|
return new MongoTemplate(mongoDbFactory());
|
35
|
}
|
36
|
|
37
|
private MongoClient getMongoClient() {
|
38
|
if(mongoConfig.getUsername() != null && mongoConfig.getPassword() != null){
|
39
|
return new MongoClient(Collections.singletonList(
|
40
|
new ServerAddress(mongoConfig.getHost(), mongoConfig.getPort())),
|
41
|
Collections.singletonList(MongoCredential.createCredential(mongoConfig.getUsername(), mongoConfig.getDatabase(), mongoConfig.getPassword().toCharArray())));
|
42
|
} else {
|
43
|
return new MongoClient(Collections.singletonList(new ServerAddress(mongoConfig.getHost(), mongoConfig.getPort())));
|
44
|
}
|
45
|
}
|
46
|
}
|