global_uri.go

package mongodb_locks

import (
	"context"
	mongodb_storage "github.com/storage-lock/go-mongodb-storage"
	storage_lock "github.com/storage-lock/go-storage-lock"
	storage_lock_factory "github.com/storage-lock/go-storage-lock-factory"
	"go.mongodb.org/mongo-driver/mongo"
)

var uriStorageLockFactoryBeanFactory *storage_lock_factory.StorageLockFactoryBeanFactory[string, *mongo.Client] = storage_lock_factory.NewStorageLockFactoryBeanFactory[string, *mongo.Client]()

func NewMongoLockByUri(ctx context.Context, uri string, lockId string) (*storage_lock.StorageLock, error) {
	factory, err := GetMongoLockFactoryByUri(ctx, uri)
	if err != nil {
		return nil, err
	}
	return factory.CreateLock(lockId)
}

func NewMongoLockByUriWithOptions(ctx context.Context, uri string, options *storage_lock.StorageLockOptions) (*storage_lock.StorageLock, error) {
	factory, err := GetMongoLockFactoryByUri(ctx, uri)
	if err != nil {
		return nil, err
	}
	return factory.CreateLockWithOptions(options)
}

func GetMongoLockFactoryByUri(ctx context.Context, uri string) (*storage_lock_factory.StorageLockFactory[*mongo.Client], error) {
	return uriStorageLockFactoryBeanFactory.GetOrInit(ctx, uri, func(ctx context.Context) (*storage_lock_factory.StorageLockFactory[*mongo.Client], error) {
		options := mongodb_storage.NewMongoStorageOptionsWithURI(uri)
		storage, err := mongodb_storage.NewMongoStorage(ctx, options)
		if err != nil {
			return nil, err
		}
		factory := storage_lock_factory.NewStorageLockFactory(storage, options.ConnectionManager)
		return factory, nil
	})
}