diff options
| author | jackyzha0 <[email protected]> | 2020-05-09 17:40:41 -0700 |
|---|---|---|
| committer | jackyzha0 <[email protected]> | 2020-05-09 17:40:41 -0700 |
| commit | 66727df2c1aa369b9250e7596fea6c8ae8385eaa (patch) | |
| tree | 540dc43adfe6f346490993d5d81b42563aeaeada /db/mongo.go | |
| parent | Merge pull request #2 from jackyzha0/api (diff) | |
| download | ctrl-v-66727df2c1aa369b9250e7596fea6c8ae8385eaa.tar.xz ctrl-v-66727df2c1aa369b9250e7596fea6c8ae8385eaa.zip | |
add document ttl
Diffstat (limited to 'db/mongo.go')
| -rw-r--r-- | db/mongo.go | 15 |
1 files changed, 15 insertions, 0 deletions
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") } |