01: /*
02: * Licensed to the Apache Software Foundation (ASF) under one or more
03: * contributor license agreements. See the NOTICE file distributed with
04: * this work for additional information regarding copyright ownership.
05: * The ASF licenses this file to You under the Apache License, Version 2.0
06: * (the "License"); you may not use this file except in compliance with
07: * the License. You may obtain a copy of the License at
08: *
09: * 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, WITHOUT
13: * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
14: * License for the specific language governing permissions and limitations
15: * under the License.
16: *
17: */
18:
19: package org.apache.jmeter.extractor;
20:
21: import org.apache.jmeter.processor.PostProcessor;
22: import org.apache.jmeter.samplers.SampleResult;
23: import org.apache.jmeter.testbeans.TestBean;
24: import org.apache.jmeter.threads.JMeterContext;
25: import org.apache.jmeter.threads.JMeterContextService;
26: import org.apache.jmeter.threads.JMeterVariables;
27: import org.apache.jmeter.util.BeanShellInterpreter;
28: import org.apache.jmeter.util.BeanShellTestElement;
29: import org.apache.jorphan.logging.LoggingManager;
30: import org.apache.jorphan.util.JMeterException;
31: import org.apache.log.Logger;
32:
33: public class BeanShellPostProcessor extends BeanShellTestElement
34: implements Cloneable, PostProcessor, TestBean {
35: private static final Logger log = LoggingManager
36: .getLoggerForClass();
37:
38: private static final long serialVersionUID = 4;
39:
40: // can be specified in jmeter.properties
41: private static final String INIT_FILE = "beanshell.postprocessor.init"; //$NON-NLS-1$
42:
43: protected String getInitFileProperty() {
44: return INIT_FILE;
45: }
46:
47: public void process() {
48: JMeterContext jmctx = JMeterContextService.getContext();
49:
50: SampleResult prev = jmctx.getPreviousResult();
51: final BeanShellInterpreter bshInterpreter = getBeanShellInterpreter();
52: if (prev == null || bshInterpreter == null) {
53: log.error("BeanShell not found");
54: return;
55: }
56:
57: JMeterVariables vars = jmctx.getVariables();
58: try {
59: // Add variables for access to context and variables
60: bshInterpreter.set("ctx", jmctx);//$NON-NLS-1$
61: bshInterpreter.set("vars", vars);//$NON-NLS-1$
62: bshInterpreter.set("prev", prev);//$NON-NLS-1$
63: bshInterpreter.set("data", prev.getResponseData());//$NON-NLS-1$
64: processFileOrScript(bshInterpreter);
65: } catch (JMeterException e) {
66: log.warn("Problem in BeanShell script " + e);
67: }
68: }
69: }
|