mirror-nourybot/pkg/db/connect.go

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)

// }