testcompiler.java

测试工具

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.threads;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.Set;

import org.apache.jmeter.assertions.Assertion;
import org.apache.jmeter.config.ConfigTestElement;
import org.apache.jmeter.control.Controller;
import org.apache.jmeter.control.TransactionController;
import org.apache.jmeter.control.TransactionSampler;
import org.apache.jmeter.engine.event.LoopIterationListener;
import org.apache.jmeter.processor.PostProcessor;
import org.apache.jmeter.processor.PreProcessor;
import org.apache.jmeter.samplers.SampleListener;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.testbeans.TestBeanHelper;
import org.apache.jmeter.testelement.AbstractTestElement;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.timers.Timer;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.HashTreeTraverser;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

public class TestCompiler implements HashTreeTraverser {
	private static final Logger log = LoggingManager.getLoggerForClass();

	private LinkedList stack = new LinkedList();

	private Map samplerConfigMap = new HashMap();

	private Map transactionControllerConfigMap = new HashMap();
	
	private HashTree testTree;

	/*
	 * This set keeps track of which ObjectPairs have been seen Its purpose is
	 * not entirely clear (please document if you know!) but it is needed,..
	 */
	private static Set pairing = new HashSet();

	//List loopIterListeners = new ArrayList();

	public TestCompiler(HashTree testTree, JMeterVariables vars) {
		this.testTree = testTree;
	}

	/**
	 * Clears the pairing Set Called by StandardJmeterEngine at the start of a
	 * test run.
	 */
	public static void initialize() {
		// synch is probably not needed as only called before run starts
		synchronized (pairing) {
			pairing.clear();
		}
	}

	public SamplePackage configureSampler(Sampler sampler) {
		SamplePackage pack = (SamplePackage) samplerConfigMap.get(sampler);
		pack.setSampler(sampler);
		configureWithConfigElements(sampler, pack.getConfigs());
		runPreProcessors(pack.getPreProcessors());
		return pack;
	}
    
    public SamplePackage configureTransactionSampler(TransactionSampler transactionSampler) {
        TransactionController controller = transactionSampler.getTransactionController();
        SamplePackage pack = (SamplePackage) transactionControllerConfigMap.get(controller);
        pack.setSampler(transactionSampler);
        return pack;
    }

	private void runPreProcessors(List preProcessors) {
		Iterator iter = preProcessors.iterator();
		while (iter.hasNext()) {
			PreProcessor ex = (PreProcessor) iter.next();
			if (log.isDebugEnabled()) {
				log.debug("Running preprocessor: " + ((AbstractTestElement) ex).getName());
			}
			TestBeanHelper.prepare((TestElement) ex);
			ex.process();
		}
	}

	public void done(SamplePackage pack) {
		pack.recoverRunningVersion();
	}

	public void addNode(Object node, HashTree subTree) {
		stack.addLast(node);
	}

	public void subtractNode() {
		log.debug("Subtracting node, stack size = " + stack.size());
		TestElement child = (TestElement) stack.getLast();
		trackIterationListeners(stack);
		if (child instanceof Sampler) {
			saveSamplerConfigs((Sampler) child);
		}
        else if(child instanceof TransactionController) {
            saveTransactionControllerConfigs((TransactionController) child);
        }
		stack.removeLast();
		if (stack.size() > 0) {
			ObjectPair pair = new ObjectPair(child, (TestElement) stack.getLast());
			synchronized (pairing) {// Called from multiple threads
				if (!pairing.contains(pair)) {
					pair.addTestElements();
					pairing.add(pair);
				}
			}
		}
	}

	private void trackIterationListeners(LinkedList p_stack) {
		TestElement child = (TestElement) p_stack.getLast();
		if (child instanceof LoopIterationListener) {
			ListIterator iter = p_stack.listIterator(p_stack.size());
			while (iter.hasPrevious()) {
				TestElement item = (TestElement) iter.previous();
				if (item == child) {
					continue;
				}
				if (item instanceof Controller) {
					TestBeanHelper.prepare(child);
					((Controller) item).addIterationListener((LoopIterationListener) child);
					break;
				}
			}
		}
	}

