package kg.apc.jmeter.reporters;
import java.awt.BorderLayout;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.Insets;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.ClipboardOwner;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JComponent;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import kg.apc.jmeter.JMeterPluginsUtils;
import kg.apc.jmeter.gui.BrowseAction;
import kg.apc.jmeter.gui.GuiBuilderHelper;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.visualizers.gui.AbstractListenerGui;
public class FlexibleFileWriterGui extends AbstractListenerGui implements ClipboardOwner {
public static final String WIKIPAGE = "FlexibleFileWriter";
private JTextField filename;
private JTextField columns;
private JCheckBox overwrite;
private JTextArea header;
private JTextArea footer;
private JButton browseButton;
private String[] fields = {
"endTime", "Epoch time when the request was ended",
"endTimeMillis", "Same as endTime, but divided by 1000 (surrogate field, eg. 1311122631.104)",
"isFailed", "If response was marked as failed (surrogate field)",
"isSuccsessful", "If response was marked as successful",
"latency", "Latency, time to first response byte received (if available)",
"latencyMicros", "Same as latency, but multiplied by 1000 (surrogate field)",
"receivedBytes", "Number of request bytes received (if available)",
"requestData", "Request data from sample",
"responseCode", "Response code (eg. 200, 404, etc.)",
"responseData", "Response data",
"responseHeaders", "Response headers (if present in sample)",
"responseMessage", "Response message (eg. OK, Not Found, etc.)",
"responseTime", "Response time, time to full response loaded",
"responseTimeMicros", "Same as responseTime, but multiplied by 1000 (surrogate field)",
"sampleLabel", "Name of the sampler that made the request",
"sentBytes", "Number of request bytes sent (if available)",
"startTime", "Epoch time when the request was started",
"startTimeMillis", "Same as startTime, but divided by 1000 (surrogate field, eg. 1311121131.062)",
"threadName", "Name of thread in Thread Group that processed the request",
"variable#<N>", "Sample variable with index N (eg. variable#2), see help for details"
};
public FlexibleFileWriterGui() {
super();
init();
initFields();
}
@Override
public String getStaticLabel() {
return JMeterPluginsUtils.prefixLabel("Flexible File Writer");
}
@Override
public String getLabelResource() {
return getClass().getCanonicalName();
}
@Override
public TestElement createTestElement() {
TestElement te = new FlexibleFileWriter();
modifyTestElement(te);
te.setComment(JMeterPluginsUtils.getWikiLinkText(WIKIPAGE));
return te;
}
@Override
public void modifyTestElement(TestElement te) {
super.configureTestElement(te);
if (te instanceof FlexibleFileWriter) {
FlexibleFileWriter fw = (FlexibleFileWriter) te;
fw.setFilename(filename.getText());
fw.setColumns(columns.getText());
fw.setOverwrite(overwrite.isSelected());
fw.setFileHeader(header.getText());
fw.setFileFooter(footer.getText());
}
}
@Override
public void clearGui() {
super.clearGui();
initFields();
}
private void initFields() {
filename.setText("testResults.txt");
columns.setText("endTimeMillis|\\t|"
+ "responseTime|\\t|latency|\\t|"
+ "sentBytes|\\t|receivedBytes|\\t|"
+ "isSuccessful|\\t|responseCode|\\r\\n");
overwrite.setSelected(false);
header.setText("endTimeMillis\tresponseTime\tlatency\tsentBytes\t"
+ "receivedBytes\tisSuccessful\tresponseCode\n");
footer.setText("");
}
@Override
public void configure(TestElement element) {
super.configure(element);
FlexibleFileWriter fw = (FlexibleFileWriter) element;
filename.setText(fw.getFilename());
columns.setText(fw.getColumns());
overwrite.setSelected(fw.isOverwrite());
header.setText(fw.getFileHeader());
footer.setText(fw.getFileFooter());
}
private void init() {
setLayout(new BorderLayout(0, 5));
setBorder(makeBorder());
add(JMeterPluginsUtils.addHelpLinkToPanel(makeTitlePanel(), WIKIPAGE), BorderLayout.NORTH);
JPanel mainPanel = new JPanel(new GridBagLayout());
GridBagConstraints labelConstraints = new GridBagConstraints();
labelConstraints.anchor = GridBagConstraints.FIRST_LINE_END;
GridBagConstraints editConstraints = new GridBagConstraints();
editConstraints.anchor = GridBagConstraints.FIRST_LINE_START;
editConstraints.weightx = 1.0;
editConstraints.fill = GridBagConstraints.HORIZONTAL;
addToPanel(mainPanel, labelConstraints, 0, 1, new JLabel("Filename: ", JLabel.RIGHT));
addToPanel(mainPanel, editConstraints, 1, 1, filename = new JTextField(20));
addToPanel(mainPanel, labelConstraints, 2, 1, browseButton = new JButton("Browse..."));
GuiBuilderHelper.strechItemToComponent(filename, browseButton);
browseButton.addActionListener(new BrowseAction(filename));
addToPanel(mainPanel, labelConstraints, 0, 2, new JLabel("Overwrite existing file: ", JLabel.RIGHT));
addToPanel(mainPanel, editConstraints, 1, 2, overwrite=new JCheckBox());
addToPanel(mainPanel, labelConstraints, 0, 3, new JLabel("Write File Header: ", JLabel.RIGHT));
header = new JTextArea();
header.setLineWrap(true);
addToPanel(mainPanel, editConstraints, 1, 3, GuiBuilderHelper.getTextAreaScrollPaneContainer(header, 3));
editConstraints.insets = new java.awt.Insets(2, 0, 0, 0);
labelConstraints.insets = new java.awt.Insets(2, 0, 0, 0);
addToPanel(mainPanel, labelConstraints, 0, 4, new JLabel("Record each sample as: ", JLabel.RIGHT));
addToPanel(mainPanel, editConstraints, 1, 4, columns = new JTextField(20));
editConstraints.insets = new java.awt.Insets(2, 0, 0, 0);
labelConstraints.insets = new java.awt.Insets(2, 0, 0, 0);
addToPanel(mainPanel, labelConstraints, 0, 5, new JLabel("Write File Footer: ", JLabel.RIGHT));
footer = new JTextArea();
footer.setLineWrap(true);
addToPanel(mainPanel, editConstraints, 1, 5, GuiBuilderHelper.getTextAreaScrollPaneContainer(footer, 3));
JPanel container = new JPanel(new BorderLayout());
container.add(mainPanel, BorderLayout.NORTH);
add(container, BorderLayout.CENTER);
add(createHelperPanel(), BorderLayout.SOUTH);
}
private JPanel createHelperPanel() {
JPanel ret = new JPanel(new GridBagLayout());
GridBagConstraints labelConstraints = new GridBagConstraints();
labelConstraints.insets = new Insets(0, 0, 10, 0);
labelConstraints.gridx = 0;
labelConstraints.fill = GridBagConstraints.HORIZONTAL;
labelConstraints.gridwidth = 2;
ret.add(new JLabel("Available sample fields (click any button to copy the field to clipboard):"), labelConstraints);
GridBagConstraints buttonConstraints = new GridBagConstraints();
buttonConstraints.insets = new Insets(4, 0, 0, 0);
buttonConstraints.gridx = 0;
buttonConstraints.fill = GridBagConstraints.HORIZONTAL;
GridBagConstraints detailConstraints = new GridBagConstraints();
detailConstraints.insets = new Insets(4, 10, 0, 0);
detailConstraints.weightx = 1.0;
detailConstraints.fill = GridBagConstraints.HORIZONTAL;
detailConstraints.gridx = 1;
detailConstraints.anchor = GridBagConstraints.WEST;
int line = 1;
CopyAction copyAction = new CopyAction();
for (int i = 0; i < fields.length / 2; i++) {
JButton fieldButton = new JButton(fields[2 * i]);
fieldButton.addActionListener(copyAction);
JTextField fieldDescription = new JTextField(fields[2 * i + 1]);
fieldDescription.setEditable(false);
fieldDescription.setBorder(null);
fieldDescription.setOpaque(false);
GuiBuilderHelper.strechItemToComponent(fieldDescription, fieldButton);
buttonConstraints.gridy = line;
detailConstraints.gridy = line;
ret.add(fieldButton, buttonConstraints);
ret.add(fieldDescription, detailConstraints);
line++;
}
return ret;
}
private void addToPanel(JPanel panel, GridBagConstraints constraints, int col, int row, JComponent component) {
constraints.gridx = col;
constraints.gridy = row;
panel.add(component, constraints);
}
@Override
public void lostOwnership(Clipboard clipboard, Transferable contents) {
// do nothing
}
private class CopyAction
implements ActionListener {
@Override
public void actionPerformed(final ActionEvent e) {
Clipboard clipboard = getToolkit().getSystemClipboard();
Transferable transferable = new Transferable() {
@Override
public Object getTransferData(DataFlavor flavor) {
if (isDataFlavorSupported(flavor)) {
return "|" + ((JButton) e.getSource()).getText() + "|";
}
return null;
}
@Override
public DataFlavor[] getTransferDataFlavors() {
return new DataFlavor[]{
DataFlavor.stringFlavor
};
}
@Override
public boolean isDataFlavorSupported(DataFlavor flavor) {
return DataFlavor.stringFlavor.equals(flavor);
}
};
clipboard.setContents(transferable, FlexibleFileWriterGui.this);
}
}
}