add AddUserInChat
This commit is contained in:
parent
6ccd5b478e
commit
3005afda25
22
db/db.go
22
db/db.go
|
@ -39,3 +39,25 @@ func AddChat(chatID int64) error {
|
|||
}
|
||||
return result.Err()
|
||||
}
|
||||
|
||||
func AddUserInChat(chatID int64, username string) error {
|
||||
if err := AddChat(chatID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
||||
defer cancel()
|
||||
|
||||
client, err := mongo.Connect(ctx, options.Client().ApplyURI(cfg.MongoURL))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
collection := client.Database(cfg.DBName).Collection(cfg.ChatsCollectionName)
|
||||
_, err = collection.UpdateOne(
|
||||
ctx,
|
||||
bson.E{Key: "chat_id", Value: chatID},
|
||||
bson.M{"addToSet": bson.M{"users": username}},
|
||||
)
|
||||
return err
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue