driver.go

package driver

import (
	"context"

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

type Driver struct {
	client *mongo.Client
	dbName string
}

type Option func(d *Driver)

func WithClient(client *mongo.Client) Option {
	return func(d *Driver) {
		d.client = client
	}
}

func WithDbName(dbName string) Option {
	return func(d *Driver) {
		d.dbName = dbName
	}
}

func NewWithOptions(options ...Option) (*Driver, error) {
	d := &Driver{}

	for _, opt := range options {
		opt(d)
	}

	if d.client == nil {
		return nil, ErrEmptyClient
	}

	return d, nil
}

func New(ctx context.Context) (*Driver, error) {
	return NewWithConfig(ctx, DefaultConfig())
}

func NewWithConfig(ctx context.Context, config Config) (*Driver, error) {
	uri, err := buildConnectionURI(&config)
	if err != nil {
		return nil, err
	}

	client, errConn := buildConnection(
		ctx,
		uri,
		config.CertPath,
		config.MinPoolSize,
		config.MaxPoolSize,
	)

	if errConn != nil {
		return nil, errConn
	}

	return &Driver{
		client: client,
		dbName: config.DBName,
	}, nil
}

func (d *Driver) Client() *mongo.Client {
	return d.client
}

func (d *Driver) DbName() string {
	return d.dbName
}

func (d *Driver) Close() error {
	return d.client.Disconnect(context.Background())
}