From bc6947f41facc3ef62fe8f2611551f48a7e3788b Mon Sep 17 00:00:00 2001 From: jackyzha0 Date: Sat, 9 May 2020 12:47:39 -0700 Subject: working con to mongo --- db/mongo.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 db/mongo.go (limited to 'db/mongo.go') diff --git a/db/mongo.go b/db/mongo.go new file mode 100644 index 0000000..4c3b88a --- /dev/null +++ b/db/mongo.go @@ -0,0 +1,39 @@ +package db + +import ( + "crypto/tls" + "fmt" + "net" + + "github.com/globalsign/mgo" + log "github.com/sirupsen/logrus" +) + +var Session *mgo.Session +var TextDB *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()) + } + + // Define connection to Databases + TextDB = Session.DB("main").C("pastes") +} -- cgit v1.2.3