// $Header: /home/cvs/jakarta-jmeter/src/monitor/model/org/apache/jmeter/monitor/parser/MonitorHandler.java,v 1.5 2004/03/20 20:36:38 sebb Exp $
/*
 * Copyright 2004 The Apache Software Foundation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.jmeter.monitor.parser;

//import java.util.List;
import java.util.Stack;

import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

import org.apache.jmeter.monitor.model.ObjectFactory;
import org.apache.jmeter.monitor.model.Connector;
import org.apache.jmeter.monitor.model.Jvm;
import org.apache.jmeter.monitor.model.Memory;
import org.apache.jmeter.monitor.model.RequestInfo;
import org.apache.jmeter.monitor.model.Status;
import org.apache.jmeter.monitor.model.StatusImpl;
import org.apache.jmeter.monitor.model.ThreadInfo;
import org.apache.jmeter.monitor.model.Worker;
import org.apache.jmeter.monitor.model.Workers;
import org.apache.jmeter.monitor.model.WorkersImpl;

public class MonitorHandler extends DefaultHandler
{
	//private boolean startDoc = false;
	//private boolean endDoc = false;
	private Stack stacktree = new Stack();

	private ObjectFactory factory = null;	
	private Status status = null;
	private Jvm jvm = null;
	private Memory memory = null;
	private Connector connector = null;
	private ThreadInfo threadinfo = null;
	private RequestInfo requestinfo = null;
	private Worker worker = null;
	private Workers workers = null;
	//private List workerslist = null;
	
    /**
     * 
     */
    public MonitorHandler()
    {
        super();
    }

	/**
	 * Set the ObjectFactory used to create new 
	 * @param factory
	 */
	public void setObjectFactory(ObjectFactory factory){
		this.factory = factory;
	}
	
	public void startDocument ()
	throws SAXException
	{
		//this.startDoc = true;
	}
	
	public void endDocument ()
	throws SAXException
	{
		//this.startDoc = false;
		//this.endDoc = true;
	}

	/**
	 * Receive notification of the start of an element.
	 *
	 * 
By default, do nothing.  Application writers may override this
	 * method in a subclass to take specific actions at the start of
	 * each element (such as allocating a new tree node or writing
	 * output to a file).
	 *
	 * @param uri
	 * @param localName The element type name.
	 * @param qName
	 * @param attributes The specified or defaulted attributes.
	 * @exception org.xml.sax.SAXException Any SAX exception, possibly
	 *            wrapping another exception.
	 * @see org.xml.sax.ContentHandler#startElement
	 */
	public void startElement (String uri, String localName,
				  String qName, Attributes attributes)
	throws SAXException
	{
		if (qName.equals(Constants.STATUS)){
			status = factory.createStatus();
			stacktree.push(status);
		} else if (qName.equals(Constants.JVM)){
			jvm = factory.createJvm();
			if (stacktree.peek() instanceof Status){
				status.setJvm(jvm);
				stacktree.push(jvm);
			}
		} else if (qName.equals(Constants.MEMORY)){
			memory = factory.createMemory();
			if (stacktree.peek() instanceof Jvm){
				stacktree.push(memory);
				if (attributes != null){
					for (int idx=0; idx < attributes.getLength(); idx++){
						String attr = attributes.getQName(idx);
						if (attr.equals(Constants.MEMORY_FREE)){
							memory.
								setFree(parseLong(attributes.getValue(idx)));
						} else if (attr.equals(Constants.MEMORY_TOTAL)){
							memory.
								setTotal(parseLong(attributes.getValue(idx)));
						} else if (attr.equals(Constants.MEMORY_MAX)){
							memory.
								setMax(parseLong(attributes.getValue(idx)));
						}
					}
				}
				jvm.setMemory(memory);
			}
		} else if (qName.equals(Constants.CONNECTOR)){
			connector = factory.createConnector();
			if (stacktree.peek() instanceof Status ||
			stacktree.peek() instanceof Connector){
				((StatusImpl)status).addConnector(connector);
				stacktree.push(connector);
				if (attributes != null){
					for (int idx=0; idx < attributes.getLength(); idx++){
						String attr = attributes.getQName(idx);
						if (attr.equals(Constants.ATTRIBUTE_NAME)){
							connector.setName(attributes.getValue(idx));
						}
					}
				}
			}
		} else if (qName.equals(Constants.THREADINFO)){
			threadinfo = factory.createThreadInfo();
			if (stacktree.peek() instanceof Connector){
				stacktree.push(threadinfo);
				connector.setThreadInfo(threadinfo);
				if (attributes != null){
					for (int idx=0; idx < attributes.getLength(); idx++){
						String attr = attributes.getQName(idx);
						if (attr.equals(Constants.MAXTHREADS)){
							threadinfo.setMaxThreads(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.MINSPARETHREADS)){
							threadinfo.setMinSpareThreads(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.MAXSPARETHREADS)){
							threadinfo.setMaxSpareThreads(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.CURRENTTHREADCOUNT)){
							threadinfo.setCurrentThreadCount(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.CURRENTBUSYTHREADS)){
							threadinfo.setCurrentThreadsBusy(
								parseInt(attributes.getValue(idx)));
						}
					}
				}
			}
		} else if (qName.equals(Constants.REQUESTINFO)){
			requestinfo = factory.createRequestInfo();
			if (stacktree.peek() instanceof Connector){
				stacktree.push(requestinfo);
				connector.setRequestInfo(requestinfo);
				if (attributes != null){
					for (int idx=0; idx < attributes.getLength(); idx++){
						String attr = attributes.getQName(idx);
						if (attr.equals(Constants.MAXTIME)){
							requestinfo.setMaxTime(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.PROCESSINGTIME)){
							requestinfo.setProcessingTime(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.REQUESTCOUNT)){
							requestinfo.setRequestCount(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.ERRORCOUNT)){
							requestinfo.setErrorCount(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.BYTESRECEIVED)){
							requestinfo.setBytesReceived(
								parseLong(attributes.getValue(idx)));
						} else if (attr.equals(Constants.BYTESSENT)){
							requestinfo.setBytesSent(
								parseLong(attributes.getValue(idx)));
						}
					}
				}
			}
		} else if (qName.equals(Constants.WORKERS)){
			workers = factory.createWorkers();
			if (stacktree.peek() instanceof Connector){
				connector.setWorkers(workers);
				stacktree.push(workers);
			}
		} else if (qName.equals(Constants.WORKER)){
			worker = factory.createWorker();
			if (stacktree.peek() instanceof Workers ||
			stacktree.peek() instanceof Worker){
				stacktree.push(worker);
				((WorkersImpl)workers).addWorker(worker);
				if (attributes != null){
					for (int idx=0; idx < attributes.getLength(); idx++){
						String attr = attributes.getQName(idx);
						if (attr.equals(Constants.STAGE)){
							worker.setStage(attributes.getValue(idx));
						} else if (attr.equals(Constants.REQUESTPROCESSINGTIME)){
							worker.setRequestProcessingTime(
								parseInt(attributes.getValue(idx)));
						} else if (attr.equals(Constants.REQUESTBYTESSENT)){
							worker.setRequestBytesSent(
								parseLong(attributes.getValue(idx)));
						} else if (attr.equals(Constants.REQUESTBYTESRECEIVED)){
							worker.setRequestBytesReceived(
								parseLong(attributes.getValue(idx)));
						} else if (attr.equals(Constants.REMOTEADDR)){
							worker.setRemoteAddr(attributes.getValue(idx));
						} else if (attr.equals(Constants.VIRTUALHOST)){
							worker.setVirtualHost(attributes.getValue(idx));
						} else if (attr.equals(Constants.METHOD)){
							worker.setMethod(attributes.getValue(idx));
						} else if (attr.equals(Constants.CURRENTURI)){
							worker.setCurrentUri(attributes.getValue(idx));
						} else if (attr.equals(Constants.CURRENTQUERYSTRING)){
							worker.setCurrentQueryString(
								attributes.getValue(idx));
						} else if (attr.equals(Constants.PROTOCOL)){
							worker.setProtocol(attributes.getValue(idx));
						}
					}
				}
			}
		}
	}
    
    
	/**
	 * Receive notification of the end of an element.
	 *
	 * 
By default, do nothing.  Application writers may override this
	 * method in a subclass to take specific actions at the end of
	 * each element (such as finalising a tree node or writing
	 * output to a file).
	 *
	 * @param uri
	 * @param localName The element type name.
	 * @param qName The specified or defaulted attributes.
	 * @exception org.xml.sax.SAXException Any SAX exception, possibly
	 *            wrapping another exception.
	 * @see org.xml.sax.ContentHandler#endElement
	 */
	public void endElement (String uri, String localName, String qName)
	throws SAXException
	{
		if (qName.equals(Constants.STATUS)){
			if (stacktree.peek() instanceof Status){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.JVM)){
			if (stacktree.peek() instanceof Jvm){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.MEMORY)){
			if (stacktree.peek() instanceof Memory){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.CONNECTOR)){
			if (stacktree.peek() instanceof Connector ||
			stacktree.peek() instanceof Connector){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.THREADINFO)){
			if (stacktree.peek() instanceof ThreadInfo){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.REQUESTINFO)){
			if (stacktree.peek() instanceof RequestInfo){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.WORKERS)){
			if (stacktree.peek() instanceof Workers){
				stacktree.pop();
			}
		} else if (qName.equals(Constants.WORKER)){
			if (stacktree.peek() instanceof Worker ||
			stacktree.peek() instanceof Worker){
				stacktree.pop();
			}
		}
	}

	/**
	 * Receive notification of character data inside an element.
	 *
	 * 
By default, do nothing.  Application writers may override this
	 * method to take specific actions for each chunk of character data
	 * (such as adding the data to a node or buffer, or printing it to
	 * a file).
	 *
	 * @param ch The characters.
	 * @param start The start position in the character array.
	 * @param length The number of characters to use from the
	 *               character array.
	 * @exception org.xml.sax.SAXException Any SAX exception, possibly
	 *            wrapping another exception.
	 * @see org.xml.sax.ContentHandler#characters
	 */
	public void characters (char ch[], int start, int length)
	throws SAXException
	{
	}

	/**
	 * Convienance method for parsing long. If the string
	 * was not a number, the method returns zero.
	 * @param data
	 * @return the value as a long
	 */    
    public long parseLong(String data){
    	long val = 0;
    	if (data.length() > 0){
			try {
				val = Long.parseLong(data);
			} catch (NumberFormatException e){
				val = 0;
			}
    	}
    	return val;
    }
    
    /**
     * Convienance method for parsing integers. 
     * @param data
     * @return the value as an integer
     */
    public int parseInt(String data){
    	int val = 0;
    	if (data.length() > 0){
			try {
				val = Integer.parseInt(data);
			} catch (NumberFormatException e){
				val = 0;
			}
    	}
    	return val;
    }
    
    /**
     * method returns the status object.
     * @return the status
     */
    public Status getContents(){
    	return this.status;
    }
}