/*
* 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.gui;
import java.awt.BorderLayout;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.util.Date;
import javax.swing.BorderFactory;
import javax.swing.JCheckBox;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextField;
import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.control.gui.LoopControlPanel;
import org.apache.jmeter.gui.util.JDateField;
import org.apache.jmeter.gui.util.VerticalPanel;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.property.BooleanProperty;
import org.apache.jmeter.testelement.property.LongProperty;
import org.apache.jmeter.threads.AbstractThreadGroup;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jmeter.util.JMeterUtils;
public class ThreadGroupGui extends AbstractThreadGroupGui implements ItemListener {
private static final long serialVersionUID = 240L;
private LoopControlPanel loopPanel;
private VerticalPanel mainPanel;
private static final String THREAD_NAME = "Thread Field";
private static final String RAMP_NAME = "Ramp Up Field";
private JTextField threadInput;
private JTextField rampInput;
private JDateField start;
private JDateField end;
private final boolean showDelayedStart;
private JCheckBox delayedStart;
private JCheckBox scheduler;
private JTextField duration;
private JTextField delay; // Relative start-up time
public ThreadGroupGui() {
this(true);
}
public ThreadGroupGui(boolean showDelayedStart) {
super();
this.showDelayedStart = showDelayedStart;
init();
initGui();
}
@Override
public TestElement createTestElement() {
ThreadGroup tg = new ThreadGroup();
modifyTestElement(tg);
return tg;
}
/**
* Modifies a given TestElement to mirror the data in the gui components.
*
* @see org.apache.jmeter.gui.JMeterGUIComponent#modifyTestElement(TestElement)
*/
@Override
public void modifyTestElement(TestElement tg) {
super.configureTestElement(tg);
if (tg instanceof AbstractThreadGroup) {
((AbstractThreadGroup) tg).setSamplerController((LoopController) loopPanel.createTestElement());
}
tg.setProperty(AbstractThreadGroup.NUM_THREADS, threadInput.getText());
tg.setProperty(ThreadGroup.RAMP_TIME, rampInput.getText());
tg.setProperty(new LongProperty(ThreadGroup.START_TIME, start.getDate().getTime()));
tg.setProperty(new LongProperty(ThreadGroup.END_TIME, end.getDate().getTime()));
if (showDelayedStart) {
tg.setProperty(ThreadGroup.DELAYED_START, delayedStart.isSelected(), false);
}
tg.setProperty(new BooleanProperty(ThreadGroup.SCHEDULER, scheduler.isSelected()));
tg.setProperty(ThreadGroup.DURATION, duration.getText());
tg.setProperty(ThreadGroup.DELAY, delay.getText());
}
@Override
public void configure(TestElement tg) {
super.configure(tg);
threadInput.setText(tg.getPropertyAsString(AbstractThreadGroup.NUM_THREADS));
rampInput.setText(tg.getPropertyAsString(ThreadGroup.RAMP_TIME));
loopPanel.configure((TestElement) tg.getProperty(AbstractThreadGroup.MAIN_CONTROLLER).getObjectValue());
if (showDelayedStart) {
delayedStart.setSelected(tg.getPropertyAsBoolean(ThreadGroup.DELAYED_START));
}
scheduler.setSelected(tg.getPropertyAsBoolean(ThreadGroup.SCHEDULER));
toggleSchedulerFields(scheduler.isSelected());
// Check if the property exists
String s = tg.getPropertyAsString(ThreadGroup.START_TIME);
if (s.length() == 0) {// Must be an old test plan
start.setDate(new Date());
end.setDate(new Date());
} else {
start.setDate(new Date(tg.getPropertyAsLong(ThreadGroup.START_TIME)));
end.setDate(new Date(tg.getPropertyAsLong(ThreadGroup.END_TIME)));
}
duration.setText(tg.getPropertyAsString(ThreadGroup.DURATION));
delay.setText(tg.getPropertyAsString(ThreadGroup.DELAY));
}
@Override
public void itemStateChanged(ItemEvent ie) {
if (ie.getItem().equals(scheduler)) {
toggleSchedulerFields(scheduler.isSelected());
}
}
/**
* @param enable boolean used to enable/disable fields related to scheduler
*/
private void toggleSchedulerFields(boolean enable) {
start.setEnabled(enable);
end.setEnabled(enable);
duration.setEnabled(enable);
delay.setEnabled(enable);
}
private JPanel createControllerPanel() {
loopPanel = new LoopControlPanel(false);
LoopController looper = (LoopController) loopPanel.createTestElement();
looper.setLoops(1);
loopPanel.configure(looper);
return loopPanel;
}
/**
* Create a panel containing the StartTime field and corresponding label.
*
* @return a GUI panel containing the StartTime field
*/
private JPanel createStartTimePanel() {
JPanel panel = new JPanel(new BorderLayout(5, 0));
JLabel label = new JLabel(JMeterUtils.getResString("starttime")); //$NON-NLS-1$
panel.add(label, BorderLayout.WEST);
start = new JDateField();
panel.add(start, BorderLayout.CENTER);
return panel;
}
/**
* Create a panel containing the EndTime field and corresponding label.
*
* @return a GUI panel containing the EndTime field
*/
private JPanel createEndTimePanel() {
JPanel panel = new JPanel(new BorderLayout(5, 0));
JLabel label = new JLabel(JMeterUtils.getResString("endtime")); // $NON-NLS-1$
panel.add(label, BorderLayout.WEST);
end = new JDateField();
panel.add(end, BorderLayout.CENTER);
return panel;
}
/**
* Create a panel containing the Duration field and corresponding label.
*
* @return a GUI panel containing the Duration field
*/
private JPanel createDurationPanel() {
JPanel panel = new JPanel(new BorderLayout(5, 0));
JLabel label = new JLabel(JMeterUtils.getResString("duration")); // $NON-NLS-1$
panel.add(label, BorderLayout.WEST);
duration = new JTextField();
panel.add(duration, BorderLayout.CENTER);
return panel;
}
/**
* Create a panel containing the Duration field and corresponding label.
*
* @return a GUI panel containing the Duration field
*/
private JPanel createDelayPanel() {
JPanel panel = new JPanel(new BorderLayout(5, 0));
JLabel label = new JLabel(JMeterUtils.getResString("delay")); // $NON-NLS-1$
panel.add(label, BorderLayout.WEST);
delay = new JTextField();
panel.add(delay, BorderLayout.CENTER);
return panel;
}
@Override
public String getLabelResource() {
return "threadgroup"; // $NON-NLS-1$
}
@Override
public void clearGui(){
super.clearGui();
initGui();
}
// Initialise the gui field values
private void initGui(){
threadInput.setText("1"); // $NON-NLS-1$
rampInput.setText("1"); // $NON-NLS-1$
loopPanel.clearGui();
if (showDelayedStart) {
delayedStart.setSelected(false);
}
scheduler.setSelected(false);
Date today = new Date();
end.setDate(today);
start.setDate(today);
delay.setText(""); // $NON-NLS-1$
duration.setText(""); // $NON-NLS-1$
}
private void init() { // WARNING: called from ctor so must not be overridden (i.e. must be private or final)
// THREAD PROPERTIES
VerticalPanel threadPropsPanel = new VerticalPanel();
threadPropsPanel.setBorder(BorderFactory.createTitledBorder(BorderFactory.createEtchedBorder(),
JMeterUtils.getResString("thread_properties"))); // $NON-NLS-1$
// NUMBER OF THREADS
JPanel threadPanel = new JPanel(new BorderLayout(5, 0));
JLabel threadLabel = new JLabel(JMeterUtils.getResString("number_of_threads")); // $NON-NLS-1$
threadPanel.add(threadLabel, BorderLayout.WEST);
threadInput = new JTextField(5);
threadInput.setName(THREAD_NAME);
threadLabel.setLabelFor(threadInput);
threadPanel.add(threadInput, BorderLayout.CENTER);
threadPropsPanel.add(threadPanel);
// RAMP-UP
JPanel rampPanel = new JPanel(new BorderLayout(5, 0));
JLabel rampLabel = new JLabel(JMeterUtils.getResString("ramp_up")); // $NON-NLS-1$
rampPanel.add(rampLabel, BorderLayout.WEST);
rampInput = new JTextField(5);
rampInput.setName(RAMP_NAME);
rampLabel.setLabelFor(rampInput);
rampPanel.add(rampInput, BorderLayout.CENTER);
threadPropsPanel.add(rampPanel);
// LOOP COUNT
threadPropsPanel.add(createControllerPanel());
if (showDelayedStart) {
delayedStart = new JCheckBox(JMeterUtils.getResString("delayed_start")); // $NON-NLS-1$
threadPropsPanel.add(delayedStart);
}
scheduler = new JCheckBox(JMeterUtils.getResString("scheduler")); // $NON-NLS-1$
scheduler.addItemListener(this);
threadPropsPanel.add(scheduler);
mainPanel = new VerticalPanel();
mainPanel.setBorder(BorderFactory.createTitledBorder(BorderFactory.createEtchedBorder(),
JMeterUtils.getResString("scheduler_configuration"))); // $NON-NLS-1$
mainPanel.add(createDurationPanel());
mainPanel.add(createDelayPanel());
mainPanel.add(createStartTimePanel());
mainPanel.add(createEndTimePanel());
toggleSchedulerFields(false);
VerticalPanel intgrationPanel = new VerticalPanel();
intgrationPanel.add(threadPropsPanel);
intgrationPanel.add(mainPanel);
add(intgrationPanel, BorderLayout.CENTER);
}
}