aboutsummaryrefslogtreecommitdiff
path: root/db/mongo.go
diff options
context:
space:
mode:
authorjackyzha0 <[email protected]>2020-05-09 20:15:59 -0700
committerjackyzha0 <[email protected]>2020-05-09 20:15:59 -0700
commitdedabf41a18820527aed9e77b75564e69c9030ce (patch)
tree9258832576f9c1bcad6126cf76ef8f7f6a128e82 /db/mongo.go
parentMerge pull request #3 from jackyzha0/doc-expiry (diff)
downloadctrl-v-dedabf41a18820527aed9e77b75564e69c9030ce.tar.xz
ctrl-v-dedabf41a18820527aed9e77b75564e69c9030ce.zip
folder refactor
Diffstat (limited to 'db/mongo.go')
-rw-r--r--db/mongo.go67
1 files changed, 0 insertions, 67 deletions
diff --git a/db/mongo.go b/db/mongo.go
deleted file mode 100644
index 4c8a739..0000000
--- a/db/mongo.go
+++ /dev/null
@@ -1,67 +0,0 @@
-package db
-
-import (
- "crypto/tls"
- "fmt"
- "net"
-
- "github.com/globalsign/mgo"
- "github.com/globalsign/mgo/bson"
- log "github.com/sirupsen/logrus"
-)
-
-var Session *mgo.Session
-var pastes *mgo.Collection
-
-func initSessions(user, pass, ip string) {
- log.Infof("attempting connection to %s", ip)
-
- // build uri string
- URIfmt := "mongodb://%s:%s@%s:27017"
- mongoURI := fmt.Sprintf(URIfmt, user, pass, ip)
- dialInfo, err := mgo.ParseURL(mongoURI)
- if err != nil {
- log.Fatalf("error parsing uri: %s", err.Error())
- }
-
- tlsConfig := &tls.Config{}
- dialInfo.DialServer = func(addr *mgo.ServerAddr) (net.Conn, error) {
- conn, err := tls.Dial("tcp", addr.String(), tlsConfig)
- return conn, err
- }
-
- Session, err = mgo.DialWithInfo(dialInfo)
- if err != nil {
- 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")
-}
-
-func insert(new Paste) error {
- return pastes.Insert(new)
-}
-
-func fetch(hash string) (Paste, error) {
- p := Paste{}
-
- q := bson.M{"hash": hash}
- err := pastes.Find(q).One(&p)
- return p, err
-}