diff options
Diffstat (limited to 'backend/db/mongo.go')
| -rw-r--r-- | backend/db/mongo.go | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/backend/db/mongo.go b/backend/db/mongo.go index dc1dd25..f7870ac 100644 --- a/backend/db/mongo.go +++ b/backend/db/mongo.go @@ -13,6 +13,7 @@ import ( "go.mongodb.org/mongo-driver/mongo/readpref" ) +var Client *mongo.Client var Session *mongo.Session var pastes *mongo.Collection @@ -25,21 +26,16 @@ func initSessions(user, pass, ip string) { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() - client, err := mongo.Connect(ctx, options.Client().ApplyURI(mongoURI).SetTLSConfig(&tls.Config{})) + c, err := mongo.Connect(ctx, options.Client().ApplyURI(mongoURI).SetTLSConfig(&tls.Config{})) + Client = c if err != nil { log.Fatalf("error establishing connection to mongo: %s", err.Error()) } - err = client.Ping(ctx, readpref.Primary()) + err = Client.Ping(ctx, readpref.Primary()) if err != nil { log.Fatalf("error pinging mongo: %s", err.Error()) } - defer func() { - if err = client.Disconnect(ctx); err != nil { - panic(err) - } - }() - // ensure expiry check expiryIndex := options.Index().SetExpireAfterSeconds(0) sessionTTL := mongo.IndexModel{ @@ -55,7 +51,7 @@ func initSessions(user, pass, ip string) { } // Define connection to Databases - pastes = client.Database("main").Collection("pastes") + pastes = Client.Database("main").Collection("pastes") _, _ = pastes.Indexes().CreateOne(ctx, sessionTTL) _, _ = pastes.Indexes().CreateOne(ctx, uniqueHashes) } |