Package com.snowflake.connectors.taskreactor.worker.registry

Class WorkerRegistryService

java.lang.Object

com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistryService

public class WorkerRegistryService
extends Object

Registry of task reactor workers.

Constructor Summary

Constructors

Constructor	Description
WorkerRegistryService(WorkerStatusRepository statusRepository, WorkerRegistry workerRegistry)	Creates a new WorkerRegistryService.

Method Summary

Modifier and Type	Method	Description
void	addNewWorkers(int workersToAdd)	Adds the specified numbers of workers to the instance.
boolean	canDeleteWorker(WorkerId workerId)	Returns whether a worker with the specified id can be deleted.
boolean	canProvisionWorker(WorkerId workerId)	Returns whether a worker with the specified id can be requested.
void	deleteWorkers(int workersToDelete)	Removes specified numbers of workers to the instance.
static WorkerRegistryService	from(Session session, Identifier instanceSchema)	Creates a new WorkerRegistryService.
int	getActiveOrRequestedWorkersCount()	Returns the number of active and requested workers.
List<WorkerId>	getActiveOrUpForDeletionWorkers()	Returns ids of active or up for deletion workers.
List<WorkerId>	getActiveProvisioningOrUpForDeletionWorkers()	Returns ids of active, provisioning or up for deletion workers.
List<WorkerId>	getActiveWorkers()	Returns ids of active workers.
List<WorkerId>	getAllWorkers()	Returns ids of all workers in one of the following states: WorkerLifecycleStatus.ACTIVE WorkerLifecycleStatus.PROVISIONING WorkerLifecycleStatus.UP_FOR_DELETION WorkerLifecycleStatus.DELETING
List<WorkerId>	getWorkersToDelete()	Returns ids of workers waiting for deletion.
List<WorkerId>	getWorkersToProvision()	Returns ids of requested workers.
void	workerDeleted(WorkerId workerId)	Sets the status of the specified worker to DELETED.
void	workerProvisioned(WorkerId workerId)	Sets the status of the specified worker to ACTIVE.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

WorkerRegistryService

public WorkerRegistryService(WorkerStatusRepository statusRepository,
                             WorkerRegistry workerRegistry)

Creates a new WorkerRegistryService.

Parameters:

statusRepository - Snowpark session object

workerRegistry - task reactor instance schema name

Method Detail

from

public static WorkerRegistryService from(Session session,
                                         Identifier instanceSchema)

Creates a new WorkerRegistryService.

Parameters:

session - Snowpark session object

instanceSchema - task reactor instance schema name

Returns:

a new service instance

getAllWorkers

public List<WorkerId> getAllWorkers()

Returns ids of all workers in one of the following states:

• WorkerLifecycleStatus.ACTIVE
• WorkerLifecycleStatus.PROVISIONING
• WorkerLifecycleStatus.UP_FOR_DELETION
• WorkerLifecycleStatus.DELETING

Returns:

ids of all workers in the above specified states

getActiveWorkers

public List<WorkerId> getActiveWorkers()

Returns ids of active workers.

Returns:

list of ids of active workers

getActiveOrRequestedWorkersCount

public int getActiveOrRequestedWorkersCount()

Returns the number of active and requested workers.

Returns:

number of active and requested workers

getWorkersToProvision

public List<WorkerId> getWorkersToProvision()

Returns ids of requested workers.

Returns:

list of ids of requested workers

getActiveProvisioningOrUpForDeletionWorkers

public List<WorkerId> getActiveProvisioningOrUpForDeletionWorkers()

Returns ids of active, provisioning or up for deletion workers.

Returns:

list of ids of active, provisioning or up for deletion workers

getActiveOrUpForDeletionWorkers

public List<WorkerId> getActiveOrUpForDeletionWorkers()

Returns ids of active or up for deletion workers.

Returns:

list of ids of active or up for deletion workers

canProvisionWorker

public boolean canProvisionWorker(WorkerId workerId)

Returns whether a worker with the specified id can be requested.

Parameters:

workerId - worker id

Returns:

whether a worker with the specified id can be requested

workerProvisioned

public void workerProvisioned(WorkerId workerId)

Sets the status of the specified worker to ACTIVE.

Parameters:

workerId - worker id

getWorkersToDelete

public List<WorkerId> getWorkersToDelete()

Returns ids of workers waiting for deletion.

Returns:

list of ids of workers waiting for deletion

canDeleteWorker

public boolean canDeleteWorker(WorkerId workerId)

Returns whether a worker with the specified id can be deleted.

Parameters:

workerId - worker id

Returns:

whether a worker with the specified id can be deleted

workerDeleted

public void workerDeleted(WorkerId workerId)

Sets the status of the specified worker to DELETED.

Parameters:

workerId - worker id

addNewWorkers

public void addNewWorkers(int workersToAdd)

Adds the specified numbers of workers to the instance.

Parameters:

workersToAdd - number of workers to add

deleteWorkers

public void deleteWorkers(int workersToDelete)

Removes specified numbers of workers to the instance.

Parameters:

workersToDelete - number of workers to delete