ydb::DistLockComponentBase Class Reference

#include <userver/ydb/dist_lock/component_base.hpp>

Detailed Description

Base class for YDB-based distlock worker components.

A component that implements a distlock with lock in YDB. Inherit from DistLockComponentBase and implement DoWork(). Lock options are configured in static config.

The class must be used for infinite loop jobs.

Static configuration example:

        # yaml
        sample-dist-lock:
            # Settings that might be used for a group of related distlock instances
            database-settings:
                # The key of the database within ydb component (NOT the actual database path)
                dbname: sampledb
                # Name of the coordination node within the database
                coordination-node: sample-node
            # Name of the semaphore within the coordination node
            semaphore-name: sample-lock

Static options of ydb::DistLockComponentBase :

Options inherited from components::ComponentBase :

See also

Periodics and DistLocks

Definition at line 41 of file component_base.hpp.

Inheritance diagram for ydb::DistLockComponentBase:

Public Member Functions
	DistLockComponentBase (const components::ComponentConfig &, const components::ComponentContext &)
bool	OwnsLock () const noexcept
	Checks if the the current service instance owns the lock.
ComponentHealth	GetComponentHealth () const override
void	OnLoadingCancelled () override
void	OnAllComponentsLoaded () override
void	OnAllComponentsAreStopping () override

Static Public Member Functions
static yaml_config::Schema	GetStaticConfigSchema ()

Protected Types
using	LoggableComponentBase = ComponentBase

Protected Member Functions
virtual void	DoWork ()=0
virtual void	DoWorkTestsuite ()
void	Start ()
	Must be called at the end of the constructor of the derived component.
void	Stop () noexcept
	Must be called in the destructor of the derived component.

Member Typedef Documentation

LoggableComponentBase

using components::ComponentBase::LoggableComponentBase = ComponentBase

protectedinherited

Deprecated:

use components::ComponentBase instead.

Definition at line 67 of file component_base.hpp.

Member Function Documentation

DoWork()

virtual void ydb::DistLockComponentBase::DoWork ( )

protectedpure virtual

Override this function with anything that must be done under the lock.

Note

DoWork must honour task cancellation and stop ASAP when it is cancelled, otherwise brain split is possible (IOW, two different users do work assuming both of them hold the lock, which is not true).

When DoWork exits for any reason, the lock is dropped, then after restart-delay the lock is attempted to be re-acquired (but by that time another host likely acquires the lock).

Example implementation

void SampleDistLock::DoWork() {
    while (!engine::current_task::ShouldCancel()) {
        // Start a new trace_id.
        auto span = tracing::Span::MakeRootSpan("sample-dist-lock");
 
        // If Foo() or anything in DoWork() throws an exception,
        // DoWork() will be restarted in `restart-delay` seconds.
        Foo();
 
        // Check for cancellation after cpu-intensive Foo().
        // You must check for cancellation at least every
        // `cancel-task-time-limit`
        // seconds to have time to notice lock prolongation failure.
        if (engine::current_task::ShouldCancel()) {
            break;
        }
 
        Bar();
 
        engine::InterruptibleSleepFor(config_source_.GetCopy(kSampleDistLockIntervalMs));
    }
}

DoWorkTestsuite()

virtual void ydb::DistLockComponentBase::DoWorkTestsuite ( )

inlineprotectedvirtual

Override this function to provide a custom testsuite handler, e.g. one that does not contain a "while not cancelled" loop. Calls DoWork by default.

In testsuite runs, the normal DoWork execution disabled by default. There is an API to call DoWorkTestsuite from testsuite, imitating waiting until DoWork runs:

    await service_client.run_distlock_task('sample-dist-lock')

Definition at line 84 of file component_base.hpp.

GetComponentHealth()

ComponentHealth components::ComponentBase::GetComponentHealth ( ) const

inlineoverridevirtualinherited

Override this function to inform the world of the state of your component.

Warning

The function is called concurrently from multiple threads.

Reimplemented from components::RawComponentBase.

Reimplemented in server::handlers::Restart.

Definition at line 35 of file component_base.hpp.

OnAllComponentsAreStopping()

void components::ComponentBase::OnAllComponentsAreStopping ( )

inlineoverridevirtualinherited

Component may use this function to stop doing work before the stop of the components that depend on it.

Base components may override it and make final to do some work before the derived object constructor is called. Don't use it otherwise.

Reimplemented from components::RawComponentBase.

Reimplemented in urabbitmq::ConsumerComponentBase, and components::Server.

Definition at line 58 of file component_base.hpp.

OnAllComponentsLoaded()

void components::ComponentBase::OnAllComponentsLoaded ( )

inlineoverridevirtualinherited

Component may use this function to finalize registration of other components that depend on it (for example, handler components register in server component, and the latter uses OnAllComponentsLoaded() to start processing requests).

Base components may override it and make final to do some work after the derived object constructor is called. Don't use it otherwise.

Reimplemented from components::RawComponentBase.

Reimplemented in urabbitmq::ConsumerComponentBase, components::Server, and server::handlers::Ping.

Definition at line 51 of file component_base.hpp.

OnLoadingCancelled()

void components::ComponentBase::OnLoadingCancelled ( )

inlineoverridevirtualinherited

Called once if the creation of any other component failed. If the current component expects some other component to take any action with the current component, this call is a signal that such action may never happen due to components loading was cancelled. Application components might not want to override it.

Reimplemented from components::RawComponentBase.

Definition at line 42 of file component_base.hpp.

OwnsLock()

bool ydb::DistLockComponentBase::OwnsLock ( ) const

noexcept

Checks if the the current service instance owns the lock.

Useful for:

1. Writing metrics only on the primary (for the given distlock) host.
2. Running on-demand work, e.g. in handlers, only on the primary host. The work must be short-lived (single requests with small timeout), otherwise brain split (where multiple hosts consider themselves primary) is possible.

---

The documentation for this class was generated from the following file:

• userver/ydb/dist_lock/component_base.hpp