用Java Swing編寫簡單的測試小工具界面（源碼）

前段時間我負責在系統中添加了HBase用於管理海量數據，功能實現後自然到了性能測試階段，性能、可靠性、穩定性，性性都不能少。

因此就有了個內部的小需求：“給測試部提供一個工具，可以向HBase寫入數據，以達到測試所需的數據量”。

由於我們的系統是由Java編寫，Client的代碼也用的Hadoop的Java API，因此小工具的界面編寫理所應當的選擇了Java Swing。

之前沒有用過Swing，本次完全是在網上一邊翻博一邊編碼。

首先是成品圖：

OK，我知道很丑，咱不就圖個實用麼……

這裡可以輸入Hadoop集群的IP，可以選擇需要操作的表，可以清空表中的記錄，還有寫入耗時統計。

下面是代碼，代碼分為兩個文件，一個是MainView負責界面展示；另一個是Worker線程，負責干活和刷新進度條與顯示數據。

MainView代碼：

1. package com.codeevoship.gui;
3. import java.awt.Container;
4. import java.awt.Font;
5. import java.awt.GridBagConstraints;
6. import java.awt.GridBagLayout;
7. import java.awt.Insets;
8. import java.awt.event.ActionEvent;
9. import java.awt.event.ActionListener;
11. import javax.swing.JButton;
12. import javax.swing.JFrame;
13. import javax.swing.JLabel;
14. import javax.swing.JOptionPane;
15. import javax.swing.JProgressBar;
16. import javax.swing.JRadioButton;
17. import javax.swing.JTextField;
19. @SuppressWarnings("serial")
20. publicclass MainWindow extends JFrame {
21. private JLabel welcomeLbl;
22. private JLabel ipLbl;
23. private JTextField ipTxt;
24. private JLabel numLbl;
25. private JTextField numTxt;
26. private JProgressBar probar;
27. private JTextField txtResultShow;
29. private JRadioButton archiveSel;
30. private JRadioButton contentSel;
31. private JRadioButton systemLogSel;
32. private JRadioButton clearDataSel;
34. private JButton startBtn;
35. private JButton exitBtn;
37. private Thread thdWorker;
39. public MainWindow() {
40. // 獲取本JFrame的ContentPane
41. Container contentPane = getContentPane();
43. GridBagLayout gridbag = new GridBagLayout();
44. contentPane.setLayout(gridbag);
45. GridBagConstraints c = new GridBagConstraints();
47. // Title
48. welcomeLbl = new JLabel("CodeEvoShip HBase 數據填充工具");
49. welcomeLbl.setFont(new Font("Serif",Font.BOLD,20));
50. c = getGridBagConstraints(0, 0, 0, 1, 0, 0,
51. GridBagConstraints.CENTER, GridBagConstraints.NONE,
52. new Insets(10, 0, 0, 0), 10, 0);
53. gridbag.setConstraints(welcomeLbl, c);
54. contentPane.add(welcomeLbl);
56. // IP
57. ipLbl = new JLabel("Cluster IP:");
58. c = getGridBagConstraints(0, 1, 1, 1, 0, 0,
59. GridBagConstraints.WEST, GridBagConstraints.NONE,
60. new Insets(10, 0, 10, 0), 10, 0);
61. gridbag.setConstraints(ipLbl, c);
62. contentPane.add(ipLbl);
64. ipTxt = new JTextField("127.0.0.1;127.0.0.2;127.0.0.3");
65. c = getGridBagConstraints(1, 1, 0, 1, 0, 0,
66. GridBagConstraints.WEST, GridBagConstraints.NONE,
67. new Insets(10, 0, 10, 0), 85, 0);
68. gridbag.setConstraints(ipTxt, c);
69. contentPane.add(ipTxt);
71. // 數量填寫
72. numLbl = new JLabel("Write Number:");
73. c = getGridBagConstraints(0, 2, 1, 1, 0, 0,
74. GridBagConstraints.WEST, GridBagConstraints.NONE,
75. new Insets(10, 0, 10, 0), 10, 0);
76. gridbag.setConstraints(numLbl, c);
77. contentPane.add(numLbl);
79. numTxt = new JTextField("10000");
80. c = getGridBagConstraints(1, 2, 0, 1, 0, 0,
81. GridBagConstraints.WEST, GridBagConstraints.NONE,
82. new Insets(10, 0, 10, 0), 230, 0);
83. gridbag.setConstraints(numTxt, c);
84. contentPane.add(numTxt);
86. // 多選框
87. archiveSel = new JRadioButton("Archive Table");
88. c = getGridBagConstraints(0, 3, 1, 1, 0, 0,
89. GridBagConstraints.WEST, GridBagConstraints.NONE,
90. new Insets(0, 0, 0, 0), 0, 0);
91. gridbag.setConstraints(archiveSel, c);
92. contentPane.add(archiveSel);
94. contentSel = new JRadioButton("Content Table");
95. c = getGridBagConstraints(1, 3, 1, 1, 0, 0,
96. GridBagConstraints.WEST, GridBagConstraints.NONE,
97. new Insets(0, 0, 0, 0), 0, 0);
98. gridbag.setConstraints(contentSel, c);
99. contentPane.add(contentSel);
101. systemLogSel = new JRadioButton("SystemLog Table");
102. c = getGridBagConstraints(2, 3, 1, 1, 0, 0,
103. GridBagConstraints.WEST, GridBagConstraints.NONE,
104. new Insets(0, 0, 5, 0), 0, 0);
105. gridbag.setConstraints(systemLogSel, c);
106. contentPane.add(systemLogSel);
108. clearDataSel = new JRadioButton("Clear old data of selected Tables!!");
109. c = getGridBagConstraints(0, 4, 0, 1, 0, 0,
110. GridBagConstraints.WEST, GridBagConstraints.NONE,
111. new Insets(0, 0, 15, 0), 0, 0);
112. gridbag.setConstraints(clearDataSel, c);
113. contentPane.add(clearDataSel);
115. // 進度條
116. probar = new JProgressBar(0, 100);
117. probar.setStringPainted(true);
118. c = getGridBagConstraints(0, 6, 0, 1, 0, 0,
119. GridBagConstraints.WEST, GridBagConstraints.NONE,
120. new Insets(5, 5, 5, 5), 250, 0);
121. gridbag.setConstraints(probar, c);
122. contentPane.add(probar);
124. txtResultShow = new JTextField("Not Begin");
125. txtResultShow.setEditable(false);
126. txtResultShow.setHorizontalAlignment(JTextField.CENTER);
127. c = getGridBagConstraints(0, 7, 0, 1, 0, 0,
128. GridBagConstraints.WEST, GridBagConstraints.NONE,
129. new Insets(0, 0, 5, 0), 325, 0);
130. gridbag.setConstraints(txtResultShow, c);
131. contentPane.add(txtResultShow);
133. //執行與退出
134. startBtn = new JButton("Start");
135. c = getGridBagConstraints(0, 5, 0, 1, 0, 0,
136. GridBagConstraints.WEST, GridBagConstraints.NONE,
137. new Insets(0, 0, 5, 0), 50, 0);
138. gridbag.setConstraints(startBtn, c);
139. startBtn.addActionListener(new ActionListener() {
140. publicvoid actionPerformed(ActionEvent arg0) {
141. if (!paraPassCheck()) {
142. return;
143. }
145. if (null != thdWorker && thdWorker.isAlive()) {
146. JOptionPane.showMessageDialog(null,"There already is a task running now， please wait");
147. return;
148. }
149. DoWork doWork = new DoWork(ipTxt.getText(),Long.parseLong(numTxt.getText()),
150. contentSel.isSelected(),archiveSel.isSelected(),
151. systemLogSel.isSelected(),clearDataSel.isSelected(),
152. probar, txtResultShow);
153. thdWorker = new Thread(doWork);
154. thdWorker.start();
155. }
156. });
157. contentPane.add(startBtn);
159. exitBtn = new JButton("Exit");
160. c = getGridBagConstraints(1, 5, 0, 1, 0, 0,
161. GridBagConstraints.EAST, GridBagConstraints.NONE,
162. new Insets(0, 0, 5, 0), 50, 0);
163. gridbag.setConstraints(exitBtn, c);
164. exitBtn.addActionListener(new ActionListener() {
165. publicvoid actionPerformed(ActionEvent arg0) {
166. System.exit(0);
167. }
168. });
169. contentPane.add(exitBtn);
170. }
172. privateboolean paraPassCheck() {
173. try {
174. Long.parseLong(numTxt.getText());
175. returntrue;
176. } catch (Exception e) {
177. JOptionPane.showMessageDialog(null,"Invalid number ");
178. returnfalse;
179. }
180. }
182. privatestatic GridBagConstraints getGridBagConstraints(int gridx,
183. int gridy, int gridwidth, int gridheight, double weightx,
184. double weighty, int anchor, int fill, Insets insets, int ipadx,
185. int ipady) {
187. returnnew GridBagConstraints(gridx, gridy, gridwidth, gridheight,
188. weightx, weighty, anchor, fill, insets, ipadx, ipady);
189. }
191. publicstaticvoid main(String args[]) {
192. MainWindow mainWindow = new MainWindow();
193. mainWindow.setTitle("CodeEvoShip");
194. mainWindow.pack();
195. mainWindow.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
196. mainWindow.setVisible(true);
197. }
199. }

