基于Golang实现MongoDB配置对象的数组元素增删操作

1. 添加新的配置数组（如rssi） #

根据需求分两种场景处理：

场景1：新增整个数组字段 #

如果要给configuration添加一个全新的数组（比如rssi），直接用$set操作符设置字段值即可：

import (
    "context"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
)

func addNewConfigArray(collection *mongo.Collection, docID interface{}) error {
    // 定义rssi数组的初始内容
    rssiArray := []bson.M{
        {
            "ident":       0,
            "description": "Signal Strength",
            "sensor_sn":   "abc123",
            "unit":        "dBm",
            "type":        "number",
            "elemID":      "rssi0",
        },
    }

    update := bson.M{
        "$set": bson.M{"configuration.rssi": rssiArray},
    }

    _, err := collection.UpdateByID(context.TODO(), docID, update)
    return err
}

场景2：往已存在的数组追加元素 #

如果目标数组已经存在，要新增传感器对象到数组中，使用$push操作符：

func pushToConfigArray(collection *mongo.Collection, docID interface{}, sensor bson.M) error {
    update := bson.M{
        "$push": bson.M{"configuration.rssi": sensor},
    }

    _, err := collection.UpdateByID(context.TODO(), docID, update)
    return err
}

// 使用示例
newSensor := bson.M{
    "ident":       1,
    "description": "Signal Strength",
    "sensor_sn":   "def456",
    "unit":        "dBm",
    "type":        "number",
    "elemID":      "rssi1",
}
err := pushToConfigArray(collection, docID, newSensor)

2. 删除数组内的对象（基于sensor_sn） #

用$pull操作符匹配sensor_sn字段，移除对应数组元素：

func removeSensorFromArray(collection *mongo.Collection, docID interface{}, configKey string, sensorSN string) error {
    // configKey为ta/rh/rssi这类数组字段名
    pullPath := "configuration." + configKey
    update := bson.M{
        "$pull": bson.M{pullPath: bson.M{"sensor_sn": sensorSN}},
    }

    _, err := collection.UpdateByID(context.TODO(), docID, update)
    return err
}

// 使用示例：删除ta数组中sensor_sn为"828082837284"的对象
err := removeSensorFromArray(collection, docID, "ta", "828082837284")

3. 补充：更新数组内的对象（基于sensor_sn） #

如果需要修改数组内某传感器的信息，结合过滤条件和$set操作：

func updateSensorInArray(collection *mongo.Collection, docID interface{}, configKey string, sensorSN string, updates bson.M) error {
    filter := bson.M{
        "_id": docID,
        "configuration." + configKey + ".sensor_sn": sensorSN,
    }

    update := bson.M{
        "$set": bson.M{"configuration." + configKey + ".$": updates},
    }

    _, err := collection.UpdateOne(context.TODO(), filter, update)
    return err
}

// 使用示例：更新ta数组中指定sensor_sn的描述和单位
updates := bson.M{
    "description": "Updated Temperature",
    "unit":        "°C",
}
err := updateSensorInArray(collection, docID, "ta", "828082837284", updates)

注意事项 #

• 操作时必须指定文档唯一标识（如_id），避免误修改其他文档
• 若sensor_sn为空字符串，删除/更新时需匹配空值条件
• 确保正确传递context，避免数据库操作超时

内容的提问来源于stack exchange，提问作者Bruce