com.spotify.helios.agent

Class ZooKeeperAgentModel

java.lang.Object

com.google.common.util.concurrent.AbstractIdleService

com.spotify.helios.agent.ZooKeeperAgentModel

All Implemented Interfaces:

com.google.common.util.concurrent.Service, AgentModel

public class ZooKeeperAgentModel
extends com.google.common.util.concurrent.AbstractIdleService
implements AgentModel

The Helios Agent's view into ZooKeeper.

This caches ZK state to local disk so the agent can continue to function in the face of a ZK outage.

Nested Class Summary

Nested classes/interfaces inherited from interface com.spotify.helios.agent.AgentModel

AgentModel.Listener

Nested classes/interfaces inherited from interface com.google.common.util.concurrent.Service

com.google.common.util.concurrent.Service.Listener, com.google.common.util.concurrent.Service.State

Constructor Summary

Constructors

Constructor and Description
ZooKeeperAgentModel(ZooKeeperClientProvider provider, String host, Path stateDirectory, TaskHistoryWriter historyWriter, List<EventSender> eventSenders, String taskStatusEventTopic)

Method Summary

Modifier and Type	Method and Description
void	addListener(AgentModel.Listener listener) Add a listener that will be notified when tasks are changed.
protected void	fireTasksUpdated()
Map<com.spotify.helios.common.descriptors.JobId,com.spotify.helios.common.descriptors.Task>	getTasks() Returns the tasks (basically, a pair of JobId and Task) for the current agent.
com.spotify.helios.common.descriptors.TaskStatus	getTaskStatus(com.spotify.helios.common.descriptors.JobId jobId) Get the TaskStatus for the job identified by jobId.
Map<com.spotify.helios.common.descriptors.JobId,com.spotify.helios.common.descriptors.TaskStatus>	getTaskStatuses() Returns the TaskStatuses for all tasks assigned to the current agent.
void	removeListener(AgentModel.Listener listener) Remove a listener that will be notified when tasks are changed.
void	removeTaskStatus(com.spotify.helios.common.descriptors.JobId jobId) Remove the TaskStatus for the job identified by jobId.
void	setTaskStatus(com.spotify.helios.common.descriptors.JobId jobId, com.spotify.helios.common.descriptors.TaskStatus status) Set the TaskStatus for the job identified by jobId.
protected void	shutDown()
protected void	startUp()

Methods inherited from class com.google.common.util.concurrent.AbstractIdleService

addListener, awaitRunning, awaitRunning, awaitTerminated, awaitTerminated, executor, failureCause, isRunning, serviceName, startAsync, state, stopAsync, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

ZooKeeperAgentModel

public ZooKeeperAgentModel(ZooKeeperClientProvider provider,
                           String host,
                           Path stateDirectory,
                           TaskHistoryWriter historyWriter,
                           List<EventSender> eventSenders,
                           String taskStatusEventTopic)
                    throws IOException,
                           InterruptedException

Throws:

IOException

InterruptedException

Method Detail

startUp

protected void startUp()
                throws Exception

Specified by:

startUp in class com.google.common.util.concurrent.AbstractIdleService

Throws:

Exception

shutDown

protected void shutDown()
                 throws Exception

Specified by:

shutDown in class com.google.common.util.concurrent.AbstractIdleService

Throws:

Exception

getTasks

public Map<com.spotify.helios.common.descriptors.JobId,com.spotify.helios.common.descriptors.Task> getTasks()

Returns the tasks (basically, a pair of JobId and Task) for the current agent.

Specified by:

getTasks in interface AgentModel

Returns:

A map of job id's to tasks.

getTaskStatuses

public Map<com.spotify.helios.common.descriptors.JobId,com.spotify.helios.common.descriptors.TaskStatus> getTaskStatuses()

Returns the TaskStatuses for all tasks assigned to the current agent.

Specified by:

getTaskStatuses in interface AgentModel

Returns:

A map of job ids to task statuses.

setTaskStatus

public void setTaskStatus(com.spotify.helios.common.descriptors.JobId jobId,
                          com.spotify.helios.common.descriptors.TaskStatus status)
                   throws InterruptedException

Set the TaskStatus for the job identified by jobId.

Specified by:

setTaskStatus in interface AgentModel

Parameters:

jobId - The job ID.

status - The job state.

Throws:

InterruptedException - If the thread is interrupted.

getTaskStatus

public com.spotify.helios.common.descriptors.TaskStatus getTaskStatus(com.spotify.helios.common.descriptors.JobId jobId)

Get the TaskStatus for the job identified by jobId.

Specified by:

getTaskStatus in interface AgentModel

Parameters:

jobId - The job ID.

Returns:

The job state.

removeTaskStatus

public void removeTaskStatus(com.spotify.helios.common.descriptors.JobId jobId)
                      throws InterruptedException

Remove the TaskStatus for the job identified by jobId.

Specified by:

removeTaskStatus in interface AgentModel

Parameters:

jobId - The job id.

Throws:

InterruptedException - If the thread is interrupted.

addListener

public void addListener(AgentModel.Listener listener)

Add a listener that will be notified when tasks are changed.

Specified by:

addListener in interface AgentModel

Parameters:

listener - A listener that will be called when the set of tasks changes.

See Also:

AgentModel.Listener

removeListener

public void removeListener(AgentModel.Listener listener)

Remove a listener that will be notified when tasks are changed.

Specified by:

removeListener in interface AgentModel

Parameters:

listener - The listener to remove.

See Also:

AgentModel.addListener(Listener)

fireTasksUpdated

protected void fireTasksUpdated()