Golang и MongoDB: DeleteMany с фильтром - PullRequest
Новая
       26

Golang и MongoDB: DeleteMany с фильтром

2 голосов
/ 19 июня 2019

Я пытаюсь читать, записывать и удалять данные из приложения Go с помощью официального драйвера mongodb для go (go.mongodb.org/mongo-driver).

Вот моя структура, которую я хочу использовать: 

Contact struct {
    ID               xid.ID `json:"contact_id" bson:"contact_id"`
    SurName          string `json:"surname" bson:"surname"`
    PreName          string `json:"prename" bson:"prename"`
}
// xid is https://github.com/rs/xid

Я опускаю код для добавления в коллекцию, так как это работает find.

Я могу получить список контактов с конкретным идентификатором contact_id, используя следующий код (сокращенно): 

filter := bson.D{}
cursor, err := contactCollection.Find(nil, filter)
for cur.Next(context.TODO()) {
  ...
}

Это работает и возвращает документы. Я думал сделать то же самое для удаления или подобранного получения: 

// delete - abbreviated
filter := bson.M{"contact_id": id}
result, _ := contactCollection.DeleteMany(nil, filter)
// result.DeletedCount is always 0, err is nil
if err != nil {
    sendError(c, err) // helper function
    return
}

c.JSON(200, gin.H{
    "ok":      true,
    "message": fmt.Sprintf("deleted %d patients", result.DeletedCount),
}) // will be called, it is part of a webservice done with gin

// get complete
func Get(c *gin.Context) {
    defer c.Done()

    id := c.Param("id")
    filter := bson.M{"contact_id": id}

    cur, err := contactCollection.Find(nil, filter)
    if err != nil {
        sendError(c, err) // helper function
        return
    } // no error

    contacts := make([]types.Contact, 0)
    for cur.Next(context.TODO()) { // nothing returned
        // create a value into which the single document can be decoded
        var elem types.Contact
        err := cur.Decode(&elem)
        if err != nil {
            sendError(c, err) // helper function
            return
        }
        contacts = append(contacts, elem)
    }

    c.JSON(200, contacts)
}

Почему тот же фильтр не работает при удалении?

Редактировать: Вставить код выглядит следующим образом: 

_, _ = contactCollection.InsertOne(context.TODO(), Contact{
    ID: "abcdefg",
    SurName: "Demo",
    PreName: "on stackoverflow",
})

Ответы [ 2 ]

2 голосов
/ 19 июня 2019

Contact.ID имеет тип xid.ID, который является байтовым массивом: 

type ID [rawLen]byte

Таким образом, код вставки, который вы указали, где вы используете литерал string для указания значения поля ID, будет ошибкой во время компиляции: 

_, _ = contactCollection.InsertOne(context.TODO(), Contact{
    ID: "abcdefg",
    SurName: "Demo",
    PreName: "on stackoverflow",
})

Позже в ваших комментариях вы пояснили, что приведенный выше код вставки был просто примером, а не тем, как вы на самом деле это делаете. В своем реальном коде вы отменяете контакт (или его поле идентификатора) из запроса.

xid.ID имеет собственную логику демаршалинга, которая может интерпретировать входные данные по-разному и может привести к ID, представляющему значение string, отличное от вашего ввода. ID.UnmarshalJSON() определяет, как идентификатор string будет преобразован в xid.ID: 

func (id *ID) UnmarshalJSON(b []byte) error {
    s := string(b)
    if s == "null" {
        *id = nilID
        return nil
    }
    return id.UnmarshalText(b[1 : len(b)-1])
}

Как видите, первый байт обрезан, и ID.UnmarshalText() делает с ним еще больше "магии" (проверьте источник, если вам интересно).

В целом, чтобы избежать таких «преобразований» в фоновом режиме без вашего ведома, используйте простой тип string для своего идентификатора и выполняйте необходимые преобразования самостоятельно, где бы вам ни понадобилось хранить / передавать свой идентификатор.

0 голосов
/ 19 июня 2019

Для поля идентификатора вы должны использовать primitive.ObjectID, предоставляемый пакетом bson. 

"go.mongodb.org/mongo-driver/bson/primitive"

ID          primitive.ObjectID `json:"_id" bson:"_id"`
...