修复mongo模块缺少databasecollection等方法的BUG

This commit is contained in:
mxd 2022-08-26 19:59:12 +08:00
parent 07062044d1
commit 48a1ba13ad

View File

@ -10,6 +10,7 @@ import org.slf4j.LoggerFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.ssssssss.magicapi.core.config.Constants;
import org.ssssssss.magicapi.core.annotation.MagicModule;
import org.ssssssss.script.annotation.Comment;
import org.ssssssss.script.convert.ClassImplicitConvert;
import org.ssssssss.script.functions.DynamicAttribute;
import org.ssssssss.script.reflection.JavaInvoker;
@ -55,6 +56,11 @@ public class MongoModule implements ClassImplicitConvert, DynamicAttribute<Mongo
JavaReflection.registerImplicitConvert(this);
}
@Comment("获取`database`")
public MongoDataBaseGetter database(String databaseName){
return getDynamicAttribute(databaseName);
}
@Override
@Transient
public MongoDataBaseGetter getDynamicAttribute(String databaseName) {
@ -92,5 +98,11 @@ public class MongoModule implements ClassImplicitConvert, DynamicAttribute<Mongo
public MongoCollection<Document> getDynamicAttribute(String key) {
return database.getCollection(key);
}
@Comment("获取`Collection`")
public MongoCollection<Document> collection(String key){
return getDynamicAttribute(key);
}
}
}