GDLMadushanka
8 years ago
parent
1375b89a4c
commit
5c782a6c15
@ -0,0 +1,335 @@
|
||||
|
||||
|
||||
//
|
||||
// Source code recreated from a .class file by IntelliJ IDEA
|
||||
// (powered by Fernflower decompiler)
|
||||
//
|
||||
|
||||
package org.wso2.iot.integration.common.extensions;
|
||||
|
||||
import org.apache.commons.lang3.ArrayUtils;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.automation.engine.context.AutomationContext;
|
||||
import org.wso2.carbon.automation.engine.context.beans.User;
|
||||
import org.wso2.carbon.automation.engine.exceptions.AutomationFrameworkException;
|
||||
import org.wso2.carbon.automation.engine.frameworkutils.CodeCoverageUtils;
|
||||
import org.wso2.carbon.automation.engine.frameworkutils.FrameworkPathUtil;
|
||||
import org.wso2.carbon.automation.engine.frameworkutils.ReportGenerator;
|
||||
import org.wso2.carbon.automation.engine.frameworkutils.TestFrameworkUtils;
|
||||
import org.wso2.carbon.automation.extensions.servers.utils.ArchiveExtractor;
|
||||
import org.wso2.carbon.automation.extensions.servers.utils.ClientConnectionUtil;
|
||||
import org.wso2.carbon.automation.extensions.servers.utils.FileManipulator;
|
||||
import org.wso2.carbon.automation.extensions.servers.utils.ServerLogReader;
|
||||
|
||||
import javax.xml.xpath.XPathExpressionException;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.util.Iterator;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
|
||||
public class CarbonServerManagerExtension {
|
||||
private static final Log log = LogFactory.getLog(CarbonServerManagerExtension.class);
|
||||
private Process process;
|
||||
private String carbonHome;
|
||||
private AutomationContext automationContext;
|
||||
private ServerLogReader inputStreamHandler;
|
||||
private ServerLogReader errorStreamHandler;
|
||||
private boolean isCoverageEnable = false;
|
||||
private String coverageDumpFilePath;
|
||||
private int portOffset = 0;
|
||||
private static final String SERVER_SHUTDOWN_MESSAGE = "Halting JVM";
|
||||
private static final String SERVER_STARTUP_MESSAGE = "Mgt Console URL";
|
||||
private static final long DEFAULT_START_STOP_WAIT_MS = 400000L;
|
||||
private static final String CMD_ARG = "cmdArg";
|
||||
private static int defaultHttpPort = Integer.parseInt("9763");
|
||||
private static int defaultHttpsPort = Integer.parseInt("9443");
|
||||
|
||||
public CarbonServerManagerExtension(AutomationContext context) {
|
||||
this.automationContext = context;
|
||||
}
|
||||
|
||||
public synchronized void startServerUsingCarbonHome(String carbonHome, Map<String, String> commandMap) throws AutomationFrameworkException {
|
||||
if(this.process == null) {
|
||||
this.portOffset = this.checkPortAvailability(commandMap);
|
||||
Process tempProcess = null;
|
||||
|
||||
try {
|
||||
if(!commandMap.isEmpty() && this.getPortOffsetFromCommandMap(commandMap) == 0) {
|
||||
System.setProperty("carbon.home", carbonHome);
|
||||
}
|
||||
|
||||
File commandDir = new File(carbonHome);
|
||||
log.info("Starting carbon server............. ");
|
||||
String scriptName = TestFrameworkUtils.getStartupScriptFileName(carbonHome);
|
||||
String[] parameters = this.expandServerStartupCommandList(commandMap);
|
||||
String[] cmdArray;
|
||||
if(System.getProperty("os.name").toLowerCase().contains("windows")) {
|
||||
commandDir = new File(carbonHome + File.separator + "bin");
|
||||
cmdArray = new String[]{"cmd.exe", "/c", scriptName + ".bat"};
|
||||
cmdArray = this.mergePropertiesToCommandArray(parameters, cmdArray);
|
||||
tempProcess = Runtime.getRuntime().exec(cmdArray, (String[])null, commandDir);
|
||||
} else {
|
||||
cmdArray = new String[]{"sh", "bin/" + scriptName + ".sh"};
|
||||
cmdArray = this.mergePropertiesToCommandArray(parameters, cmdArray);
|
||||
tempProcess = Runtime.getRuntime().exec(cmdArray, (String[])null, commandDir);
|
||||
}
|
||||
|
||||
this.errorStreamHandler = new ServerLogReader("errorStream", tempProcess.getErrorStream());
|
||||
this.inputStreamHandler = new ServerLogReader("inputStream", tempProcess.getInputStream());
|
||||
this.inputStreamHandler.start();
|
||||
this.errorStreamHandler.start();
|
||||
Runtime.getRuntime().addShutdownHook(new Thread() {
|
||||
public void run() {
|
||||
try {
|
||||
CarbonServerManagerExtension.this.serverShutdown(CarbonServerManagerExtension.this.portOffset);
|
||||
} catch (Exception var2) {
|
||||
CarbonServerManagerExtension.log.error("Error while server shutdown ..", var2);
|
||||
}
|
||||
|
||||
}
|
||||
});
|
||||
ClientConnectionUtil.waitForPort(defaultHttpPort + this.portOffset, 300000L, false, (String)this.automationContext.getInstance().getHosts().get("default"));
|
||||
long time = System.currentTimeMillis() + 60000L;
|
||||
|
||||
while(true) {
|
||||
if(this.inputStreamHandler.getOutput().contains("Mgt Console URL") || System.currentTimeMillis() >= time) {
|
||||
int httpsPort = defaultHttpsPort + this.portOffset;
|
||||
String backendURL = this.automationContext.getContextUrls().getSecureServiceUrl().replaceAll("(:\\d+)", ":" + httpsPort);
|
||||
User superUser = this.automationContext.getSuperTenant().getTenantAdmin();
|
||||
ClientConnectionUtil.waitForLogin(backendURL, superUser);
|
||||
log.info("Server started successfully.");
|
||||
break;
|
||||
}
|
||||
}
|
||||
} catch (XPathExpressionException | IOException var13) {
|
||||
throw new IllegalStateException("Unable to start server", var13);
|
||||
}
|
||||
|
||||
this.process = tempProcess;
|
||||
}
|
||||
}
|
||||
|
||||
private int checkPortAvailability(Map<String, String> commandMap) throws AutomationFrameworkException {
|
||||
int portOffset = this.getPortOffsetFromCommandMap(commandMap);
|
||||
if(ClientConnectionUtil.isPortOpen(defaultHttpPort + portOffset)) {
|
||||
throw new AutomationFrameworkException("Unable to start carbon server on port " + (defaultHttpPort + portOffset) + " : Port already in use");
|
||||
} else if(ClientConnectionUtil.isPortOpen(defaultHttpsPort + portOffset)) {
|
||||
throw new AutomationFrameworkException("Unable to start carbon server on port " + (defaultHttpsPort + portOffset) + " : Port already in use");
|
||||
} else {
|
||||
return portOffset;
|
||||
}
|
||||
}
|
||||
|
||||
private String[] mergePropertiesToCommandArray(String[] parameters, String[] cmdArray) {
|
||||
if(parameters != null) {
|
||||
cmdArray = this.mergerArrays(cmdArray, parameters);
|
||||
}
|
||||
|
||||
return cmdArray;
|
||||
}
|
||||
|
||||
public synchronized String setUpCarbonHome(String carbonServerZipFile) throws IOException, AutomationFrameworkException {
|
||||
if(this.process != null) {
|
||||
return this.carbonHome;
|
||||
} else {
|
||||
int indexOfZip = carbonServerZipFile.lastIndexOf(".zip");
|
||||
if(indexOfZip == -1) {
|
||||
throw new IllegalArgumentException(carbonServerZipFile + " is not a zip file");
|
||||
} else {
|
||||
String fileSeparator = File.separator.equals("\\")?"\\":"/";
|
||||
if(fileSeparator.equals("\\")) {
|
||||
carbonServerZipFile = carbonServerZipFile.replace("/", "\\");
|
||||
}
|
||||
|
||||
String extractedCarbonDir = carbonServerZipFile.substring(carbonServerZipFile.lastIndexOf(fileSeparator) + 1, indexOfZip);
|
||||
FileManipulator.deleteDir(extractedCarbonDir);
|
||||
String extractDir = "carbontmp" + System.currentTimeMillis();
|
||||
String baseDir = System.getProperty("basedir", ".") + File.separator + "target";
|
||||
log.info("Extracting carbon zip file.. ");
|
||||
(new ArchiveExtractor()).extractFile(carbonServerZipFile, baseDir + File.separator + extractDir);
|
||||
this.carbonHome = (new File(baseDir)).getAbsolutePath() + File.separator + extractDir + File.separator + extractedCarbonDir;
|
||||
|
||||
try {
|
||||
this.isCoverageEnable = Boolean.parseBoolean(this.automationContext.getConfigurationValue("//coverage"));
|
||||
} catch (XPathExpressionException var8) {
|
||||
throw new AutomationFrameworkException("Coverage configuration not found in automation.xml", var8);
|
||||
}
|
||||
|
||||
if(this.isCoverageEnable) {
|
||||
this.instrumentForCoverage();
|
||||
}
|
||||
|
||||
return this.carbonHome;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void serverShutdown(int portOffset) throws AutomationFrameworkException {
|
||||
if(this.process != null) {
|
||||
log.info("Shutting down server..");
|
||||
if(ClientConnectionUtil.isPortOpen(Integer.parseInt("9443") + portOffset)) {
|
||||
int httpsPort = defaultHttpsPort + portOffset;
|
||||
String url = null;
|
||||
|
||||
try {
|
||||
url = this.automationContext.getContextUrls().getBackEndUrl();
|
||||
} catch (XPathExpressionException var10) {
|
||||
throw new AutomationFrameworkException("Get context failed", var10);
|
||||
}
|
||||
|
||||
String backendURL = url.replaceAll("(:\\d+)", ":" + httpsPort);
|
||||
|
||||
try {
|
||||
ClientConnectionUtil.sendForcefulShutDownRequest(backendURL, this.automationContext.getSuperTenant().getContextUser().getUserName(), this.automationContext.getSuperTenant().getContextUser().getPassword());
|
||||
} catch (AutomationFrameworkException var8) {
|
||||
throw new AutomationFrameworkException("Get context failed", var8);
|
||||
} catch (XPathExpressionException var9) {
|
||||
throw new AutomationFrameworkException("Get context failed", var9);
|
||||
}
|
||||
|
||||
long time = System.currentTimeMillis() + 300000L;
|
||||
|
||||
while(!this.inputStreamHandler.getOutput().contains("Halting JVM") && System.currentTimeMillis() < time) {
|
||||
;
|
||||
}
|
||||
|
||||
log.info("Server stopped successfully...");
|
||||
}
|
||||
|
||||
this.inputStreamHandler.stop();
|
||||
this.errorStreamHandler.stop();
|
||||
this.process.destroy();
|
||||
this.process = null;
|
||||
if(this.isCoverageEnable) {
|
||||
try {
|
||||
log.info("Generating Jacoco code coverage...");
|
||||
this.generateCoverageReport(new File(this.carbonHome + File.separator + "repository" + File.separator + "components" + File.separator + "plugins" + File.separator));
|
||||
} catch (IOException var7) {
|
||||
log.error("Failed to generate code coverage ", var7);
|
||||
throw new AutomationFrameworkException("Failed to generate code coverage ", var7);
|
||||
}
|
||||
}
|
||||
|
||||
if(portOffset == 0) {
|
||||
System.clearProperty("carbon.home");
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private void generateCoverageReport(File classesDir) throws IOException, AutomationFrameworkException {
|
||||
CodeCoverageUtils.executeMerge(FrameworkPathUtil.getJacocoCoverageHome(), FrameworkPathUtil.getCoverageMergeFilePath());
|
||||
ReportGenerator reportGenerator = new ReportGenerator(new File(FrameworkPathUtil.getCoverageMergeFilePath()), classesDir, new File(CodeCoverageUtils.getJacocoReportDirectory()), (File)null);
|
||||
reportGenerator.create();
|
||||
log.info("Jacoco coverage dump file path : " + FrameworkPathUtil.getCoverageDumpFilePath());
|
||||
log.info("Jacoco class file path : " + classesDir);
|
||||
log.info("Jacoco coverage HTML report path : " + CodeCoverageUtils.getJacocoReportDirectory() + File.separator + "index.html");
|
||||
}
|
||||
|
||||
public synchronized void restartGracefully() throws AutomationFrameworkException {
|
||||
try {
|
||||
int httpsPort = defaultHttpsPort + this.portOffset;
|
||||
String backendURL = this.automationContext.getContextUrls().getSecureServiceUrl().replaceAll("(:\\d+)", ":" + httpsPort);
|
||||
User superUser = this.automationContext.getSuperTenant().getTenantAdmin();
|
||||
ClientConnectionUtil.sendGraceFullRestartRequest(backendURL, superUser.getUserName(), superUser.getPassword());
|
||||
} catch (XPathExpressionException var5) {
|
||||
throw new AutomationFrameworkException("restart failed", var5);
|
||||
}
|
||||
|
||||
long time = System.currentTimeMillis() + 300000L;
|
||||
|
||||
while(!this.inputStreamHandler.getOutput().contains("Halting JVM") && System.currentTimeMillis() < time) {
|
||||
;
|
||||
}
|
||||
|
||||
time = System.currentTimeMillis();
|
||||
|
||||
while(System.currentTimeMillis() < time + 5000L) {
|
||||
;
|
||||
}
|
||||
|
||||
try {
|
||||
ClientConnectionUtil.waitForPort(Integer.parseInt((String)this.automationContext.getInstance().getPorts().get("https")), (String)this.automationContext.getInstance().getHosts().get("default"));
|
||||
ClientConnectionUtil.waitForLogin(this.automationContext);
|
||||
} catch (XPathExpressionException var4) {
|
||||
throw new AutomationFrameworkException("Connection attempt to carbon server failed", var4);
|
||||
}
|
||||
}
|
||||
|
||||
private String[] expandServerStartupCommandList(Map<String, String> commandMap) {
|
||||
if(commandMap != null && commandMap.size() != 0) {
|
||||
String[] cmdParaArray = null;
|
||||
String cmdArg = null;
|
||||
if(commandMap.containsKey("cmdArg")) {
|
||||
cmdArg = (String)commandMap.get("cmdArg");
|
||||
cmdParaArray = cmdArg.trim().split("\\s+");
|
||||
commandMap.remove("cmdArg");
|
||||
}
|
||||
|
||||
String[] parameterArray = new String[commandMap.size()];
|
||||
int arrayIndex = 0;
|
||||
Set<Map.Entry<String, String>> entries = commandMap.entrySet();
|
||||
|
||||
String parameter;
|
||||
for(Iterator i$ = entries.iterator(); i$.hasNext(); parameterArray[arrayIndex++] = parameter) {
|
||||
Map.Entry<String, String> entry = (Map.Entry)i$.next();
|
||||
String key = (String)entry.getKey();
|
||||
String value = (String)entry.getValue();
|
||||
if(value != null && !value.isEmpty()) {
|
||||
parameter = key + "=" + value;
|
||||
} else {
|
||||
parameter = key;
|
||||
}
|
||||
}
|
||||
|
||||
if(cmdArg != null) {
|
||||
commandMap.put("cmdArg", cmdArg);
|
||||
}
|
||||
|
||||
if(cmdParaArray != null && cmdParaArray.length != 0) {
|
||||
return (String[]) ArrayUtils.addAll(parameterArray, cmdParaArray);
|
||||
} else {
|
||||
return parameterArray;
|
||||
}
|
||||
} else {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private int getPortOffsetFromCommandMap(Map<String, String> commandMap) {
|
||||
return commandMap.containsKey("-DportOffset")?Integer.parseInt((String)commandMap.get("-DportOffset")):0;
|
||||
}
|
||||
|
||||
private String[] mergerArrays(String[] array1, String[] array2) {
|
||||
return (String[])ArrayUtils.addAll(array1, array2);
|
||||
}
|
||||
|
||||
private void insertJacocoAgentToShellScript(String scriptName) throws IOException {
|
||||
String jacocoAgentFile = CodeCoverageUtils.getJacocoAgentJarLocation();
|
||||
this.coverageDumpFilePath = FrameworkPathUtil.getCoverageDumpFilePath();
|
||||
CodeCoverageUtils.insertStringToFile(new File(this.carbonHome + File.separator + "bin" + File.separator + scriptName + ".sh"), new File(this.carbonHome + File.separator + "tmp" + File.separator + scriptName + ".sh"), "-Dwso2.server.standalone=true", "-javaagent:" + jacocoAgentFile + "=destfile=" + this.coverageDumpFilePath + "" + ",append=true,includes=" + CodeCoverageUtils.getInclusionJarsPattern(":") + " \\");
|
||||
}
|
||||
|
||||
private void insertJacocoAgentToBatScript(String scriptName) throws IOException {
|
||||
String jacocoAgentFile = CodeCoverageUtils.getJacocoAgentJarLocation();
|
||||
this.coverageDumpFilePath = FrameworkPathUtil.getCoverageDumpFilePath();
|
||||
CodeCoverageUtils.insertJacocoAgentToStartupBat(new File(this.carbonHome + File.separator + "bin" + File.separator + scriptName + ".bat"), new File(this.carbonHome + File.separator + "tmp" + File.separator + scriptName + ".bat"), "-Dcatalina.base", "-javaagent:" + jacocoAgentFile + "=destfile=" + this.coverageDumpFilePath + "" + ",append=true,includes=" + CodeCoverageUtils.getInclusionJarsPattern(":"));
|
||||
}
|
||||
|
||||
private void instrumentForCoverage() throws IOException, AutomationFrameworkException {
|
||||
String scriptName = TestFrameworkUtils.getStartupScriptFileName(this.carbonHome);
|
||||
if(System.getProperty("os.name").toLowerCase().contains("windows")) {
|
||||
this.insertJacocoAgentToBatScript(scriptName);
|
||||
if(log.isDebugEnabled()) {
|
||||
log.debug("Included files " + CodeCoverageUtils.getInclusionJarsPattern(":"));
|
||||
log.debug("Excluded files " + CodeCoverageUtils.getExclusionJarsPattern(":"));
|
||||
}
|
||||
} else {
|
||||
this.insertJacocoAgentToShellScript(scriptName);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue