|
@@ -1,11 +1,11 @@
|
|
|
package org.ssssssss.magicapi.spring.boot.starter;
|
|
|
|
|
|
-import com.mongodb.MongoClient;
|
|
|
import com.mongodb.client.FindIterable;
|
|
|
import com.mongodb.client.MongoCollection;
|
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
+import org.springframework.data.mongodb.core.MongoTemplate;
|
|
|
import org.ssssssss.magicapi.modules.MongoCollectionExtension;
|
|
|
import org.ssssssss.magicapi.modules.MongoFindIterableExtension;
|
|
|
import org.ssssssss.magicapi.modules.MongoModule;
|
|
@@ -14,17 +14,17 @@ import org.ssssssss.script.reflection.AbstractReflection;
|
|
|
/**
|
|
|
* mongo配置
|
|
|
*/
|
|
|
-@ConditionalOnBean(MongoClient.class)
|
|
|
@Configuration
|
|
|
+@ConditionalOnBean(MongoTemplate.class)
|
|
|
public class MagicMongoAutoConfiguration {
|
|
|
|
|
|
/**
|
|
|
* 注入mongo模块
|
|
|
*/
|
|
|
@Bean
|
|
|
- public MongoModule mongoFunctions(MongoClient mongoClient) {
|
|
|
+ public MongoModule mongoFunctions(MongoTemplate mongoTemplate) {
|
|
|
AbstractReflection.getInstance().registerExtensionClass(MongoCollection.class, MongoCollectionExtension.class);
|
|
|
AbstractReflection.getInstance().registerExtensionClass(FindIterable.class, MongoFindIterableExtension.class);
|
|
|
- return new MongoModule(mongoClient);
|
|
|
+ return new MongoModule(mongoTemplate);
|
|
|
}
|
|
|
}
|