upda/server/service_lock_mem.go

package server

import (
	"context"
	"errors"
	"git.myservermanager.com/varakh/upda/util"
	"go.uber.org/zap"
)

type lockMemService struct {
	registry *util.InMemoryLockRegistry
}

var (
	errLockMemNotReleased = newServiceError(Conflict, errors.New("lock service: could not release lock"))
)

func newLockMemService() lockService {
	zap.L().Info("Initialized in-memory locking service")
	return &lockMemService{registry: util.NewInMemoryLockRegistry()}
}

func (s *lockMemService) lock(ctx context.Context, resource string) (appLock, error) {
	if resource == "" {
		return nil, errorValidationNotBlank
	}

	zap.L().Sugar().Debugf("Trying to lock '%s'", resource)

	s.registry.Lock(resource)

	zap.L().Sugar().Debugf("Locked '%s'", resource)

	l := &inMemoryLock{
		registry: s.registry,
		resource: resource,
	}

	return l, nil
}

func (s *lockMemService) lockWithOptions(ctx context.Context, resource string, options ...appLockOption) (appLock, error) {
	// ignores any options
	return s.lock(ctx, resource)
}

var _ appLock = (*inMemoryLock)(nil)

type inMemoryLock struct {
	registry *util.InMemoryLockRegistry
	resource string
}

func (r inMemoryLock) unlock(ctx context.Context) error {
	zap.L().Sugar().Debugf("Unlocking '%s'", r.resource)

	if err := r.registry.Unlock(r.resource); err != nil {
		return errLockMemNotReleased
	}
	return nil
}