如何检查 MongoDB Golang 集合是否存在

Question

我是 GO 语言的新手，我正在使用 MongoDB。我正在 Angular 4 上为应用程序及其前端创建后端。我想检查集合是否存在。

这是我的代码，我已经使用 nil 检查了它。

collection := GetCollection("users")    
fmt.Println("collection", collection)   
if collection == nil {      
   fmt.Println("Collection is empty")   
}

我创建了一个 GetCollection 函数，当我们向它传递集合名称时，它会返回一个集合。 那么，如果没有集合，我如何检查它是否存在？ 我尝试了很多事情但都失败了。

Answer 1

您可以简单地使用

Database.CollectionNames()

方法返回给定数据库中存在的集合名称。它返回一个切片，您必须在其中检查您的集合是否已列出。

sess := ... // obtain session
db := sess.DB("") // Get db, use db name if not given in connection url

names, err := db.CollectionNames()
if err != nil {
    // Handle error
    log.Printf("Failed to get coll names: %v", err)
    return
}

// Simply search in the names slice, e.g.
for _, name := range names {
    if name == "collectionToCheck" {
        log.Printf("The collection exists!")
        break
    }
}

但正如 Neil Lunn 在他的评论中所写，你不应该需要这个。您应该更改逻辑以使用 MongoDB，而不是依赖此检查。如果您尝试插入文档，则会自动创建集合，并且从不存在的集合中查询不会产生错误（当然也不会产生结果）。

Answer 2

您可以使用 go.mongodb.org/mongo-driver 包中的 ListCollectionNames 函数。

示例代码：

func dbHasCollection(db *mongo.Database,collectionName string) bool{
    coll, _ := db.ListCollectionNames(context.Background(), bson.D{{"name", collectionName}})
    return len(coll)==1
}