From 66727df2c1aa369b9250e7596fea6c8ae8385eaa Mon Sep 17 00:00:00 2001 From: jackyzha0 Date: Sat, 9 May 2020 17:40:41 -0700 Subject: add document ttl --- db/mongo.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'db/mongo.go') diff --git a/db/mongo.go b/db/mongo.go index 83e71f7..4c8a739 100644 --- a/db/mongo.go +++ b/db/mongo.go @@ -35,6 +35,21 @@ func initSessions(user, pass, ip string) { log.Fatalf("error establishing connection to mongo: %s", err.Error()) } + // ensure expiry check + sessionTTL := mgo.Index{ + Key: []string{"expiry"}, + ExpireAfter: 0, + } + + // ensure hashes are unique + uniqueHashes := mgo.Index{ + Key: []string{"hash"}, + Unique: true, + } + + _ = Session.DB("main").C("pastes").EnsureIndex(sessionTTL) + _ = Session.DB("main").C("pastes").EnsureIndex(uniqueHashes) + // Define connection to Databases pastes = Session.DB("main").C("pastes") } -- cgit v1.2.3