/*
* Licensed under the GPL License. You may not use this file except in
* compliance with the License. You may obtain a copy of the License at
*
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*
* THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED
* WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
* MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
*/
package com.googlecode.psiprobe.controllers;
import com.googlecode.psiprobe.beans.ContainerListenerBean;
import com.googlecode.psiprobe.model.ApplicationResource;
import com.googlecode.psiprobe.model.DataSourceInfo;
import com.googlecode.psiprobe.model.TomcatTestReport;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.catalina.Context;
import org.springframework.web.servlet.ModelAndView;
/**
* "Quick check" controller.
*
* @author Vlad Ilyushchenko
*/
public class TomcatAvailabilityController extends TomcatContainerController {
private ContainerListenerBean containerListenerBean;
public ContainerListenerBean getContainerListenerBean() {
return containerListenerBean;
}
public void setContainerListenerBean(ContainerListenerBean containerListenerBean) {
this.containerListenerBean = containerListenerBean;
}
public ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response) throws Exception {
long start = System.currentTimeMillis();
TomcatTestReport tomcatTestReport = new TomcatTestReport();
//
// check datasource status
//
tomcatTestReport.setDatasourceUsageScore(0);
boolean allContextsAvailable = true;
if (getContainerWrapper().getResourceResolver().supportsPrivateResources()) {
for (Iterator it = getContainerWrapper().getTomcatContainer().findContexts().iterator(); it.hasNext();) {
//
// make sure we skip ROOT application
//
Context appContext = (Context) it.next();
allContextsAvailable = allContextsAvailable && appContext.getAvailable();
List applicationResources = getContainerWrapper().getResourceResolver().getApplicationResources(appContext);
for (Iterator it2 = applicationResources.iterator(); it2.hasNext();) {
ApplicationResource appResource = (ApplicationResource) it2.next();
DataSourceInfo dsi = appResource.getDataSourceInfo();
if (dsi != null && dsi.getBusyScore() > tomcatTestReport.getDatasourceUsageScore()) {
tomcatTestReport.setContextName(appContext.getName());
tomcatTestReport.setDatasourceUsageScore(dsi.getBusyScore());
tomcatTestReport.setDataSourceName(appResource.getName());
}
}
}
tomcatTestReport.setWebappAvailabilityTest(allContextsAvailable ?
TomcatTestReport.TEST_PASSED : TomcatTestReport.TEST_FAILED);
} else {
List l = getContainerWrapper().getResourceResolver().getApplicationResources();
for (int i = 0; i < l.size(); i++) {
ApplicationResource resource = (ApplicationResource)l.get(i);
DataSourceInfo dsi = resource.getDataSourceInfo();
if (dsi != null && dsi.getBusyScore() > tomcatTestReport.getDatasourceUsageScore()) {
tomcatTestReport.setDatasourceUsageScore(dsi.getBusyScore());
tomcatTestReport.setDataSourceName(resource.getName());
}
}
}
tomcatTestReport.setDatasourceTest(TomcatTestReport.TEST_PASSED);
//
// try to allocate some memory
//
String word = "hello";
int count = tomcatTestReport.getDefaultMemorySize() / word.length();
try {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
for (; count > 0; count --) {
bos.write(word.getBytes());
}
tomcatTestReport.setMemoryTest(TomcatTestReport.TEST_PASSED);
} catch (Throwable e) {
tomcatTestReport.setMemoryTest(TomcatTestReport.TEST_FAILED);
//
// make sure we always re-throw ThreadDeath
//
if (e instanceof ThreadDeath) {
throw (ThreadDeath) e;
}
}
//
// try to open some files
//
File tmpDir = new File(System.getProperty("java.io.tmpdir"));
int fileCount = tomcatTestReport.getDefaultFileCount();
List files = new ArrayList();
List fileStreams = new ArrayList();
try {
for (; fileCount > 0; fileCount --) {
File file = new File(tmpDir, "tctest_" + fileCount);
FileOutputStream fos = new FileOutputStream(file);
files.add(file);
fileStreams.add(fos);
fos.write("this is a test".getBytes());
}
tomcatTestReport.setFileTest(TomcatTestReport.TEST_PASSED);
} catch (IOException e) {
tomcatTestReport.setFileTest(TomcatTestReport.TEST_FAILED);
} finally {
for (int i = 0; i < fileStreams.size(); i++) {
try {
((FileOutputStream) fileStreams.get(i)).close();
} catch (IOException e) {
}
}
for (int i = 0; i < files.size(); i++) {
((File) files.get(i)).delete();
}
}
tomcatTestReport.setTestDuration(System.currentTimeMillis() - start);
long maxServiceTime = 0;
//
// check the maximum execution time
//
/*
List pools = containerListenerBean.getThreadPools();
for (int iPool = 0; iPool < pools.size(); iPool++) {
ThreadPool threadPool = (ThreadPool) pools.get(iPool);
List threads = threadPool.getRequestProcessors();
for (int iThread = 0; iThread < threads.size(); iThread++) {
RequestProcessor rp = (RequestProcessor) threads.get(iThread);
if (rp.getStage() == 3) {
//
// the request processor is in SERVICE state
//
maxServiceTime = Math.max(maxServiceTime, rp.getProcessingTime());
}
}
}
*/
tomcatTestReport.setMaxServiceTime(maxServiceTime);
return new ModelAndView(getViewName(), "testReport", tomcatTestReport);
}
}