Class InMemoryWorkerRegistry

  • All Implemented Interfaces:
    com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry

    public class InMemoryWorkerRegistry
    extends Object
    implements com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
    In memory implementation of WorkerRegistry.
    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void clear()
      Clears this registry.
      int getWorkerCountWithStatuses​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus... statuses)  
      List<com.snowflake.connectors.taskreactor.worker.WorkerId> getWorkerIds​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus... statuses)  
      void insertWorkers​(int workersToInsert)  
      void setWorkersStatus​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus status, List<com.snowflake.connectors.taskreactor.worker.WorkerId> workerIds)  
      Map<com.snowflake.connectors.taskreactor.worker.WorkerId,​com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus> store()
      Returns a map backing this registry.
      long updateWorkersStatus​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus newStatus, com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus currentStatus, Collection<com.snowflake.connectors.taskreactor.worker.WorkerId> workerIds)  
    • Constructor Detail

      • InMemoryWorkerRegistry

        public InMemoryWorkerRegistry()
    • Method Detail

      • insertWorkers

        public void insertWorkers​(int workersToInsert)
        Specified by:
        insertWorkers in interface com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
      • setWorkersStatus

        public void setWorkersStatus​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus status,
                                     List<com.snowflake.connectors.taskreactor.worker.WorkerId> workerIds)
        Specified by:
        setWorkersStatus in interface com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
      • updateWorkersStatus

        public long updateWorkersStatus​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus newStatus,
                                        com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus currentStatus,
                                        Collection<com.snowflake.connectors.taskreactor.worker.WorkerId> workerIds)
        Specified by:
        updateWorkersStatus in interface com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
      • getWorkerIds

        public List<com.snowflake.connectors.taskreactor.worker.WorkerId> getWorkerIds​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus... statuses)
        Specified by:
        getWorkerIds in interface com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
      • getWorkerCountWithStatuses

        public int getWorkerCountWithStatuses​(com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus... statuses)
        Specified by:
        getWorkerCountWithStatuses in interface com.snowflake.connectors.taskreactor.worker.registry.WorkerRegistry
      • store

        public Map<com.snowflake.connectors.taskreactor.worker.WorkerId,​com.snowflake.connectors.taskreactor.worker.registry.WorkerLifecycleStatus> store()
        Returns a map backing this registry.
        Returns:
        map backing this registry
      • clear

        public void clear()
        Clears this registry.