dataaccess.go

package mongodata

import (
	"context"

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

type DataAccess struct {
	client   *mongo.Client
	database *mongo.Database
}

func Connect(
	ctx context.Context,
	uri string,
	databaseName string,
) (*DataAccess, error) {
	dataAccess := new(DataAccess)
	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
	if err != nil {
		return nil, err
	}

	if err := client.Ping(ctx, readpref.Primary()); err != nil {
		return nil, err
	}

	dataAccess.client = client
	dataAccess.database = client.Database(databaseName)
	return dataAccess, nil
}

func (da *DataAccess) Collection(collectionName string) *mongo.Collection {
	return da.database.Collection(collectionName)
}

func (da *DataAccess) Close(ctx context.Context) error {
	err := da.client.Disconnect(ctx)
	return err
}