private void setSvnCredential(CIJob job) throws JDOMException, IOException {
S_LOGGER.debug("Entering Method CIManagerImpl.setSvnCredential");
try {
String jenkinsTemplateDir = Utility.getJenkinsTemplateDir();
String credentialFilePath = jenkinsTemplateDir + job.getRepoType() + HYPHEN + CREDENTIAL_XML;
if (debugEnabled) {
S_LOGGER.debug("credentialFilePath ... " + credentialFilePath);
}
File credentialFile = new File(credentialFilePath);
SvnProcessor processor = new SvnProcessor(credentialFile);
// DataInputStream in = new DataInputStream(new FileInputStream(credentialFile));
// while (in.available() != 0) {
// System.out.println(in.readLine());
// }
// in.close();
processor.changeNodeValue("credentials/entry//userName", job.getUserName());
processor.changeNodeValue("credentials/entry//password", job.getPassword());
processor.writeStream(new File(Utility.getJenkinsHome() + File.separator + job.getName()));
// jenkins home location
String jenkinsJobHome = System.getenv(JENKINS_HOME);
StringBuilder builder = new StringBuilder(jenkinsJobHome);
builder.append(File.separator);
processor.writeStream(new File(builder.toString() + CI_CREDENTIAL_XML));
} catch (Exception e) {
S_LOGGER.error(
"Entered into the catch block of CIManagerImpl.setSvnCredential "
+ e.getLocalizedMessage());
}
}
private static void usage() {
System.err.println("usage: java net.tinyos.drain.Drip <opts>");
System.err.println(" -d <data value>");
System.err.println(" -c <channel id>");
System.err.println(" -w : send message with wakeup bit set");
System.err.println(" -h, --help : this information");
System.exit(1);
}
public void test_init_absolutePath() throws ServletException {
String os = System.getProperty("os.name");
if (os.startsWith("Windows")) {
config.parameterValue = "C:/Temp";
} else {
config.parameterValue = "/abc/def";
}
servlet.init(config);
assertEquals("prefix value", config.parameterValue, servlet.prefix);
}
private void send(Message m) {
try {
moteIF.send(MoteIF.TOS_BCAST_ADDR, m);
} catch (IOException e) {
e.printStackTrace();
System.out.println("ERROR: Can't send message");
System.exit(1);
} catch (Exception e) {
e.printStackTrace();
}
}
public static void main(String[] args) throws CleverException {
in = Initiator.getInstance(); // creo un oggetto Initiator
in.start(); // faccio partire l'initiator
while (true) {
try {
Thread.sleep(1000000);
} catch (InterruptedException ex) {
System.exit(1);
}
}
}
public Drip(int id) {
log.info("Started id=" + id);
try {
moteIF = new MoteIF();
moteIF.registerListener(new DripMsg(), this);
} catch (Exception e) {
System.out.println("ERROR: Couldn't contact serial forwarder.");
System.exit(1);
}
this.id = id;
}
/**
* Starts Grid instance. Note that if grid is already started, then it will be looked up and
* returned from this method.
*
* @return Started grid.
*/
private Grid startGrid() {
Properties props = System.getProperties();
gridName = props.getProperty(GRIDGAIN_NAME.name());
if (!props.containsKey(GRIDGAIN_NAME.name()) || G.state(gridName) != GridFactoryState.STARTED) {
selfStarted = true;
// Set class loader for the spring.
ClassLoader curCl = Thread.currentThread().getContextClassLoader();
// Add no-op logger to remove no-appender warning.
Appender app = new NullAppender();
Logger.getRootLogger().addAppender(app);
try {
Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
Grid grid = G.start(cfgPath);
gridName = grid.name();
System.setProperty(GRIDGAIN_NAME.name(), grid.name());
return grid;
} catch (GridException e) {
throw new GridRuntimeException("Failed to start grid: " + cfgPath, e);
} finally {
Logger.getRootLogger().removeAppender(app);
Thread.currentThread().setContextClassLoader(curCl);
}
}
return G.grid(gridName);
}
public static void main(String[] args) {
parseArgs(args);
Drip drip = new Drip(channel);
TestDripMsg msg = new TestDripMsg();
msg.set_data((short) data);
if (wakeup) {
drip.sendWakeup(msg, TestDripMsg.DEFAULT_MESSAGE_SIZE);
} else {
drip.send(msg, TestDripMsg.DEFAULT_MESSAGE_SIZE);
}
System.exit(0);
}
private static void processFile(
String pid, File child, File outFilePath, String guidelineId, String htmlFilePath) {
// Write to output directory
// outFilePath should be path+pid+extension
String caseData = null;
SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
java.util.Date startTime = new java.util.Date();
try {
caseData = readFileAsString(child.getPath());
} catch (java.io.IOException e) {
logger.error("Error reading data file ", e);
System.exit(-1);
}
String recommendations = null;
try {
System.out.println("-----------------------------------------");
recommendations =
pca.topLevelComputeAdvisory(pid, caseData, formatter.format(startTime), guidelineId, pid);
System.out.println("-----------------------------------------");
// String fileName = outFilePath+pid+fileExtension
// File outFilePath = new File(fileName);;
logger.warn("Output recommenations to: " + outFilePath);
if (outFilePath.exists()) {
outFilePath.delete();
}
try {
PrintWriter out = new PrintWriter(outFilePath.getPath());
out.print(recommendations);
out.flush();
// System.out.println("****READY FILE PROCESSING****");
// create Ready file
// String readyFileName;
// readyFileName = outFilePath.getPath();
// System.out.println("Ready File Name Is: " + readyFileName); //Remove
// File readyFile = new File(readyFileName.replace(".xml","_READY.txt"));
// try {
// readyFile.createNewFile();
// System.out.println("****SUCCESS****"+readyFile.getPath());
// } catch (java.io.IOException e0) {
// System.out.println("Error creating READY file");
// }
} catch (FileNotFoundException e) {
e.printStackTrace();
}
} catch (PCA_Session_Exception e1) {
e1.printStackTrace();
}
}
private void setMailCredential(CIJob job) {
if (debugEnabled) {
S_LOGGER.debug("Entering Method CIManagerImpl.setMailCredential");
}
try {
String jenkinsTemplateDir = Utility.getJenkinsTemplateDir();
String mailFilePath = jenkinsTemplateDir + MAIL + HYPHEN + CREDENTIAL_XML;
if (debugEnabled) {
S_LOGGER.debug("configFilePath ... " + mailFilePath);
}
File mailFile = new File(mailFilePath);
SvnProcessor processor = new SvnProcessor(mailFile);
// DataInputStream in = new DataInputStream(new FileInputStream(mailFile));
// while (in.available() != 0) {
// System.out.println(in.readLine());
// }
// in.close();
// Mail have to go with jenkins running email address
InetAddress ownIP = InetAddress.getLocalHost();
processor.changeNodeValue(
CI_HUDSONURL,
HTTP_PROTOCOL
+ PROTOCOL_POSTFIX
+ ownIP.getHostAddress()
+ COLON
+ job.getJenkinsPort()
+ FORWARD_SLASH
+ CI
+ FORWARD_SLASH);
processor.changeNodeValue("smtpAuthUsername", job.getSenderEmailId());
processor.changeNodeValue("smtpAuthPassword", job.getSenderEmailPassword());
processor.changeNodeValue("adminAddress", job.getSenderEmailId());
// jenkins home location
String jenkinsJobHome = System.getenv(JENKINS_HOME);
StringBuilder builder = new StringBuilder(jenkinsJobHome);
builder.append(File.separator);
processor.writeStream(new File(builder.toString() + CI_MAILER_XML));
} catch (Exception e) {
S_LOGGER.error(
"Entered into the catch block of CIManagerImpl.setMailCredential "
+ e.getLocalizedMessage());
}
}
private void initSSL()
throws NoSuchAlgorithmException, CertificateException, FileNotFoundException, IOException,
KeyStoreException, UnrecoverableKeyException, KeyManagementException {
KeyStore keyStore = KeyStore.getInstance(KeyStore.getDefaultType());
dLog.trace("Loading keystore");
keyStore.load(new FileInputStream(KEYSTORE_PATH), KEYSTORE_PASSWORD.toCharArray());
dLog.trace("Initializing KeyManagerFactory");
KeyManagerFactory kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
kmf.init(keyStore, KEYSTORE_PASSWORD.toCharArray());
dLog.trace("Initializing SSL Context");
System.setProperty("https.protocols", "TLSv1");
SSLContext ctx = SSLContext.getInstance("TLS");
// Note: only care about encryption at this time (Confidentiality), will add trust later
// (Integrity)
ctx.init(kmf.getKeyManagers(), trustAllCerts, new java.security.SecureRandom());
dLog.trace("Retreiving SSL socket factory");
factory = ctx.getSocketFactory();
}
private static int mergeSameAltitude(
final CellData[] cellData, final int start, final int first, int count) {
for (int i = start; i < first; ++i) {
int j = first;
while (j < first + count) {
if (!CellData.collide(cellData[i], cellData[j])) {
++j;
} else {
final CellData old = cellData[i];
cellData[i] = cellData[i].createMerged(cellData[j]);
if (--count == 0) {
continue;
}
System.arraycopy(cellData, j + 1, cellData, j, count);
cellData[j + count] = old;
}
}
}
return count;
}
private void loadKB(String kbURL, String guidelineId) {
System.out.println("Starting to load KB");
System.out.println("------------------------------------------------------------------");
try {
PCAImp = new PCAServer_i();
PCAImp.kbManager = new KBHandler(kbURL); // Sever loads the KB
// java.util.Date finishedKB = new java.util.Date();
// logger.warn("finished loading KB "+(finishedKB.getTime() - startTime.getTime())+ "
// milliseconds after start.");
pca = PCAImp.open_pca_session();
if ((guidelineId != null) && (guidelineId != "")) {
pca.setGuideline(guidelineId); // Specifies the guideline to use
} else logger.error("No GUIDELINEID specified");
} catch (Exception se) {
logger.error("Exception raised during initialization " + se.toString());
System.exit(1);
}
pca.finishSession();
System.out.println("------------------------------------------------------------------");
System.out.println("Done loading KB");
}
private String[] loadKBInfo(String mode, String parentPath) {
// Results: [0] = kbFilePath
// [1] = kbID
String[] results = new String[2];
results[0] = "";
results[1] = "";
String configFilePath = parentPath + "config/";
File configFileDir = new File(configFilePath);
if (configFileDir.isDirectory() == false) {
System.out.println(configFileDir.getName() + " directory is missing. Exiting");
System.exit(0);
}
String configFName = parentPath + "config/KB MAP.txt";
TStringList sl = new TStringList();
try {
sl.ReadFromFile(configFName);
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
for (int i = 0; i < sl.Count(); i++) {
String s = sl.getString(i);
String lbl = s.split("=")[0];
if (lbl.equals("")) continue;
String[] lblParts = lbl.split(":");
if (lblParts.length < 2) continue;
if (lblParts[0].equals(mode) == false) continue;
if (lblParts[1].equals("KB")) {
String kbFilePath = parentPath + "kbs/" + s.split("=")[1].trim();
results[0] = kbFilePath;
} else if (lblParts[1].equals("ID")) {
results[1] = s.split("=")[1].trim(); // kb ID string
}
}
return results;
}
public WDEManagerDB(String name) {
super(name);
String eManagerDir = System.getProperty("em.home");
String log4jConfigFile = eManagerDir + "/config/log4j.properties";
PropertyConfigurator.configure(log4jConfigFile);
}
public static void main(String[] argv) throws InterruptedException {
// Arguments:
// 1st = Mode
// 2nd = Parent Path -- (Parent path must have children /data and /output
System.out.println("+----------------------------------+");
System.out.println("| Running Athena Pallas Processor |");
System.out.println("| 9/12/12 |");
System.out.println("+----------------------------------+");
Athena_Server_Obj = new ATHENA_Server_Local();
BasicConfigurator.configure();
Logger.getRootLogger().setLevel(Level.WARN);
// Get and verify parameters
if (argv.length < 2) {
System.out.println("Not enough parameters.");
System.out.println("Expected 2 parameters: 1st is MODE, 2nd is path to working directory.");
System.out.println("Exiting.");
System.exit(0);
}
String mode = argv[0];
String parentPath = argv[1];
File inFilePath = new File(parentPath + "astronautdata/");
File outFilePath = new File(parentPath + "output/");
File kbFilePath = new File(parentPath + "kbs/");
if (mode.equals("")) {
System.out.println("Parameter 1 is empty. Should be RUNMODE (e.g. 'HTN'). Exiting");
System.exit(0);
}
if (inFilePath.isDirectory() == false) {
System.out.println(inFilePath.getName() + " directory is missing. Exiting");
System.exit(0);
}
if (outFilePath.isDirectory() == false) {
System.out.println(outFilePath.getName() + " directory is missing. Exiting");
System.exit(0);
}
if (kbFilePath.isDirectory() == false) {
System.out.println(kbFilePath.getName() + " directory is missing. Exiting");
System.exit(0);
}
String kbID = "";
TStringList IgnoredFileNames = new TStringList();
// Main Loop
while (mainLoopRunning) {
if (Verbose == true) {
// System.out.println("New Loop");
}
fileProcessed = false;
if (Verbose == true) {
// System.out.println("Ignoring #" + IgnoredFileNames.Count() + " files.");
}
// (re)load if needed
if (loadKB == true) {
String[] kbInfo = Athena_Server_Obj.loadKBInfo(mode, parentPath);
String kbFilenamePath = kbInfo[0];
kbID = kbInfo[1];
Athena_Server_Obj.loadKB(kbFilenamePath, kbID);
loadKB = false;
}
File[] listOfFiles = inFilePath.listFiles();
// for (File child : inFilePath.listFiles()) {
for (int i = 0; i < listOfFiles.length; i++) {
File child = listOfFiles[i];
if (".".equals(child.getName()) || "..".equals(child.getName())) {
continue;
}
if (child.getName().toUpperCase().equals(mode + "-CMD.TXT")) {
try {
handleCommands(child.getPath(), outFilePath, mode);
// fileProcessed = true;
if (child.delete() == false) {
System.out.println("Unable to delete: " + child.getPath());
Thread.sleep(1000);
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
// continue;
}
if (IgnoredFileNames.IndexOf(child.getName()) > -1) {
// if (Verbose == true) {System.out.print("~");}
continue;
}
if (Verbose == true) {
System.out.println("Considering file: [" + child.getName() + "]");
}
String[] namePieces = child.getName().split("_");
// (should only open filenames with format of <MODE>_<DFN>.xml)
if (Verbose == true) {
System.out.println("File Name Is: " + child.getName());
}
String fileExt = GetFileExtension(child.getName());
if (Verbose == true) {
System.out.println("File Ext Is: " + fileExt);
}
// At times, there seems to be a race condition, where this java process
// tries to load in the file before the mump process is done writting
// So I have modified the mumps process to that it outputs TWO files.
// 1st the <MODE>_<DFN>.xml file
// 2nd, after finished with above, it outputs <MODE>_<DFN>_READY.TXT
// So the 2nd file must exist before opening the <MODE>_<DFN>.xml
// The 2nd file need not be opened or processed. Just the fact that
// it exists on in the directory is signal that it is OK to process.
if (fileExt.equals("xml") == false) {
if (Verbose == true) {
System.out.println("Skipping file due to wrong extension.");
}
IgnoredFileNames.Add(child.getName());
continue;
}
if (((namePieces.length > 1) && (namePieces[0].equals(mode))) == false) {
if (Verbose == true) {
System.out.println("Skipping file due to wrong mode.");
}
IgnoredFileNames.Add(child.getName());
continue;
}
Boolean finishFileExists = finishFileExist(child.getPath());
if (Verbose == true) {
System.out.println("Ready File Exists?: " + finishFileExists);
}
// NOTE: after processing the <MODE>_<DFN>.xml, both files should be deleted
// NOTE: in java, one can't use '=' to compare strings -- that just compares a pointer to
// the string. Must use ".equals()"
if (finishFileExists == true) {
System.out.println("Found File to Process: [" + child.getName() + "]");
fileProcessed = true;
String patientID = namePieces[1];
patientID = patientID.split("\\.")[0];
File outFileName =
new File(outFilePath.getPath() + "/" + mode + "_" + patientID + ".xml");
String htmlFile = outFilePath.getPath() + "/" + mode + "_" + patientID + ".html";
System.out.println("SENDING TO: [" + outFileName.getPath() + "]");
String readyFileName;
readyFileName = outFileName.getPath();
try {
processFile(patientID, child, outFileName, kbID, htmlFile);
} catch (Throwable t) {
try {
FileWriter fwrite = new FileWriter(outFileName);
fwrite.write(
"-1^CDSS/Clinical Decision Support Engine(Athena) Error. " + t.getMessage());
fwrite.flush();
fwrite.close();
} catch (IOException e3) {
System.out.println("Could not create error message");
}
System.out.println("Error Processing Recommendation: " + t.getMessage());
// return;
}
System.out.println("Ready.TXT File Name Is: " + readyFileName); // Remove
File readyFile = new File(readyFileName.replace(".xml", "_READY.TXT"));
try {
readyFile.createNewFile();
FileWriter fstream = new FileWriter(readyFile);
BufferedWriter out = new BufferedWriter(fstream);
out.write("DONE");
out.close();
System.out.println("****SUCCESS****" + readyFile.getPath());
} catch (java.io.IOException e0) {
System.out.println("Error creating READY file");
}
if (child.delete() == false) {
System.out.println("Unable to delete: " + child.getName());
Thread.sleep(1000);
}
// Delete completed file
File f = new File(getFinishFileName(child.getPath()));
if (f.delete() == false) {
System.out.println("Unable to delete: " + f.getName());
Thread.sleep(1000);
}
}
}
if (fileProcessed == false) {
Thread.sleep(1000);
if (Verbose == true) {
System.out.print(".");
}
} else {
if (Verbose == true) {
System.out.println("Found process, so won't sleep...");
}
}
mainLoopRunning = false;
} // for loop
System.out.println("Finished with run.");
System.exit(0);
} // Main routine
/**
* @param since last modified time to use
* @param req
* @param url if null, ignored
* @param redirCount number of redirs we've done
*/
public static HttpData getDataOnce(
HttpServletRequest req,
HttpServletResponse res,
long since,
String surl,
int redirCount,
int timeout)
throws IOException, HttpException, DataSourceException, MalformedURLException {
HttpMethodBase request = null;
HostConfiguration hcfg = new HostConfiguration();
/*
[todo hqm 2006-02-01] Anyone know why this code was here? It is setting
the mime type to something which just confuses the DHTML parser.
if (res != null) {
res.setContentType("application/x-www-form-urlencoded;charset=UTF-8");
}
*/
try {
// TODO: [2002-01-09 bloch] cope with cache-control
// response headers (no-store, no-cache, must-revalidate,
// proxy-revalidate).
if (surl == null) {
surl = getURL(req);
}
if (surl == null || surl.equals("")) {
throw new MalformedURLException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="url is empty or null"
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-312"));
}
String reqType = "";
String headers = "";
if (req != null) {
reqType = req.getParameter("reqtype");
headers = req.getParameter("headers");
}
boolean isPost = false;
mLogger.debug("reqtype = " + reqType);
if (reqType != null && reqType.equals("POST")) {
request = new LZPostMethod();
request.setRequestHeader("Content-Type", "application/x-www-form-urlencoded;charset=UTF-8");
isPost = true;
mLogger.debug("setting POST req method");
} else if (reqType != null && reqType.equals("PUT")) {
request = new LZPutMethod();
// todo [hqm 2007] treat PUT like POST?
isPost = true;
mLogger.debug("setting PUT req method");
} else if (reqType != null && reqType.equals("DELETE")) {
request = new LZDeleteMethod();
mLogger.debug("setting DELETE req method");
} else {
mLogger.debug("setting GET (default) req method");
request = new LZGetMethod();
}
request.setHttp11(mUseHttp11);
// Proxy the request headers
if (req != null) {
LZHttpUtils.proxyRequestHeaders(req, request);
}
// Set headers from query string
if (headers != null && headers.length() > 0) {
StringTokenizer st = new StringTokenizer(headers, "\n");
while (st.hasMoreTokens()) {
String h = st.nextToken();
int i = h.indexOf(":");
if (i > -1) {
String n = h.substring(0, i);
String v = h.substring(i + 2, h.length());
request.setRequestHeader(n, v);
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="setting header " + p[0] + "=" + p[1]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-359", new Object[] {n, v}));
}
}
}
mLogger.debug("Parsing url");
URI uri = LZHttpUtils.newURI(surl);
try {
hcfg.setHost(uri);
} catch (Exception e) {
throw new MalformedURLException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="can't form uri from " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-376", new Object[] {surl}));
}
// This gets us the url-encoded (escaped) path and query string
String path = uri.getEscapedPath();
String query = uri.getEscapedQuery();
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="encoded path: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-389", new Object[] {path}));
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="encoded query: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-397", new Object[] {query}));
// This call takes a decoded (unescaped) path
request.setPath(path);
boolean hasQuery = (query != null && query.length() > 0);
String rawcontent = null;
// Newer rawpost protocol puts lzpostbody as a separate
// top level query arg in the request.
rawcontent = req.getParameter("lzpostbody");
if (isPost) {
// Older rawpost protocol put the "lzpostbody" arg
// embedded in the "url" args's query args
if (rawcontent == null && hasQuery) {
rawcontent = findQueryArg("lzpostbody", query);
}
if (rawcontent != null) {
// Get the unescaped query string
((EntityEnclosingMethod) request).setRequestBody(rawcontent);
} else if (hasQuery) {
StringTokenizer st = new StringTokenizer(query, "&");
while (st.hasMoreTokens()) {
String it = st.nextToken();
int i = it.indexOf("=");
if (i > 0) {
String n = it.substring(0, i);
String v = it.substring(i + 1, it.length());
// POST encodes values during request
((PostMethod) request).addParameter(n, URLDecoder.decode(v, "UTF-8"));
} else {
mLogger.warn(
/* (non-Javadoc)
* @i18n.test
* @org-mes="ignoring bad token (missing '=' char) in query string: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-429", new Object[] {it}));
}
}
}
} else {
// This call takes an encoded (escaped) query string
request.setQueryString(query);
}
// Put in the If-Modified-Since headers
if (since != -1) {
String lms = LZHttpUtils.getDateString(since);
request.setRequestHeader(LZHttpUtils.IF_MODIFIED_SINCE, lms);
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="proxying lms: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-450", new Object[] {lms}));
}
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="setting up http client"
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-460"));
HttpClient htc = null;
if (mConnectionMgr != null) {
htc = new HttpClient(mConnectionMgr);
} else {
htc = new HttpClient();
}
htc.setHostConfiguration(hcfg);
// This is the data timeout
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="timeout set to " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-478", new Object[] {new Integer(timeout)}));
htc.setTimeout(timeout);
// Set connection timeout the same
htc.setConnectionTimeout(mConnectionTimeout);
// Set timeout for getting a connection
htc.setHttpConnectionFactoryTimeout(mConnectionPoolTimeout);
// TODO: [2003-03-05 bloch] this should be more configurable (per app?)
if (!isPost) {
request.setFollowRedirects(mFollowRedirects > 0);
}
long t1 = System.currentTimeMillis();
mLogger.debug("starting remote request");
int rc = htc.executeMethod(hcfg, request);
String status = HttpStatus.getStatusText(rc);
if (status == null) {
status = "" + rc;
}
mLogger.debug(
/* (non-Javadoc)
* @i18n.test
* @org-mes="remote response status: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-504", new Object[] {status}));
HttpData data = null;
if (isRedirect(rc) && mFollowRedirects > redirCount) {
String loc = request.getResponseHeader("Location").toString();
String hostURI = loc.substring(loc.indexOf(": ") + 2, loc.length());
mLogger.info(
/* (non-Javadoc)
* @i18n.test
* @org-mes="Following URL from redirect: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-517", new Object[] {hostURI}));
long t2 = System.currentTimeMillis();
if (timeout > 0) {
timeout -= (t2 - t1);
if (timeout < 0) {
throw new InterruptedIOException(
/* (non-Javadoc)
* @i18n.test
* @org-mes=p[0] + " timed out after redirecting to " + p[1]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-529", new Object[] {surl, loc}));
}
}
data = getDataOnce(req, res, since, hostURI, redirCount++, timeout);
} else {
data = new HttpData(request, rc);
}
if (req != null && res != null) {
// proxy response headers
LZHttpUtils.proxyResponseHeaders(request, res, req.isSecure());
}
return data;
} catch (ConnectTimeoutException ce) {
// Transduce to an InterrupedIOException, since lps takes these to be timeouts.
if (request != null) {
request.releaseConnection();
}
throw new InterruptedIOException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="connecting to " + p[0] + ":" + p[1] + " timed out beyond " + p[2] + " msecs."
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(),
"051018-557",
new Object[] {
hcfg.getHost(), new Integer(hcfg.getPort()), new Integer(mConnectionTimeout)
}));
} catch (HttpRecoverableException hre) {
if (request != null) {
request.releaseConnection();
}
throw hre;
} catch (HttpException e) {
if (request != null) {
request.releaseConnection();
}
throw e;
} catch (IOException ie) {
if (request != null) {
request.releaseConnection();
}
throw ie;
} catch (RuntimeException e) {
if (request != null) {
request.releaseConnection();
}
throw e;
}
}
public static Data getHTTPData(
HttpServletRequest req, HttpServletResponse res, String surl, long since)
throws DataSourceException, IOException {
int tries = 1;
// timeout msecs of time we're allowed in this routine
// we must return or throw an exception. 0 means infinite.
int timeout = mTimeout;
if (req != null) {
String timeoutParm = req.getParameter("timeout");
if (timeoutParm != null) {
timeout = Integer.parseInt(timeoutParm);
}
}
long t1 = System.currentTimeMillis();
long elapsed = 0;
if (surl == null) {
surl = getURL(req);
}
while (true) {
String m = null;
long tout;
if (timeout > 0) {
tout = timeout - elapsed;
if (tout <= 0) {
throw new InterruptedIOException(
/* (non-Javadoc)
* @i18n.test
* @org-mes=p[0] + " timed out"
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-194", new Object[] {surl}));
}
} else {
tout = 0;
}
try {
HttpData data = getDataOnce(req, res, since, surl, 0, (int) tout);
if (data.code >= 400) {
data.release();
throw new DataSourceException(errorMessage(data.code));
}
return data;
} catch (HttpRecoverableException e) {
// This type of exception should be retried.
if (tries++ > mMaxRetries) {
throw new InterruptedIOException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="too many retries, exception: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-217", new Object[] {e.getMessage()}));
}
mLogger.warn(
/* (non-Javadoc)
* @i18n.test
* @org-mes="retrying a recoverable exception: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-226", new Object[] {e.getMessage()}));
} catch (HttpException e) {
String msg =
/* (non-Javadoc)
* @i18n.test
* @org-mes="HttpException: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-235", new Object[] {e.getMessage()});
throw new IOException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="HttpException: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-235", new Object[] {e.getMessage()}));
} catch (IOException e) {
try {
Class ssle = Class.forName("javax.net.ssl.SSLException");
if (ssle.isAssignableFrom(e.getClass())) {
throw new DataSourceException(
/* (non-Javadoc)
* @i18n.test
* @org-mes="SSL exception: " + p[0]
*/
org.openlaszlo.i18n.LaszloMessages.getMessage(
HTTPDataSource.class.getName(), "051018-256", new Object[] {e.getMessage()}));
}
} catch (ClassNotFoundException cfne) {
}
throw e;
}
long t2 = System.currentTimeMillis();
elapsed = (t2 - t1);
}
}
public static void main(String[] args) throws Exception {
Config.hadoop_mode = false;
if (args.length == 2 && args[0].equals("args")) // needed for mrql.flink script
args = args[1].substring(1).split("!");
for (String arg : args) {
Config.hadoop_mode |= arg.equals("-local") || arg.equals("-dist");
Config.bsp_mode |= arg.equals("-bsp");
Config.spark_mode |= arg.equals("-spark");
Config.flink_mode |= arg.equals("-flink");
}
;
Config.map_reduce_mode = !Config.bsp_mode && !Config.spark_mode && !Config.flink_mode;
initialize_evaluator();
if (Config.hadoop_mode) {
conf = Evaluator.evaluator.new_configuration();
GenericOptionsParser gop = new GenericOptionsParser(conf, args);
conf = gop.getConfiguration();
args = gop.getRemainingArgs();
}
;
Config.parse_args(args, conf);
Config.hadoop_mode = Config.local_mode || Config.distributed_mode;
if (!Config.info) {
for (Enumeration en = LogManager.getCurrentLoggers(); en.hasMoreElements(); )
((Logger) en.nextElement()).setLevel(Level.WARN);
LogManager.getRootLogger().setLevel(Level.WARN);
}
;
Evaluator.evaluator.init(conf);
new TopLevel();
System.out.print("Apache MRQL version " + version + " (");
if (Config.compile_functional_arguments) System.out.print("compiled ");
else System.out.print("interpreted ");
if (Config.hadoop_mode) {
if (Config.local_mode) System.out.print("local ");
else if (Config.distributed_mode) System.out.print("distributed ");
if (Config.spark_mode) System.out.println("Spark mode using " + Config.nodes + " tasks)");
else if (Config.flink_mode)
System.out.println("Flink mode using " + Config.nodes + " tasks)");
else if (Config.bsp_mode)
System.out.println("Hama BSP mode over " + Config.nodes + " BSP tasks)");
else if (Config.nodes > 0)
System.out.println("Hadoop MapReduce mode with " + Config.nodes + " reducers)");
else if (!Config.local_mode)
System.out.println("Hadoop MapReduce mode with 1 reducer, use -nodes to change it)");
else System.out.println("Hadoop MapReduce mode)");
} else if (Config.bsp_mode) System.out.println("in-memory BSP mode)");
else System.out.println("in-memory Java mode)");
if (Config.interactive) {
System.out.println("Type quit to exit");
ConsoleReader reader = new ConsoleReader();
reader.setBellEnabled(false);
History history = new History(new File(System.getProperty("user.home") + "/.mrqlhistory"));
reader.setHistory(history);
reader.setUseHistory(false);
try {
loop:
while (true) {
String line = "";
String s = "";
try {
if (Config.hadoop_mode && Config.bsp_mode) Config.write(Plan.conf);
do {
s = reader.readLine("> ");
if (s != null && (s.equals("quit") || s.equals("exit"))) break loop;
if (s != null) line += " " + s;
} while (s == null || s.indexOf(";") <= 0);
line = line.substring(1);
history.addToHistory(line);
parser = new MRQLParser(new MRQLLex(new StringReader(line)));
MRQLLex.reset();
parser.parse();
} catch (EOFException x) {
break;
} catch (Exception x) {
if (x.getMessage() != null) System.out.println(x);
} catch (Error x) {
System.out.println(x);
}
}
} finally {
if (Config.hadoop_mode) {
Plan.clean();
Evaluator.evaluator.shutdown(Plan.conf);
}
;
if (Config.compile_functional_arguments) Compiler.clean();
}
} else
try {
if (Config.hadoop_mode && Config.bsp_mode) Config.write(Plan.conf);
try {
parser = new MRQLParser(new MRQLLex(new FileInputStream(query_file)));
} catch (Exception e) {
// when the query file is in HDFS
Path path = new Path(query_file);
FileSystem fs = path.getFileSystem(conf);
parser = new MRQLParser(new MRQLLex(fs.open(path)));
}
;
parser.parse();
} finally {
if (Config.hadoop_mode) {
Plan.clean();
Evaluator.evaluator.shutdown(Plan.conf);
}
;
if (Config.compile_functional_arguments) Compiler.clean();
}
}
public static void main(String args[]) {
final String S_ProcName = "CFInternetSaxMySqlLoaderCLI.main() ";
initConsoleLog();
int numArgs = args.length;
if (numArgs >= 2) {
String homeDirName = System.getProperty("HOME");
if (homeDirName == null) {
homeDirName = System.getProperty("user.home");
if (homeDirName == null) {
log.message(S_ProcName + "ERROR: Home directory not set");
return;
}
}
File homeDir = new File(homeDirName);
if (!homeDir.exists()) {
log.message(S_ProcName + "ERROR: Home directory \"" + homeDirName + "\" does not exist");
return;
}
if (!homeDir.isDirectory()) {
log.message(
S_ProcName + "ERROR: Home directory \"" + homeDirName + "\" is not a directory");
return;
}
CFInternetConfigurationFile cFInternetConfig = new CFInternetConfigurationFile();
String cFInternetConfigFileName = homeDir.getPath() + File.separator + ".cfinternetmysqlrc";
cFInternetConfig.setFileName(cFInternetConfigFileName);
File cFInternetConfigFile = new File(cFInternetConfigFileName);
if (!cFInternetConfigFile.exists()) {
cFInternetConfig.setDbServer("127.0.0.1");
cFInternetConfig.setDbPort(3306);
cFInternetConfig.setDbDatabase("CFINet24");
cFInternetConfig.setDbUserName("root");
cFInternetConfig.setDbPassword("edit-me-please");
cFInternetConfig.save();
log.message(
S_ProcName
+ "INFO: Created configuration file "
+ cFInternetConfigFileName
+ ", please edit configuration and restart.");
return;
}
if (!cFInternetConfigFile.isFile()) {
log.message(
S_ProcName
+ "ERROR: Proposed configuration file "
+ cFInternetConfigFileName
+ " is not a file.");
return;
}
if (!cFInternetConfigFile.canRead()) {
log.message(
S_ProcName
+ "ERROR: Permission denied attempting to read configuration file "
+ cFInternetConfigFileName);
return;
}
cFInternetConfig.load();
boolean fastExit = false;
CFInternetClientConfigurationFile cFDbTestClientConfig =
new CFInternetClientConfigurationFile();
String cFDbTestClientConfigFileName =
homeDir.getPath() + File.separator + ".cfdbtestclientrc";
cFDbTestClientConfig.setFileName(cFDbTestClientConfigFileName);
File cFDbTestClientConfigFile = new File(cFDbTestClientConfigFileName);
if (!cFDbTestClientConfigFile.exists()) {
String cFDbTestKeyStoreFileName = homeDir.getPath() + File.separator + ".msscfjceks";
cFDbTestClientConfig.setKeyStore(cFDbTestKeyStoreFileName);
InetAddress localHost;
try {
localHost = InetAddress.getLocalHost();
} catch (UnknownHostException e) {
localHost = null;
}
if (localHost == null) {
log.message(S_ProcName + "ERROR: LocalHost is null");
return;
}
String hostName = localHost.getHostName();
if ((hostName == null) || (hostName.length() <= 0)) {
log.message("ERROR: LocalHost.HostName is null or empty");
return;
}
String userName = System.getProperty("user.name");
if ((userName == null) || (userName.length() <= 0)) {
log.message("ERROR: user.name is null or empty");
return;
}
String deviceName =
hostName.replaceAll("[^\\w]", "_").toLowerCase()
+ "-"
+ userName.replaceAll("[^\\w]", "_").toLowerCase();
cFDbTestClientConfig.setDeviceName(deviceName);
cFDbTestClientConfig.save();
log.message(
S_ProcName
+ "INFO: Created CFInternet client configuration file "
+ cFDbTestClientConfigFileName);
fastExit = true;
}
if (!cFDbTestClientConfigFile.isFile()) {
log.message(
S_ProcName
+ "ERROR: Proposed client configuration file "
+ cFDbTestClientConfigFileName
+ " is not a file.");
fastExit = true;
}
if (!cFDbTestClientConfigFile.canRead()) {
log.message(
S_ProcName
+ "ERROR: Permission denied attempting to read client configuration file "
+ cFDbTestClientConfigFileName);
fastExit = true;
}
cFDbTestClientConfig.load();
if (fastExit) {
return;
}
// Configure logging
Properties sysProps = System.getProperties();
sysProps.setProperty("log4j.rootCategory", "WARN");
sysProps.setProperty(
"org.apache.commons.logging.Log", "org.apache.commons.logging.impl.Log4JLogger");
Logger httpLogger = Logger.getLogger("org.apache.http");
httpLogger.setLevel(Level.WARN);
ICFInternetSchema cFInternetSchema = new CFInternetMySqlSchema();
cFInternetSchema.setConfigurationFile(cFInternetConfig);
ICFInternetSchemaObj cFInternetSchemaObj = new CFInternetSchemaObj();
cFInternetSchemaObj.setBackingStore(cFInternetSchema);
CFInternetSaxLoaderCLI cli = new CFInternetSaxMySqlLoaderCLI();
CFInternetSaxLoader loader = cli.getSaxLoader();
loader.setSchemaObj(cFInternetSchemaObj);
cFInternetSchema.connect();
String url = args[1];
if (numArgs >= 5) {
cli.setClusterName(args[2]);
cli.setTenantName(args[3]);
cli.setSecUserName(args[4]);
} else {
cli.setClusterName("default");
cli.setTenantName("system");
cli.setSecUserName("system");
}
loader.setUseCluster(cli.getClusterObj());
loader.setUseTenant(cli.getTenantObj());
try {
cFInternetSchema.beginTransaction();
cFInternetSchemaObj.setSecCluster(cli.getClusterObj());
cFInternetSchemaObj.setSecTenant(cli.getTenantObj());
cFInternetSchemaObj.setSecUser(cli.getSecUserObj());
cFInternetSchemaObj.setSecSession(cli.getSecSessionObj());
CFSecurityAuthorization auth = new CFSecurityAuthorization();
auth.setSecCluster(cFInternetSchemaObj.getSecCluster());
auth.setSecTenant(cFInternetSchemaObj.getSecTenant());
auth.setSecSession(cFInternetSchemaObj.getSecSession());
cFInternetSchemaObj.setAuthorization(auth);
applyLoaderOptions(loader, args[0]);
if (numArgs >= 5) {
cli.evaluateRemainingArgs(args, 5);
} else {
cli.evaluateRemainingArgs(args, 2);
}
loader.parseFile(url);
cFInternetSchema.commit();
cFInternetSchema.disconnect(true);
} catch (Exception e) {
log.message(
S_ProcName + "EXCEPTION: Could not parse XML file \"" + url + "\": " + e.getMessage());
e.printStackTrace(System.out);
} catch (Error e) {
log.message(
S_ProcName + "ERROR: Could not parse XML file \"" + url + "\": " + e.getMessage());
e.printStackTrace(System.out);
} finally {
if (cFInternetSchema.isConnected()) {
cFInternetSchema.rollback();
cFInternetSchema.disconnect(false);
}
}
} else {
log.message(
S_ProcName
+ "ERROR: Expected at least two argument specifying the loader options and the name of the XML file to parse. The first argument may be empty.");
}
}
/**
* Test suite for distributing JUnit3 tests. Simply add tests to this suite just like you would for
* regular JUnit3 suites, and these tests will be executed in parallel on the grid. Note that if
* there are no other grid nodes, this suite will still ensure parallel test execution within single
* JVM.
*
* <p>Below is an example of distributed JUnit3 test suite:
*
* <pre name="code" class="java">
* public class GridJunit3ExampleTestSuite {
* // Standard JUnit3 static suite method.
* public static TestSuite suite() {
* TestSuite suite = new GridJunit3TestSuite("Example Grid Test Suite");
*
* // Add tests.
* suite.addTestSuite(TestA.class);
* suite.addTestSuite(TestB.class);
* suite.addTestSuite(TestC.class);
*
* return suite;
* }
* }
* </pre>
*
* If you have four tests A, B, C, and D, and if you need to run A and B sequentially, then you
* should create a nested test suite with test A and B as follows:
*
* <pre name="code" class="java">
* public class GridJunit3ExampleTestSuite {
* // Standard JUnit3 static suite method.
* public static TestSuite suite() {
* TestSuite suite = new GridJunit3TestSuite("Example Grid Test Suite");
*
* // Nested test suite to run tests A and B sequentially.
* TestSuite nested = new TestSuite("Example Nested Sequential Suite");
*
* nested.addTestSuite(TestA.class);
* nested.addTestSuite(TestB.class);
*
* // Add tests A and B.
* suite.addTest(nested);
*
* // Add other tests.
* suite.addTestSuite(TestC.class);
*
* return suite;
* }
* }
* </pre>
*
* <p>Note that you can also grid-enable existing JUnit3 tests using {@link
* GridifyTest @GridifyTest} annotation which you can attach to your {@code suite()} methods of
* existing test suite. Refer to {@link GridifyTest @GridifyTest} documentation for more
* information.
*
* <p>Also note that some tests can only be executed locally mostly due to some environment issues.
* However they still can benefit from parallel execution with other tests. GridGain supports it via
* {@link GridJunit3LocalTestSuite} suites that can be added to {@code GridJunit3TestSuite}. Refer
* to {@link GridJunit3LocalTestSuite} documentation for more information.
*
* <h1 class="header">Logging</h1>
*
* When running distributed JUnit, all the logging that is done to {@link System#out} or {@link
* System#err} is preserved. GridGain will accumulate all logging that is done on remote nodes, send
* them back to originating node and associate all log statements with their corresponding tests.
* This way, for example, if you are running tests from and IDEA or Eclipse (or any other IDE) you
* would still see the logs as if it was a local run. However, since remote nodes keep all log
* statements done within a single individual test case in memory, you must make sure that enough
* memory is allocated on every node and that individual test cases do not spit out gigabytes of log
* statements. Also note, that logs will be sent back to originating node upon completion of every
* test, so don't be alarmed if you don't see any log statements for a while and then all of them
* appear at once.
*
* <p>GridGain achieves such log transparency via reassigning {@link System#out} or {@link
* System#err} to internal {@link PrintStream} implementation. However, when using {@code Log4J} (or
* any other logging framework) within your tests you must make sure that it is configured with
* {@link ConsoleAppender} and that {@link ConsoleAppender#setFollow(boolean)} attribute is set to
* {@code true}. Logging to files is not supported yet and is planned for next point release.
*
* <p>
*
* <h1 class="header">Test Suite Nesting</h1>
*
* {@code GridJunit3TestSuite} instances can be nested within each other as deep as needed. However
* all nested distributed test suites will be treated just like regular JUnit test suites, and not
* as distributed test suites. This approach becomes convenient when you have several distributed
* test suites that you would like to be able to execute separately in distributed fashion, but at
* the same time you would like to be able to execute them as a part of larger distributed suites.
*
* <p>
*
* <h1 class="header">Configuration</h1>
*
* To run distributed JUnit tests you need to start other instances of GridGain. You can do so by
* running {@code GRIDGAIN_HOME/bin/ggjunit.{sh|bat}} script, which will start default
* configuration. If configuration other than default is required, then use regular {@code
* GRIDGAIN_HOME/bin/ggstart.{sh|bat}} script and pass your own Spring XML configuration file as a
* parameter to the script.
*
* <p>You can use the following configuration parameters to configure distributed test suite
* locally. Note that many parameters can be overridden by setting corresponding VM parameters
* defined in {@link GridTestVmParameters} at VM startup.
*
* <table class="doctable">
* <tr>
* <th>GridConfiguration Method</th>
* <th>Default Value</th>
* <th>Description</th>
* </tr>
* <tr>
* <td>{@link #setDisabled(boolean) setDisabled(boolean)}</td>
* <td>{@code false}</td>
* <td>
* If {@code true} then GridGain will be turned off and suite will run locally.
* This value can be overridden by setting {@link GridTestVmParameters#GRIDGAIN_DISABLED} VM
* parameter to {@code true}. This parameter comes handy when you would like to
* turn off GridGain without changing the actual code.
* </td>
* </tr>
* <tr>
* <td>{@link #setConfigurationPath(String) setConfigurationPath(String)}</td>
* <td>{@link #DFLT_JUNIT_CONFIG DFLT_JUNIT_CONFIG}</td>
* <td>
* Optional path to GridGain Spring XML configuration file for running JUnit tests. This
* property can be overridden by setting {@link GridTestVmParameters#GRIDGAIN_CONFIG} VM
* parameter. Note that the value can be either absolute value or relative to
* ${GRIDGAIN_HOME} installation folder.
* </td>
* </tr>
* <tr>
* <td>{@link #setRouterClassName(String) setRouterClassName(String)}</td>
* <td>{@link #DFLT_JUNIT_ROUTER DFLT_JUNIT_ROUTER}</td>
* <td>
* Optional name of test router class that implements {@link GridTestRouter} interface.
* If not provided, then tests will be routed in round-robin fashion using default
* {@link GridTestRouterAdapter}. The value of this parameter can be overridden by setting
* {@link GridTestVmParameters#GRIDGAIN_TEST_ROUTER} VM parameter to the name of your
* own customer router class.
* </td>
* </tr>
* <tr>
* <td>{@link #setRouterClass(Class) setRouterClass(Class)}</td>
* <td>{@code null}</td>
* <td>
* Same as {@link #setRouterClassName(String) setRouterClassName(String)}, but sets the
* actual class instead of the name.
* </td>
* </tr>
* <tr>
* <td>{@link #setTimeout(long) setTimeout(long)}</td>
* <td>{@code 0} which means that tests will never timeout.</td>
* <td>
* Maximum timeout value in milliseconds after which test suite will return without
* waiting for the remaining tests to complete. This value can be overridden by setting
* {@link GridTestVmParameters#GRIDGAIN_TEST_TIMEOUT} VM parameter to the timeout value
* for the tests.
* </td>
* </tr>
* </table>
*
* @author 2005-2011 Copyright (C) GridGain Systems, Inc.
* @version 3.1.1c.19062011
*/
public class GridJunit3TestSuite extends TestSuite {
/**
* Default GridGain configuration file for JUnits (value is {@code
* config/junit/junit-spring.xml}).
*/
public static final String DFLT_JUNIT_CONFIG = "config/junit/junit-spring.xml";
/**
* Default JUnit test router (value is {@link GridTestRouterAdapter
* GridTestRouterAdapter.class.getName()}).
*/
public static final String DFLT_JUNIT_ROUTER = GridTestRouterAdapter.class.getName();
/** */
private final Collection<String> locTests = new HashSet<String>();
/** JUnit3 JavaAssist proxy. */
private final GridJunit3ProxyFactory factory = new GridJunit3ProxyFactory();
/** Flag indicating whether grid was started in this suite. */
private boolean selfStarted;
/** Junit3 Spring configuration path. */
private String cfgPath =
System.getProperty(GRIDGAIN_CONFIG.name()) == null
? DFLT_JUNIT_CONFIG
: System.getProperty(GRIDGAIN_CONFIG.name());
/**
* Check if GridGain is disabled by checking {@link GridTestVmParameters#GRIDGAIN_DISABLED} system
* property.
*/
private boolean isDisabled = Boolean.getBoolean(GRIDGAIN_DISABLED.name());
/** JUnit test router class name. */
private String routerClsName =
System.getProperty(GRIDGAIN_TEST_ROUTER.name()) == null
? DFLT_JUNIT_ROUTER
: System.getProperty(GRIDGAIN_TEST_ROUTER.name());
/** JUnit test router class. */
private Class<? extends GridTestRouter> routerCls;
/**
* Local suite in case if grid is disabled or if this is a nested suite within other distributed
* suite.
*/
private TestSuite copy;
/** JUnit grid name. */
private String gridName;
/** Test timeout. */
private long timeout =
Long.getLong(GRIDGAIN_TEST_TIMEOUT.name()) == null
? 0
: Long.getLong(GRIDGAIN_TEST_TIMEOUT.name());
/** */
private ClassLoader clsLdr;
/** Empty test suite. */
public GridJunit3TestSuite() {
if (copy == null) {
copy = new TestSuite();
}
}
/** @param name Test suite name. */
public GridJunit3TestSuite(String name) {
super(name);
if (copy == null) {
copy = new TestSuite(name);
}
}
/**
* Test suite for one class.
*
* @param cls Class for test suite.
*/
public GridJunit3TestSuite(Class<? extends TestCase> cls) {
super(cls);
if (copy == null) {
copy = new TestSuite(cls);
}
}
/**
* Test suite for a given test class with specified test name.
*
* @param cls Test class.
* @param name Test name.
*/
public GridJunit3TestSuite(Class<? extends TestCase> cls, String name) {
super(cls, name);
if (copy == null) {
copy = new TestSuite(cls, name);
}
}
/**
* Copies non-distributed test suite into distributed one.
*
* @param suite Test suite to copy.
*/
public GridJunit3TestSuite(TestSuite suite) {
super(suite.getName());
if (copy == null) {
copy = new TestSuite(suite.getName());
}
for (int i = 0; i < suite.testCount(); i++) {
addTest(suite.testAt(i));
}
}
/**
* Empty test suite with given class loader.
*
* @param clsLdr Tests class loader.
*/
public GridJunit3TestSuite(ClassLoader clsLdr) {
this();
assert clsLdr != null;
this.clsLdr = clsLdr;
}
/**
* @param name Test suite name.
* @param clsLdr Tests class loader.
*/
public GridJunit3TestSuite(String name, ClassLoader clsLdr) {
this(name);
assert clsLdr != null;
this.clsLdr = clsLdr;
}
/**
* Test suite for one class.
*
* @param cls Class for test suite.
* @param clsLdr Tests class loader.
*/
public GridJunit3TestSuite(Class<? extends TestCase> cls, ClassLoader clsLdr) {
this(cls);
assert clsLdr != null;
this.clsLdr = clsLdr;
}
/**
* Test suite for a given test class with specified test name.
*
* @param cls Test class.
* @param name Test name.
* @param clsLdr Tests class loader.
*/
public GridJunit3TestSuite(Class<? extends TestCase> cls, String name, ClassLoader clsLdr) {
this(cls, name);
assert clsLdr != null;
this.clsLdr = clsLdr;
}
/**
* Copies non-distributed test suite into distributed one.
*
* @param suite Test suite to copy.
* @param clsLdr Tests class loader.
*/
public GridJunit3TestSuite(TestSuite suite, ClassLoader clsLdr) {
this(suite);
assert clsLdr != null;
this.clsLdr = clsLdr;
}
/**
* Sets path to GridGain configuration file. By default {@code
* {GRIDGAIN_HOME}/config/junit/junit-spring.xml} is used.
*
* @param cfgPath Path to GridGain configuration file.
*/
public void setConfigurationPath(String cfgPath) {
this.cfgPath = cfgPath;
}
/**
* Gets path to GridGain configuration file. By default {@code
* {GRIDGAIN_HOME}/config/junit/junit-spring.xml} is used.
*
* @return Path to GridGain configuration file.
*/
public String getConfigurationPath() {
return cfgPath;
}
/**
* Disables GridGain. If set to {@code true} then this suite will execute locally as if GridGain
* was not in a picture at all.
*
* @param disabled If set to {@code true} then this suite will execute locally as if GridGain was
* not in a picture at all.
*/
public void setDisabled(boolean disabled) {
isDisabled = disabled;
}
/**
* Gets flag indicating whether GridGain should be enabled or not. If set to {@code true} then
* this suite will execute locally as if GridGain was not in a picture at all.
*
* @return Flag indicating whether GridGain should be enabled or not. If set to {@code true} then
* this suite will execute locally as if GridGain was not in a picture at all.
*/
public boolean isDisabled() {
return isDisabled;
}
/**
* Sets name of class for routing JUnit tests. By default {@link #DFLT_JUNIT_ROUTER} class name is
* used.
*
* @param routerClsName Junit test router class name.
*/
public void setRouterClassName(String routerClsName) {
this.routerClsName = routerClsName;
}
/**
* Gets JUnit test router class name.
*
* @return JUnit test router class name.
*/
public String getRouterClassName() {
return routerClsName;
}
/**
* Sets router class. By default {@link GridTestRouterAdapter} is used.
*
* @param routerCls Router class to use for test routing.
*/
public void setRouterClass(Class<? extends GridTestRouter> routerCls) {
this.routerCls = routerCls;
}
/**
* Gets router class used for test routing.
*
* @return Router class used for test routing.
*/
public Class<? extends GridTestRouter> getRouterClass() {
return routerCls;
}
/**
* Gets identical suite for local (non-distributed) execution.
*
* @return Local suite.
*/
public TestSuite getLocalCopy() {
return copy;
}
/** {@inheritDoc} */
@Override
public void setName(String name) {
if (copy != null) {
copy.setName(name);
}
super.setName(name);
}
/**
* Gets timeout for running distributed test suite.
*
* @return Timeout for tests.
*/
public long getTimeout() {
return timeout;
}
/**
* Sets timeout for running distributed test suite. By default, test execution does not expire.
*
* @param timeout Timeout for tests.
*/
public void setTimeout(long timeout) {
this.timeout = timeout;
}
/** {@inheritDoc} */
@Override
public Test testAt(int index) {
return isDisabled ? copy.testAt(index) : super.testAt(index);
}
/** {@inheritDoc} */
@Override
public int testCount() {
return isDisabled ? copy.testCount() : super.testCount();
}
/** {@inheritDoc} */
@Override
public Enumeration<Test> tests() {
return isDisabled ? copy.tests() : super.tests();
}
/**
* The added suite will be always executed locally, but in parallel with other locally or remotely
* running tests. This comes handy for tests that cannot be distributed for some environmental
* reasons, but still would benefit from parallel execution.
*
* <p>Note, that local suites will be executed on local node even if grid topology only allows
* remote nodes.
*
* @param localSuite Test to execute locally in parallel with other local or distributed tests.
*/
@SuppressWarnings({"TypeMayBeWeakened"})
public void addTest(GridJunit3LocalTestSuite localSuite) {
if (!locTests.contains(localSuite.getName())) {
locTests.add(localSuite.getName());
}
addTest((Test) localSuite);
}
/**
* Adds a test to be executed on the grid. In case of test suite, all tests inside of test suite
* will be executed sequentially on some remote node.
*
* @param test Test to add.
*/
@Override
public void addTest(Test test) {
if (copy == null) {
copy = new TestSuite(getName());
}
// Add test to the list of local ones.
if (test instanceof GridJunit3LocalTestSuite) {
String testName = ((TestSuite) test).getName();
if (!locTests.contains(testName)) {
locTests.add(testName);
}
}
if (test instanceof GridJunit3TestSuite) {
copy.addTest(((GridJunit3TestSuite) test).copy);
super.addTest(new GridJunit3TestSuiteProxy(((GridJunit3TestSuite) test).copy, factory));
} else if (test instanceof GridJunit3TestSuiteProxy) {
copy.addTest(((GridJunit3TestSuiteProxy) test).getOriginal());
super.addTest(test);
} else if (test instanceof GridJunit3TestCaseProxy) {
copy.addTest(((GridJunit3TestCaseProxy) test).getGridGainJunit3OriginalTestCase());
super.addTest(test);
} else if (test instanceof TestSuite) {
copy.addTest(test);
super.addTest(new GridJunit3TestSuiteProxy((TestSuite) test, factory));
} else {
assert test instanceof TestCase
: "Test must be either instance of TestSuite or TestCase: " + test;
copy.addTest(test);
super.addTest(factory.createProxy((TestCase) test));
}
}
/**
* Creates JUnit test router. Note that router must have a no-arg constructor.
*
* @return JUnit router instance.
*/
@SuppressWarnings({"unchecked"})
private GridTestRouter createRouter() {
try {
if (routerCls == null) {
routerCls = (Class<? extends GridTestRouter>) Class.forName(routerClsName);
} else {
routerClsName = routerCls.getName();
}
return routerCls.newInstance();
} catch (ClassNotFoundException e) {
throw new GridRuntimeException("Failed to initialize JUnit router: " + routerClsName, e);
} catch (IllegalAccessException e) {
throw new GridRuntimeException("Failed to initialize JUnit router: " + routerClsName, e);
} catch (InstantiationException e) {
throw new GridRuntimeException("Failed to initialize JUnit router: " + routerClsName, e);
}
}
/**
* Runs all tests belonging to this test suite on the grid.
*
* @param result Test result collector.
*/
@Override
public void run(TestResult result) {
if (isDisabled) {
copy.run(result);
} else {
GridTestRouter router = createRouter();
Grid grid = startGrid();
try {
List<GridTaskFuture<?>> futs = new ArrayList<GridTaskFuture<?>>(testCount());
List<GridJunit3SerializableTest> tests =
new ArrayList<GridJunit3SerializableTest>(testCount());
for (int i = 0; i < testCount(); i++) {
Test junit = testAt(i);
GridJunit3SerializableTest test;
if (junit instanceof TestSuite) {
test = new GridJunit3SerializableTestSuite((TestSuite) junit);
} else {
assert junit instanceof TestCase
: "Test must be either TestSuite or TestCase: " + junit;
test = new GridJunit3SerializableTestCase((TestCase) junit);
}
tests.add(test);
if (clsLdr == null) {
clsLdr = U.detectClassLoader(junit.getClass());
}
futs.add(
grid.execute(
new GridJunit3Task(junit.getClass(), clsLdr),
new GridJunit3Argument(router, test, locTests.contains(test.getName())),
timeout));
}
for (int i = 0; i < testCount(); i++) {
GridTaskFuture<?> fut = futs.get(i);
GridJunit3SerializableTest origTest = tests.get(i);
try {
GridJunit3SerializableTest resTest = (GridJunit3SerializableTest) fut.get();
origTest.setResult(resTest);
origTest.getTest().run(result);
} catch (GridException e) {
handleFail(result, origTest, e);
}
}
} finally {
stopGrid();
}
}
}
/**
* Handles test fail.
*
* @param result Test result.
* @param origTest Original JUnit test.
* @param e Exception thrown from grid.
*/
private void handleFail(TestResult result, GridJunit3SerializableTest origTest, Throwable e) {
// Simulate that all tests were run.
origTest.getTest().run(result);
// For the tests suite we assume that all tests failed because
// entire test suite execution failed and there is no way to get
// broken tests.
if (origTest.getTest() instanceof GridJunit3TestSuiteProxy) {
TestSuite suite = (((TestSuite) origTest.getTest()));
for (int j = 0; j < suite.testCount(); j++) {
result.addError(suite.testAt(j), e);
}
} else if (origTest.getTest() instanceof GridJunit3TestCaseProxy) {
result.addError(origTest.getTest(), e);
}
}
/**
* Starts Grid instance. Note that if grid is already started, then it will be looked up and
* returned from this method.
*
* @return Started grid.
*/
private Grid startGrid() {
Properties props = System.getProperties();
gridName = props.getProperty(GRIDGAIN_NAME.name());
if (!props.containsKey(GRIDGAIN_NAME.name()) || G.state(gridName) != GridFactoryState.STARTED) {
selfStarted = true;
// Set class loader for the spring.
ClassLoader curCl = Thread.currentThread().getContextClassLoader();
// Add no-op logger to remove no-appender warning.
Appender app = new NullAppender();
Logger.getRootLogger().addAppender(app);
try {
Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
Grid grid = G.start(cfgPath);
gridName = grid.name();
System.setProperty(GRIDGAIN_NAME.name(), grid.name());
return grid;
} catch (GridException e) {
throw new GridRuntimeException("Failed to start grid: " + cfgPath, e);
} finally {
Logger.getRootLogger().removeAppender(app);
Thread.currentThread().setContextClassLoader(curCl);
}
}
return G.grid(gridName);
}
/** Stops grid only if it was started by this test suite. */
private void stopGrid() {
// Only stop grid if it was started here.
if (selfStarted) {
G.stop(gridName, true);
}
}
}
protected long getCurrentTimeMillis() {
return System.currentTimeMillis();
}
