|
|
package main
import ( "os"
"github.com/fatih/color" mgo "gopkg.in/mgo.v2" )
const mongoip = "127.0.0.1:27017"
func main() { session, err := getSession() check(err)
var databases []string if len(os.Args) > 1 { for i, arg := range os.Args { if i > 0 { databases = append(databases, arg) } } } for _, database := range databases { db := getDatabase(session, database) color.Yellow("delete database: " + database) err := db.DropDatabase() check(err) } }
func getSession() (*mgo.Session, error) { session, err := mgo.Dial("mongodb://" + mongoip) 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 getDatabase(session *mgo.Session, database string) *mgo.Database {
D := session.DB(database) return D } func getCollection(session *mgo.Session, database string, collection string) *mgo.Collection {
c := session.DB(database).C(collection) return c }
|