68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
|
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
|
||
|
|
}
|
||
|
|
}
|