package db import ( "context" "time" "github.com/lyx0/nourybot/pkg/config" log "github.com/sirupsen/logrus" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" ) type Channel struct { Name string `bson:"name,omitempty"` Connect bool `bson:"connect,omitempty"` } func Connect() *mongo.Client { conf := config.LoadConfig() client, err := mongo.NewClient(options.Client().ApplyURI(conf.MongoURI)) if err != nil { log.Fatal(err) } ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() err = client.Connect(ctx) if err != nil { log.Fatal(err) } // defer client.Disconnect(ctx) err = client.Ping(context.TODO(), nil) if err != nil { log.Fatal(err) } log.Info("Connected to MongoDB!") databases, err := client.ListDatabaseNames(ctx, bson.M{}) if err != nil { log.Fatal(err) } _ = databases // log.Info(databases) return client } /* List databases */ /* Iterate a cursor */ // log.Info(channels) // import ( // "context" // "fmt" // "time" // "github.com/lyx0/nourybot/pkg/config" // log "github.com/sirupsen/logrus" // "go.mongodb.org/mongo-driver/bson" // "go.mongodb.org/mongo-driver/mongo" // "go.mongodb.org/mongo-driver/mongo/options" // ) // func Connect() { // conf := config.LoadConfig() // client, err := mongo.NewClient(options.Client().ApplyURI(conf.MongoURI)) // if err != nil { // log.Fatal(err) // } // ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) // err = client.Connect(ctx) // if err != nil { // log.Fatal(err) // } // defer client.Disconnect(ctx) // /* // List databases // */ // databases, err := client.ListDatabaseNames(ctx, bson.M{}) // if err != nil { // log.Fatal(err) // } // fmt.Println(databases) // // Interact with data // type Channel struct { // Name string `bson:"name,omitempty"` // } // /* // Get my collection instance // */ // collection := client.Database("nourybot").Collection("channels") // /* // Insert channel // */ // // chnl := []interface{}{ // // bson.D{{"name", "nouryqt"}}, // // bson.D{{"name", "nourybot"}}, // // } // // res, insertErr := collection.InsertMany(ctx, chnl) // // if insertErr != nil { // // log.Fatal(insertErr) // // } // // log.Info(res) // /* // Iterate a cursor // */ // cur, currErr := collection.Find(ctx, bson.D{}) // if currErr != nil { // panic(currErr) // } // defer cur.Close(ctx) // var channels []Channel // if err = cur.All(ctx, &channels); err != nil { // panic(err) // } // log.Info(channels) // }