diff options
| author | Ryan Mehri <[email protected]> | 2020-05-09 18:59:03 -0600 |
|---|---|---|
| committer | GitHub <[email protected]> | 2020-05-09 18:59:03 -0600 |
| commit | 4a748323975922e0cb559c9623572e251b4a5db0 (patch) | |
| tree | 540dc43adfe6f346490993d5d81b42563aeaeada /db/mongo.go | |
| parent | Merge pull request #2 from jackyzha0/api (diff) | |
| parent | add document ttl (diff) | |
| download | ctrl-v-4a748323975922e0cb559c9623572e251b4a5db0.tar.xz ctrl-v-4a748323975922e0cb559c9623572e251b4a5db0.zip | |
Merge pull request #3 from jackyzha0/doc-expiry
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") } |