From dedabf41a18820527aed9e77b75564e69c9030ce Mon Sep 17 00:00:00 2001 From: jackyzha0 Date: Sat, 9 May 2020 20:15:59 -0700 Subject: folder refactor --- db/mongo.go | 67 ------------------------------------------------------------- 1 file changed, 67 deletions(-) delete mode 100644 db/mongo.go (limited to 'db/mongo.go') 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 -} -- cgit v1.2.3