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

Class ScriptLoadBalancer

java.lang.Object

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

All Implemented Interfaces:

com.netflix.genie.core.services.ClusterLoadBalancer, org.springframework.core.Ordered

@Component
 @ConditionalOnProperty(value="genie.jobs.clusters.loadBalancers.script.enabled",
                       havingValue="true")
public class ScriptLoadBalancer
extends java.lang.Object
implements com.netflix.genie.core.services.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, com.netflix.genie.core.services.impl.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.
com.netflix.genie.common.dto.Cluster	selectCluster(java.util.Set<com.netflix.genie.common.dto.Cluster> clusters, com.netflix.genie.common.dto.JobRequest jobRequest)

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")
                                     com.netflix.genie.core.services.impl.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 com.netflix.genie.common.dto.Cluster selectCluster(@Nonnull @NonNull @NotEmpty
                                                          java.util.Set<com.netflix.genie.common.dto.Cluster> clusters,
                                                          @Nonnull @NonNull
                                                          com.netflix.genie.common.dto.JobRequest jobRequest)
                                                   throws com.netflix.genie.common.exceptions.GenieException

Specified by:

selectCluster in interface com.netflix.genie.core.services.ClusterLoadBalancer

Throws:

com.netflix.genie.common.exceptions.GenieException

getOrder

public int getOrder()

Specified by:

getOrder in interface com.netflix.genie.core.services.ClusterLoadBalancer

Specified by:

getOrder in interface org.springframework.core.Ordered

refresh

public void refresh()

Check if the script file needs to be refreshed.