|
|
package main
import ( "encoding/json" "fmt" "io/ioutil"
mgo "gopkg.in/mgo.v2" )
//MongoConfig stores the configuration of mongodb to connect
type MongoConfig struct { Ip string `json:"ip"` Database string `json:"database"` }
var mongoConfig MongoConfig
func readMongodbConfig(path string) { file, e := ioutil.ReadFile(path) if e != nil { fmt.Println("error:", e) } content := string(file) json.Unmarshal([]byte(content), &mongoConfig) }
func getSession() (*mgo.Session, error) { session, err := mgo.Dial("mongodb://" + mongoConfig.Ip) if err != nil { panic(err) } //defer session.Close()
// Optional. Switch the session to a monotonic behavior.
session.SetMode(mgo.Monotonic, true)
// Optional. Switch the session to a monotonic behavior.
session.SetMode(mgo.Monotonic, true)
return session, err } func getCollection(session *mgo.Session, collection string) *mgo.Collection {
c := session.DB(mongoConfig.Database).C(collection) return c }
|