noury/mirror-nourybot
1
0
Fork 0
mirror of https://github.com/lyx0/nourybot.git synced 2024-11-13 19:49:55 +01:00
Code Issues Projects Releases Packages Wiki Activity

use cancel to not leak memory

Browse source
This commit is contained in:
lyx0 2021-10-28 15:22:19 +02:00
parent df2aad908f
commit 20f9470ce8
3 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pkg/db/addchannel.go
View file

@ -13,7 +13,9 @@ func AddChannel(channelName string) {
client := Connect() client := Connect()
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
/* /*
Get my collection instance Get my collection instance
*/ */

5
pkg/db/connect.go
View file

@ -19,7 +19,10 @@ func Connect() *mongo.Client {
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
err = client.Connect(ctx) err = client.Connect(ctx)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)

4
pkg/db/initialjoin.go
View file

@ -12,7 +12,9 @@ func InitialJoin(nb *bot.Bot) {
client := Connect() client := Connect()
collection := client.Database("nourybot").Collection("channels") collection := client.Database("nourybot").Collection("channels")
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
defer client.Disconnect(ctx) defer client.Disconnect(ctx)
cur, currErr := collection.Find(ctx, bson.D{}) cur, currErr := collection.Find(ctx, bson.D{})