com.netflix.genie.web.services.loadbalancers.script

Class ScriptLoadBalancer

java.lang.Object

com.netflix.genie.web.services.loadbalancers.script.ScriptLoadBalancer

All Implemented Interfaces:

ClusterLoadBalancer, org.springframework.core.Ordered

@Component
 @ConditionalOnProperty(value="genie.jobs.clusters.loadBalancers.script.enabled",
                       havingValue="true")
public class ScriptLoadBalancer
extends java.lang.Object
implements ClusterLoadBalancer

An implementation of the ClusterLoadBalancer interface which uses user a supplied script to make decisions based on the list of clusters and the job request supplied.

The contract between the script and the Java code is that the script will be supplied global variables clusters and jobRequest which will be JSON strings representing the list (array) of clusters matching the cluster criteria tags and the job request that kicked off this evaluation. The code expects the script to either return the id of the cluster if one is selected or null if none was selected.

Since:

3.1.0

Field Summary

Fields inherited from interface com.netflix.genie.core.services.ClusterLoadBalancer

DEFAULT_ORDER

Fields inherited from interface org.springframework.core.Ordered

HIGHEST_PRECEDENCE, LOWEST_PRECEDENCE

Constructor Summary

Constructors

Constructor and Description
ScriptLoadBalancer(org.springframework.core.task.AsyncTaskExecutor asyncTaskExecutor, org.springframework.scheduling.TaskScheduler taskScheduler, GenieFileTransferService fileTransferService, org.springframework.core.env.Environment environment, com.fasterxml.jackson.databind.ObjectMapper mapper, com.netflix.spectator.api.Registry registry) Constructor.

Method Summary

Modifier and Type	Method and Description
int	getOrder()
void	refresh() Check if the script file needs to be refreshed.
Cluster	selectCluster(java.util.Set<Cluster> clusters, JobRequest jobRequest) Return best cluster to run job on.

Methods inherited from class java.lang.Object

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

Constructor Detail

ScriptLoadBalancer

@Autowired
public ScriptLoadBalancer(@Qualifier(value="genieAsyncTaskExecutor")
                                     org.springframework.core.task.AsyncTaskExecutor asyncTaskExecutor,
                                     @Qualifier(value="genieTaskScheduler")
                                     org.springframework.scheduling.TaskScheduler taskScheduler,
                                     @Qualifier(value="cacheGenieFileTransferService")
                                     GenieFileTransferService fileTransferService,
                                     org.springframework.core.env.Environment environment,
                                     com.fasterxml.jackson.databind.ObjectMapper mapper,
                                     com.netflix.spectator.api.Registry registry)

Constructor.

Parameters:

asyncTaskExecutor - The asynchronous task executor to use to run the load balancer script in

taskScheduler - The task scheduler to schedule the script refresh task with

fileTransferService - The file transfer service to use to download the script

environment - The program environment to get properties from

mapper - The object mapper to use to serialize objects to JSON for binding with scripts

registry - The metrics registry to use for collecting metrics

Method Detail

selectCluster

public Cluster selectCluster(@Nonnull @NonNull @NotEmpty
                             java.util.Set<Cluster> clusters,
                             @Nonnull @NonNull
                             JobRequest jobRequest)
                      throws GenieException

Return best cluster to run job on.

Specified by:

selectCluster in interface ClusterLoadBalancer

Parameters:

clusters - An immutable, non-empty list of available clusters to choose from

jobRequest - The job request these clusters are being load balanced for

Returns:

the "best" cluster to run job on or null if no cluster selected

Throws:

GenieException - if there is any error

getOrder

public int getOrder()

Specified by:

getOrder in interface ClusterLoadBalancer

Specified by:

getOrder in interface org.springframework.core.Ordered

refresh

public void refresh()

Check if the script file needs to be refreshed.