Worker代碼：

1. package com.codeevoship.gui;
3. import javax.swing.JProgressBar;
4. import javax.swing.JTextField;
6. publicclass DoWork implements Runnable {
8. JProgressBar probar;
9. JTextField curTxtShow;
10. long totalNum = 0;
11. long curNum = 0;
12. long startTime = 0;
13. long curTime = 0;
15. public DoWork(String ip, long num, boolean isWriteCon, boolean isWriteArc, boolean isWriteSys, boolean isClearData, JProgressBar probar, JTextField curTxtShow) {
16. System.out.println("IP:" + ip + ", Num:" + num);
17. System.out.println("Write Content[" + isWriteCon + "], Write Archive[" + isWriteArc + "], Write SystemLog[" + isWriteSys + "].");
18. startTime = 0;
19. curNum = 0;
20. totalNum = num;
21. this.probar = probar;
22. this.curTxtShow = curTxtShow;
23. }
25. @Override
26. publicvoid run() {
27. startTime = System.currentTimeMillis();
28. // 開始工作
29. System.out.println("Do Work Start!");
30. StringBuilder sb = new StringBuilder();
31. while (true) {
32. if (curNum >= totalNum) {
33. break;
34. }
36. try {
37. // TODO: Real work here
38. Thread.sleep(10);
40. // 更新界面顯示
41. curNum += 1;
42. probar.setValue((int) (curNum*100/totalNum));
43. curTime = System.currentTimeMillis();
44. sb.setLength(0);
45. sb.append(curNum);
46. sb.append('/');
47. sb.append(totalNum);
48. sb.append(" cost time: ");
49. sb.append((curTime - startTime)/1000);
50. sb.append('s');
51. curTxtShow.setText(sb.toString());
53. } catch (Exception e) {
54. e.printStackTrace();
55. }
56. }
57. System.out.println("Do Work End!");
58. }
59. }

大家如需要可隨意使用，改成自己的小工具，只要把Worker中的// TODO: 改為實際運行的程序即可。