com.netflix.genie.web.data.services.jpa

Class JpaAgentConnectionPersistenceServiceImpl

java.lang.Object

com.netflix.genie.web.data.services.jpa.JpaAgentConnectionPersistenceServiceImpl

All Implemented Interfaces:

AgentConnectionPersistenceService

@Transactional
public class JpaAgentConnectionPersistenceServiceImpl
extends java.lang.Object
implements AgentConnectionPersistenceService

JPA implementation of AgentConnectionPersistenceService.

Since:

4.0.0

Constructor Summary

Constructors

Constructor and Description
JpaAgentConnectionPersistenceServiceImpl(JpaAgentConnectionRepository agentConnectionRepository) Constructor.

Method Summary

Modifier and Type	Method and Description
long	getNumAgentConnectionsOnServer(@NotBlank java.lang.String hostname) Get the number of active agent connections on Genie server with the given hostname.
java.util.Optional<java.lang.String>	lookupAgentConnectionServer(@NotBlank java.lang.String jobId) Lookup the hostname/address of the server with an active connection to a given agent.
void	removeAgentConnection(@NotBlank java.lang.String jobId, @NotBlank java.lang.String hostname) Remove an existing connection currently active from the given agent to the local node.
int	removeAllAgentConnectionToServer(@NotBlank java.lang.String hostname) Drop all records of agent connections to the specified hostname.
void	saveAgentConnection(@NotBlank java.lang.String jobId, @NotBlank java.lang.String hostname) Store a new connection currently active from the given agent to the given node.

Methods inherited from class java.lang.Object

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

Constructor Detail

JpaAgentConnectionPersistenceServiceImpl

public JpaAgentConnectionPersistenceServiceImpl(JpaAgentConnectionRepository agentConnectionRepository)

Constructor.

Parameters:

agentConnectionRepository - agent connection repository

Method Detail

saveAgentConnection

public void saveAgentConnection(@NotBlank
                                @NotBlank java.lang.String jobId,
                                @NotBlank
                                @NotBlank java.lang.String hostname)

Store a new connection currently active from the given agent to the given node.

Specified by:

saveAgentConnection in interface AgentConnectionPersistenceService

Parameters:

jobId - the id of a job the agent is running

hostname - the server owning the connection

removeAgentConnection

public void removeAgentConnection(@NotBlank
                                  @NotBlank java.lang.String jobId,
                                  @NotBlank
                                  @NotBlank java.lang.String hostname)

Remove an existing connection currently active from the given agent to the local node. The entity won't be deleted if it was already modified by a different server.

Specified by:

removeAgentConnection in interface AgentConnectionPersistenceService

Parameters:

jobId - the id of a job the agent is running

hostname - the hostname expected to be associated to the connection

lookupAgentConnectionServer

@Transactional(readOnly=true)
public java.util.Optional<java.lang.String> lookupAgentConnectionServer(@NotBlank
                                                                                                      @NotBlank java.lang.String jobId)

Lookup the hostname/address of the server with an active connection to a given agent.

Specified by:

lookupAgentConnectionServer in interface AgentConnectionPersistenceService

Parameters:

jobId - the id of a job the agent is running

Returns:

the server hostname or empty

getNumAgentConnectionsOnServer

@Transactional(readOnly=true)
public long getNumAgentConnectionsOnServer(@NotBlank
                                                                         @NotBlank java.lang.String hostname)

Get the number of active agent connections on Genie server with the given hostname.

Specified by:

getNumAgentConnectionsOnServer in interface AgentConnectionPersistenceService

Parameters:

hostname - The server hostname to get the count for

Returns:

The number of agents currently connected to the given server

removeAllAgentConnectionToServer

public int removeAllAgentConnectionToServer(@NotBlank
                                            @NotBlank java.lang.String hostname)

Description copied from interface: AgentConnectionPersistenceService

Drop all records of agent connections to the specified hostname. This can be used to conveniently clean data related to a host when failure of the latter is detected.

Specified by:

removeAllAgentConnectionToServer in interface AgentConnectionPersistenceService

Parameters:

hostname - the server hostname

Returns:

the number of connections purged from persistent storage