Save This Page
Home » jakarta-jmeter-2.3.4_src » org.apache.jmeter.functions » [javadoc | source]
    1   /*
    2    * Licensed to the Apache Software Foundation (ASF) under one or more
    3    * contributor license agreements.  See the NOTICE file distributed with
    4    * this work for additional information regarding copyright ownership.
    5    * The ASF licenses this file to You under the Apache License, Version 2.0
    6    * (the "License"); you may not use this file except in compliance with
    7    * the License.  You may obtain a copy of the License at
    8    *
    9    *   http://www.apache.org/licenses/LICENSE-2.0
   10    *
   11    * Unless required by applicable law or agreed to in writing, software
   12    * distributed under the License is distributed on an "AS IS" BASIS,
   13    * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   14    * See the License for the specific language governing permissions and
   15    * limitations under the License.
   16    *
   17    */
   18   
   19   package org.apache.jmeter.functions;
   20   
   21   import java.util.Collection;
   22   import java.util.LinkedList;
   23   import java.util.List;
   24   
   25   import org.apache.jmeter.engine.util.CompoundVariable;
   26   import org.apache.jmeter.samplers.SampleResult;
   27   import org.apache.jmeter.samplers.Sampler;
   28   import org.apache.jmeter.threads.JMeterVariables;
   29   import org.apache.jmeter.util.JMeterUtils;
   30   
   31   public class IterationCounter extends AbstractFunction {
   32   
   33       private static final List desc = new LinkedList();
   34   
   35       private static final String KEY = "__counter"; //$NON-NLS-1$
   36   
   37       private ThreadLocal perThreadInt;
   38   
   39       private Object[] variables;
   40   
   41       private int globalCounter;//MAXINT = 2,147,483,647
   42   
   43       private void init(){
   44          synchronized(this){
   45              globalCounter=0;
   46          }
   47          perThreadInt = new ThreadLocal(){
   48               protected synchronized Object initialValue() {
   49                   return new Integer(0);
   50               }
   51           };
   52       }
   53   
   54       static {
   55           desc.add(JMeterUtils.getResString("iteration_counter_arg_1")); //$NON-NLS-1$
   56           desc.add(JMeterUtils.getResString("function_name_paropt")); //$NON-NLS-1$
   57       }
   58   
   59       public IterationCounter() {
   60           init();
   61       }
   62   
   63       /*
   64        * (non-Javadoc)
   65        *
   66        * @see org.apache.jmeter.functions.Function#execute(SampleResult, Sampler)
   67        */
   68       public synchronized String execute(SampleResult previousResult, Sampler currentSampler)
   69               throws InvalidVariableException {
   70   
   71           new Integer(1);
   72           globalCounter++;
   73   
   74           JMeterVariables vars = getVariables();
   75   
   76           boolean perThread = Boolean.valueOf(((CompoundVariable) variables[0]).execute()).booleanValue();
   77   
   78           String varName = ""; //$NON-NLS-1$
   79           if (variables.length >=2) {// Ensure variable has been provided
   80               varName = ((CompoundVariable) variables[1]).execute().trim();
   81           }
   82   
   83           String counterString = ""; //$NON-NLS-1$
   84   
   85           if (perThread) {
   86               int threadCounter;
   87               threadCounter = ((Integer) perThreadInt.get()).intValue() + 1;
   88               perThreadInt.set(new Integer(threadCounter));
   89               counterString = String.valueOf(threadCounter);
   90           } else {
   91               counterString = String.valueOf(globalCounter);
   92           }
   93   
   94           // vars will be null on Test Plan
   95           if (vars != null && varName.length() > 0) {
   96               vars.put(varName, counterString);
   97           }
   98           return counterString;
   99       }
  100   
  101       /*
  102        * (non-Javadoc)
  103        *
  104        * @see org.apache.jmeter.functions.Function#setParameters(Collection)
  105        */
  106       public synchronized void setParameters(Collection parameters) throws InvalidVariableException {
  107           checkParameterCount(parameters, 1, 2);
  108           variables = parameters.toArray();
  109       }
  110   
  111       /*
  112        * (non-Javadoc)
  113        *
  114        * @see org.apache.jmeter.functions.Function#getReferenceKey()
  115        */
  116       public String getReferenceKey() {
  117           return KEY;
  118       }
  119   
  120       /*
  121        * (non-Javadoc)
  122        *
  123        * @see org.apache.jmeter.functions.Function#getArgumentDesc()
  124        */
  125       public List getArgumentDesc() {
  126           return desc;
  127       }
  128   }

Save This Page
Home » jakarta-jmeter-2.3.4_src » org.apache.jmeter.functions » [javadoc | source]