在 Go 中使用 MongoDb 默认驱动程序创建唯一索引

Question

我在使用Go 的官方 MongoDB 驱动程序为某些数据创建唯一索引时遇到一些问题。

所以我有一个这样的结构：

type Product struct {
    ID        primitive.ObjectID `json:"_id" bson:"_id"`
    Name      string             `json:"name" bson:"name"`
    Price     float64            `json:"price" bson:"price"`
    Attribute []Attribute        `json:"attribute" bson:"attribute"`
    Category  string             `json:"category" bson:"category"`
}

然后我想为该name属性创建一个唯一索引。我尝试在我的函数中做类似的事情Create（对于产品）

func Create(c echo.Context) error {

    //unique index here 
    indexModel, err := productCollection.Indexes().CreateOne(context.Background(),
        IndexModel{
            Keys:    bsonx.Doc{{"name", bsonx.Int32(1)}},
            Options: options.Index().SetUnique(true),
        })
    if err != nil {
        log.Fatalf("something went wrong: %+v", err)
    }

    //create the product here
    p := new(Product)
    if err := c.Bind(p); err != nil {
        log.Fatalf("Could not bind request to struct: %+v", err)
        return util.SendError(c, "500", "something went wrong", "failed")
    }
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    result, _ := productCollection.InsertOne(ctx, p)

    return util.SendSuccess(c, result.InsertedID)

}

问题是我不完全知道如何indexModel在创建产品之前作为上下文中的选项传递。另外，我不确定我正在做的事情是否只创建一次索引（这就是我想要做的）。如果我能指出如何做到这一点的正确方向，我将不胜感激。

我正在使用 Go 的 echo 框架，以防万一这提供了更多上下文。

Answer 1

您的代码将在每次调用函数时尝试创建索引Create，这可以工作，但效率不是很高，因为索引只需要创建一次（或者如果索引不存在）。我建议让索引的创建由设置数据库连接的函数负责，或者由初始化数据库连接后立即运行的函数负责，这样它只运行一次（当程序启动时）。

由于该collection.Indexes().CreateOne(...)方法在尝试创建已存在的索引时不会返回错误，因此您可以使用此方法来确保索引存在，类似于非官方 MongoDB 驱动程序的功能EnsureIndex。

或者，您可以按照 Cahaba Data 在其答案中建议的做法，即使索引创建成为一项一次性数据库管理任务，而不是您的应用程序的责任。我之前曾与一些同事就此进行过讨论，虽然我确实同意这在技术上是一项一次性数据库管理任务，但我仍然更喜欢让我的应用程序在启动时确保它所需的索引实际上在那里，创建如果他们不是的话。毕竟，如果有人忘记或创建了错误的索引，我的应用程序可能会出现故障/瘫痪。它还提供了关于您的应用程序需要/使用哪些索引的良好文档，而不必与数据库管理员进行沟通并希望他/她/它会正确记录它。当然，请随意形成您自己的意见:)