diff --git a/mongo-plus-core/src/main/java/com/mongoplus/manager/MongoPlusClient.java b/mongo-plus-core/src/main/java/com/mongoplus/manager/MongoPlusClient.java index e26557a19e4ab0852d33e3fc85263fee55b26e1b..953a113f1bd9fa830a644d699e5cc0541a8cdd44 100644 --- a/mongo-plus-core/src/main/java/com/mongoplus/manager/MongoPlusClient.java +++ b/mongo-plus-core/src/main/java/com/mongoplus/manager/MongoPlusClient.java @@ -12,10 +12,7 @@ import com.mongoplus.model.BaseProperty; import com.mongoplus.toolkit.StringUtils; import org.bson.Document; -import java.util.ArrayList; -import java.util.Arrays; -import java.util.List; -import java.util.Map; +import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.stream.Collectors; @@ -198,6 +195,15 @@ public class MongoPlusClient { this.mongoDatabase = mongoDatabase; } + + public void dropCollection(Class clazz) { + Optional.ofNullable(getCollectionManager(clazz).getCollection(clazz)).ifPresent(MongoCollection::drop); + } + + public void dropCollection(String database, String collectionName) { + Optional.ofNullable(getCollectionManager(database).getCollection(collectionName)).ifPresent(MongoCollection::drop); + } + @Override public String toString() { return "ConnectionManager{" +