1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
|
/* Copyright (c) 2009 Peter Troshin
*
* JAva Bioinformatics Analysis Web Services (JABAWS) @version: 1.0
*
* This library is free software; you can redistribute it and/or modify it under the terms of the
* Apache License version 2 as published by the Apache Software Foundation
*
* This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
* even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Apache
* License for more details.
*
* A copy of the license is in apache_license.txt. It is also available here:
* @see: http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Any republication or derived work distributed in source code form
* must include this copyright and license notice.
*/
package compbio.engine;
import java.util.List;
import org.apache.log4j.Logger;
import compbio.data.sequence.FastaSequence;
import compbio.engine.client.Executable;
import compbio.engine.local.LocalExecutorService;
import compbio.metadata.Limit;
import compbio.metadata.PresetManager;
/**
* This class decides where to execute the job. If the local engine is enabled
* in the configuration file and it has free threads and the size of the tasks
* permits the local execution, then the local execution will be favoured.
*
* @author pvtroshin
* @version 1.0 March 2009
*/
public class LoadBalancer {
private static Logger log = Logger.getLogger(LoadBalancer.class);
private LoadBalancer() {
}
public static Executable.ExecProvider getEngine(Executable<?> executable) {
if (LocalExecutorService.getExecutor().canAcceptMoreWork()) {
log.debug("LOCAL engine HAS FREE threads will execute ... ");
return Executable.ExecProvider.Local;
}
log.debug("NO free threads on the LOCAL engine! Targeting for CLUSTER execution... ");
return Executable.ExecProvider.Cluster;
}
public static <T, V> Executable.ExecProvider getEngine(
Executable<V> executable, List<FastaSequence> dataSet) {
// If data set is deemed too big for local execution, than give a
// cluster engine
// If limit is not defined then defaults to executing on the cluster
Limit<V> limit = executable.getLimit(PresetManager.LOCAL_ENGINE_LIMIT_PRESET);
log.trace("Inspecting whether the job can be executed locally using limit: " + limit);
if (limit == null || limit.isExceeded(dataSet)) {
log.debug("Job EXCEEDS LOCAL execution LIMIT targeting for cluster execution! ");
return Executable.ExecProvider.Cluster;
}
log.debug("Job FITS into the LOCAL execution limit consulting load balancer... ");
// Even if the data satisfies criteria for local execution it may still
// be executed on the cluster as the maximum capacity for local engine
// may be reached.
return getEngine(executable);
}
}
|