initmongo.go

package main

import (
	"context"
	"fmt"
	"io/ioutil"
	"log"
	"strings"
	"time"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/mongo/readpref"
)

const (
	todoList = "todolist"
	category = "category"
	db       = "mtdlBot"
	URI      = "uri.txt"
)

func GetURI(filename string) string {
	uri, err := ioutil.ReadFile(filename)
	if err != nil {
		panic(err)
	}
	uriRep := strings.Replace(string(uri), "\n", "", 1)
	return uriRep
}

func InitMongo() (*mongo.Client, *mongo.Collection, *mongo.Collection) {
	//Создаём нового клиента базы данных с поключением по указанному URL
	client, err := mongo.NewClient(options.Client().
		ApplyURI(GetURI(URI)))
	if err != nil {
		log.Fatal(err)
	}

	//Подключаемся к базе данных
	fmt.Println("Client connecting...")
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second) //ждём ответа в течении 10 сек
	err = client.Connect(ctx)                                           //если ответа нет, вернёт ошибку
	if err != nil {
		log.Fatal(err)
	}

	//проверяем подклчение
	fmt.Println("Ping connecting...")
	ctx, _ = context.WithTimeout(context.Background(), 10*time.Second) //ждём ответа в течении 10 сек
	err = client.Ping(ctx, readpref.Primary())                         //отправляет сигнал ping, чтобы проверить, может ли клиент быть подключен к базе данных
	if err != nil {
		log.Fatal(err)
	}

	collectionTodos := client.Database(db).Collection(todoList)
	collectionCategory := client.Database(db).Collection(category)

	return client, collectionTodos, collectionCategory
}