cvs commit: jakarta-tomcat/src/share/org/apache/tomcat/request AccountingInterceptor.java

[co...@locus.apache.org]

costin      00/04/17 14:03:37

  Added:       src/share/org/apache/tomcat/request
                        AccountingInterceptor.java
  Log:
  Accounting intercetor - will time the request and create a trace file.
  
  Revision  Changes    Path
  1.1                  jakarta-tomcat/src/share/org/apache/tomcat/request/AccountingInterceptor.java
  
  Index: AccountingInterceptor.java
  ===================================================================
  /*
   * ====================================================================
   *
   * The Apache Software License, Version 1.1
   *
   * Copyright (c) 1999 The Apache Software Foundation.  All rights
   * reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions
   * are met:
   *
   * 1. Redistributions of source code must retain the above copyright
   *    notice, this list of conditions and the following disclaimer.
   *
   * 2. Redistributions in binary form must reproduce the above copyright
   *    notice, this list of conditions and the following disclaimer in
   *    the documentation and/or other materials provided with the
   *    distribution.
   *
   * 3. The end-user documentation included with the redistribution, if
   *    any, must include the following acknowlegement:
   *       "This product includes software developed by the
   *        Apache Software Foundation (http://www.apache.org/)."
   *    Alternately, this acknowlegement may appear in the software itself,
   *    if and wherever such third-party acknowlegements normally appear.
   *
   * 4. The names "The Jakarta Project", "Tomcat", and "Apache Software
   *    Foundation" must not be used to endorse or promote products derived
   *    from this software without prior written permission. For written
   *    permission, please contact apache@apache.org.
   *
   * 5. Products derived from this software may not be called "Apache"
   *    nor may "Apache" appear in their names without prior written
   *    permission of the Apache Group.
   *
   * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
   * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
   * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
   * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
   * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
   * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
   * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
   * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
   * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   * SUCH DAMAGE.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   * [Additional notices, if required by prior licensing conditions]
   *
   */
  
  
  package org.apache.tomcat.request;
  
  import org.apache.tomcat.core.*;
  import org.apache.tomcat.util.*;
  import java.io.*;
  import java.net.*;
  import java.util.*;
  import javax.servlet.http.*;
  
  /**
   * Time counting. 
   *
   */
  public class AccountingInterceptor extends  BaseInterceptor implements RequestInterceptor {
      ContextManager cm;
      int debug=0;
      boolean acc=true;
      String trace="trace.log";
      BufferedOutputStream logF=null;
  
      public AccountingInterceptor() {
      }
  
      public void setDebug( int i ) {
  	debug=i;
      }
  
      public void setTrace( String file ) {
  	trace=file;
      }
      
      public void setContextManager( ContextManager cm ) {
  	this.cm=cm;
      }
  
      /** Called when the ContextManger is started
       */
      public void engineInit(ContextManager cm) throws TomcatException {
  	try {
  	    logF=new BufferedOutputStream( new FileOutputStream( trace ));
  	} catch(IOException ex ) {
  	    ex.printStackTrace();
  	}
      }
  
  
      public int requestMap(Request request ) {
  	if( acc && request instanceof RequestImpl ) {
  	    ((RequestImpl)request).setAccount( RequestImpl.ACC_PRE_RMAP, System.currentTimeMillis() );
  	}
  	return 0;
      }
  
      public int contextMap( Request request ) {
  	if( acc && request instanceof RequestImpl ) {
  	    ((RequestImpl)request).setAccount( RequestImpl.ACC_PRE_CMAP, System.currentTimeMillis() );
  	}
  	return 0;
      }
  
      public int authenticate(Request request, Response response) {
  	if( acc && request instanceof RequestImpl ) {
  	    ((RequestImpl)request).setAccount( RequestImpl.ACC_POST_MAP, System.currentTimeMillis() );
  	}
  	return 0;
      }
  
      public int authorize(Request request, Response response) {
  	return 0;
      }
  
  
      public int preService(Request request, Response response) {
  	if( acc && request instanceof RequestImpl ) {
  	    ((RequestImpl)request).setAccount( RequestImpl.ACC_PRE_SERVICE, System.currentTimeMillis() );
  	}
  	return 0;
      }
  
      public int beforeBody( Request rrequest, Response response ) {
  	return 0;
      }
  
      public int beforeCommit( Request request, Response response) {
  	return 0;
      }
  
  
      public int afterBody( Request request, Response response) {
  	return 0;
      }
  
      public int postService(Request request, Response response) {
  	if( acc && request instanceof RequestImpl ) {
  	    ((RequestImpl)request).setAccount( RequestImpl.ACC_POST_SERVICE, System.currentTimeMillis() );
  
  	    long t1=((RequestImpl)request).getAccount( RequestImpl.ACC_PRE_CMAP );
  	    long t2=((RequestImpl)request).getAccount( RequestImpl.ACC_PRE_RMAP );
  	    long t3=((RequestImpl)request).getAccount( RequestImpl.ACC_POST_MAP );
  	    long t4=((RequestImpl)request).getAccount( RequestImpl.ACC_PRE_SERVICE );
  	    long t5=((RequestImpl)request).getAccount( RequestImpl.ACC_POST_SERVICE );
  
  	    long t21=t2-t1;
  	    long t31=t3-t1;
  	    long t54=t5-t4;
  	    long t41=t4-t1;
  
  	    long tout=((RequestImpl)request).getAccount( RequestImpl.ACC_OUT_COUNT );
  	    StringBuffer sb=new StringBuffer();
  	    // ContextMap, Map, Service, Pre-Service-Overhead
  	    sb.append(t21).append(",");
  	    sb.append(t31).append(",");
  	    sb.append(t54).append(",");
  	    sb.append(tout).append(",");
  	    sb.append(t41).append("\n");
  	    ct++;
  	    try {
  		if( logF!=null ) logF.write(sb.toString().getBytes());
  		if( (ct % 64) == 0 ) logF.flush();
  	    } catch( IOException ex ) {
  		ex.printStackTrace();
  	    }
  	}
  
  	
  	return 0;
      }
  
      static int ct=0;
      
      
  }