package dao import ( "cds/models" "context" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/mongo" ) func GetGroupById(id string) (group *models.Group, err error) { db, _ := get() objectId, err := primitive.ObjectIDFromHex(id) defer disconnect(db.Client()) coll := db.Collection("groups") var result models.Group err = coll.FindOne(context.TODO(), bson.D{{"_id", objectId}}).Decode(&result) if err != nil { return nil, err } else { return &result, nil } } func GetGroupByMember(id string) (group []models.Group, err error) { db, _ := get() objectId, err := primitive.ObjectIDFromHex(id) defer disconnect(db.Client()) coll := db.Collection("groups") cursor, err := coll.Find(context.TODO(), bson.D{{"members", objectId}}) defer cursor.Close(context.TODO()) if err != nil { return nil, err } err = cursor.All(context.TODO(), &group) if err != nil { return nil, err } else { return group, nil } } func CheckMemberInGroup(idUser string, idGroup string) (ret bool, err error) { db, _ := get() userObject, err := primitive.ObjectIDFromHex(idUser) groupObject, err := primitive.ObjectIDFromHex(idGroup) defer disconnect(db.Client()) coll := db.Collection("groups") err = coll.FindOne(context.TODO(), bson.D{ {"_id", groupObject}, {"members", userObject}, }).Err() if err == mongo.ErrNoDocuments { return false, nil } else if err != nil { return false, err } else { return true, nil } }