	public void processPath() {
	}

	private void saveSamplerConfigs(Sampler sam) {
		List configs = new LinkedList();
		List modifiers = new LinkedList();
		List controllers = new LinkedList();
		List responseModifiers = new LinkedList();
		List listeners = new LinkedList();
		List timers = new LinkedList();
		List assertions = new LinkedList();
		LinkedList posts = new LinkedList();
		LinkedList pres = new LinkedList();
		for (int i = stack.size(); i > 0; i--) {
			addDirectParentControllers(controllers, (TestElement) stack.get(i - 1));
			Iterator iter = testTree.list(stack.subList(0, i)).iterator();
			List tempPre = new LinkedList();
			List tempPost = new LinkedList();
			while (iter.hasNext()) {
				TestElement item = (TestElement) iter.next();
				if ((item instanceof ConfigTestElement)) {
					configs.add(item);
				}
				if (item instanceof SampleListener) {
					listeners.add(item);
				}
				if (item instanceof Timer) {
					timers.add(item);
				}
				if (item instanceof Assertion) {
					assertions.add(item);
				}
				if (item instanceof PostProcessor) {
					tempPost.add(item);
				}
				if (item instanceof PreProcessor) {
					tempPre.add(item);
				}
			}
			pres.addAll(0, tempPre);
			posts.addAll(0, tempPost);
		}

		SamplePackage pack = new SamplePackage(configs, modifiers, responseModifiers, listeners, timers, assertions,
				posts, pres, controllers);
		pack.setSampler(sam);
		pack.setRunningVersion(true);
		samplerConfigMap.put(sam, pack);
	}
    
    private void saveTransactionControllerConfigs(TransactionController tc) {
        List configs = new LinkedList();
        List modifiers = new LinkedList();
        List controllers = new LinkedList();
        List responseModifiers = new LinkedList();
        List listeners = new LinkedList();
        List timers = new LinkedList();
        List assertions = new LinkedList();
        LinkedList posts = new LinkedList();
        LinkedList pres = new LinkedList();
        for (int i = stack.size(); i > 0; i--) {
            addDirectParentControllers(controllers, (TestElement) stack.get(i - 1));
            Iterator iter = testTree.list(stack.subList(0, i)).iterator();
            while (iter.hasNext()) {
                TestElement item = (TestElement) iter.next();
                if (item instanceof SampleListener) {
                    listeners.add(item);
                }
                if (item instanceof Assertion) {
                    assertions.add(item);
                }
            }
        }

        SamplePackage pack = new SamplePackage(configs, modifiers, responseModifiers, listeners, timers, assertions,
                posts, pres, controllers);
        pack.setSampler(new TransactionSampler(tc, tc.getName()));
        pack.setRunningVersion(true);
        transactionControllerConfigMap.put(tc, pack);
    }

	/**
	 * @param controllers
	 * @param i
	 */
	private void addDirectParentControllers(List controllers, TestElement maybeController) {
		if (maybeController instanceof Controller) {
			log.debug("adding controller: " + maybeController + " to sampler config");
			controllers.add(maybeController);
		}
	}

	private static class ObjectPair
	{
		TestElement child, parent;

		public ObjectPair(TestElement one, TestElement two) {
			this.child = one;
			this.parent = two;
		}

		public void addTestElements() {
			if (parent instanceof Controller && (child instanceof Sampler || child instanceof Controller)) {
				parent.addTestElement(child);
			}
		}

		public int hashCode() {
			return child.hashCode() + parent.hashCode();
		}

		public boolean equals(Object o) {
			if (o instanceof ObjectPair) {
				return child == ((ObjectPair) o).child && parent == ((ObjectPair) o).parent;
			}
			return false;
		}
	}

	private void configureWithConfigElements(Sampler sam, List configs) {
		Iterator iter = configs.iterator();
		while (iter.hasNext()) {
			ConfigTestElement config = (ConfigTestElement) iter.next();
			sam.addTestElement(config);
		}
	}
}