Author：Pirate Leo

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

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

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

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

轉載請註明出處，謝謝。

首先是成品圖：

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

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

下面是程式碼，程式碼分為兩個檔案，一個是MainView負責介面展示；另一個是Worker執行緒，負責幹活和重新整理進度條與顯示資料。

MainView程式碼：

package com.codeevoship.gui;  
  
import java.awt.Container;  
import java.awt.Font;  
import java.awt.GridBagConstraints;  
import java.awt.GridBagLayout;  
import java.awt.Insets;  
import java.awt.event.ActionEvent;  
import java.awt.event.ActionListener;  
  
import javax.swing.JButton;  
import javax.swing.JFrame;  
import javax.swing.JLabel;  
import javax.swing.JOptionPane;  
import javax.swing.JProgressBar;  
import javax.swing.JRadioButton;  
import javax.swing.JTextField;  
  
@SuppressWarnings("serial")  
public class MainWindow extends JFrame {  
    private JLabel welcomeLbl;  
    private JLabel ipLbl;  
    private JTextField ipTxt;  
    private JLabel numLbl;  
    private JTextField numTxt;  
    private JProgressBar probar;  
    private JTextField txtResultShow;  
  
    private JRadioButton archiveSel;  
    private JRadioButton contentSel;  
    private JRadioButton systemLogSel;  
    private JRadioButton clearDataSel;  
      
    private JButton startBtn;  
    private JButton exitBtn;  
      
    private Thread thdWorker;  
      
    public MainWindow() {  
        // 獲取本JFrame的ContentPane  
        Container contentPane = getContentPane();  
          
        GridBagLayout gridbag = new GridBagLayout();  
        contentPane.setLayout(gridbag);  
        GridBagConstraints c = new GridBagConstraints();  
          
        // Title  
        welcomeLbl = new JLabel("CodeEvoShip HBase 資料填充工具");  
        welcomeLbl.setFont(new Font("Serif",Font.BOLD,20));  
        c = getGridBagConstraints(0, 0, 0, 1, 0, 0,  
                GridBagConstraints.CENTER, GridBagConstraints.NONE,   
                new Insets(10, 0, 0, 0), 10, 0);  
        gridbag.setConstraints(welcomeLbl, c);  
        contentPane.add(welcomeLbl);  
          
        // IP  
        ipLbl = new JLabel("Cluster IP:");  
        c = getGridBagConstraints(0, 1, 1, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(10, 0, 10, 0), 10, 0);  
        gridbag.setConstraints(ipLbl, c);  
        contentPane.add(ipLbl);  
              
        ipTxt = new JTextField("127.0.0.1;127.0.0.2;127.0.0.3");      
        c = getGridBagConstraints(1, 1, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(10, 0, 10, 0), 85, 0);  
        gridbag.setConstraints(ipTxt, c);  
        contentPane.add(ipTxt);  
          
        // 數量填寫  
        numLbl = new JLabel("Write Number:");  
        c = getGridBagConstraints(0, 2, 1, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(10, 0, 10, 0), 10, 0);  
        gridbag.setConstraints(numLbl, c);  
        contentPane.add(numLbl);  
                  
        numTxt = new JTextField("10000");  
        c = getGridBagConstraints(1, 2, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(10, 0, 10, 0), 230, 0);  
        gridbag.setConstraints(numTxt, c);  
        contentPane.add(numTxt);  
          
        // 多選框  
        archiveSel = new JRadioButton("Archive Table");       
        c = getGridBagConstraints(0, 3, 1, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 0, 0), 0, 0);  
        gridbag.setConstraints(archiveSel, c);  
        contentPane.add(archiveSel);  
          
        contentSel = new JRadioButton("Content Table");       
        c = getGridBagConstraints(1, 3, 1, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 0, 0), 0, 0);  
        gridbag.setConstraints(contentSel, c);  
        contentPane.add(contentSel);  
          
        systemLogSel = new JRadioButton("SystemLog Table");       
        c = getGridBagConstraints(2, 3, 1, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 5, 0), 0, 0);  
        gridbag.setConstraints(systemLogSel, c);  
        contentPane.add(systemLogSel);  
          
        clearDataSel = new JRadioButton("Clear old data of selected Tables!!");       
        c = getGridBagConstraints(0, 4, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 15, 0), 0, 0);  
        gridbag.setConstraints(clearDataSel, c);  
        contentPane.add(clearDataSel);  
          
        // 進度條  
        probar = new JProgressBar(0, 100);  
        probar.setStringPainted(true);  
        c = getGridBagConstraints(0, 6, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(5, 5, 5, 5), 250, 0);  
        gridbag.setConstraints(probar, c);  
        contentPane.add(probar);  
          
        txtResultShow = new JTextField("Not Begin");  
        txtResultShow.setEditable(false);  
        txtResultShow.setHorizontalAlignment(JTextField.CENTER);  
        c = getGridBagConstraints(0, 7, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 5, 0), 325, 0);  
        gridbag.setConstraints(txtResultShow, c);  
        contentPane.add(txtResultShow);  
  
        //執行與退出  
        startBtn = new JButton("Start");  
        c = getGridBagConstraints(0, 5, 0, 1, 0, 0,  
                GridBagConstraints.WEST, GridBagConstraints.NONE,   
                new Insets(0, 0, 5, 0), 50, 0);  
        gridbag.setConstraints(startBtn, c);  
        startBtn.addActionListener(new ActionListener() {  
            public void actionPerformed(ActionEvent arg0) {  
                if (!paraPassCheck()) {  
                    return;   
                }  
                  
                if (null != thdWorker && thdWorker.isAlive()) {  
                    JOptionPane.showMessageDialog(null,"There already is a task running now， please wait");  
                    return;  
                }  
                DoWork doWork = new DoWork(ipTxt.getText(),Long.parseLong(numTxt.getText()),  
                        contentSel.isSelected(),archiveSel.isSelected(),  
                        systemLogSel.isSelected(),clearDataSel.isSelected(),  
                        probar, txtResultShow);  
                thdWorker = new Thread(doWork);  
                thdWorker.start();  
            }  
        });  
        contentPane.add(startBtn);  
          
        exitBtn = new JButton("Exit");  
        c = getGridBagConstraints(1, 5, 0, 1, 0, 0,  
                GridBagConstraints.EAST, GridBagConstraints.NONE,   
                new Insets(0, 0, 5, 0), 50, 0);  
        gridbag.setConstraints(exitBtn, c);  
        exitBtn.addActionListener(new ActionListener() {  
            public void actionPerformed(ActionEvent arg0) {  
                System.exit(0);  
            }  
        });  
        contentPane.add(exitBtn);  
    }  
      
    private boolean paraPassCheck() {  
        try {  
            Long.parseLong(numTxt.getText());  
            return true;  
        } catch (Exception e) {  
            JOptionPane.showMessageDialog(null,"Invalid number ");  
            return false;  
        }  
    }  
      
    private static GridBagConstraints getGridBagConstraints(int gridx,  
            int gridy, int gridwidth, int gridheight, double weightx,  
            double weighty, int anchor, int fill, Insets insets, int ipadx,  
            int ipady) {  
  
        return new GridBagConstraints(gridx, gridy, gridwidth, gridheight,  
                weightx, weighty, anchor, fill, insets, ipadx, ipady);  
    }  
      
    public static void main(String args[]) {  
        MainWindow mainWindow = new MainWindow();  
        mainWindow.setTitle("CodeEvoShip");  
        mainWindow.pack();  
        mainWindow.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);  
        mainWindow.setVisible(true);  
    }  
  
}  
 

package com.codeevoship.gui;  
  
import javax.swing.JProgressBar;  
import javax.swing.JTextField;  
  
public class DoWork implements Runnable {  
  
    JProgressBar probar;  
    JTextField curTxtShow;  
    long totalNum = 0;  
    long curNum = 0;  
    long startTime = 0;  
    long curTime = 0;  
  
    public DoWork(String ip, long num, boolean isWriteCon, boolean isWriteArc, boolean isWriteSys, boolean isClearData, JProgressBar probar, JTextField curTxtShow) {  
        System.out.println("IP:" + ip + ", Num:" + num);  
        System.out.println("Write Content[" + isWriteCon + "], Write Archive[" + isWriteArc + "], Write SystemLog[" + isWriteSys + "].");  
        startTime = 0;  
        curNum = 0;  
        totalNum = num;  
        this.probar = probar;  
        this.curTxtShow = curTxtShow;  
    }  
      
    @Override  
    public void run() {  
        startTime = System.currentTimeMillis();  
        // 開始工作  
        System.out.println("Do Work Start!");  
        StringBuilder sb = new StringBuilder();  
        while (true) {  
            if (curNum >= totalNum) {  
                break;  
            }  
  
            try {  
                // TODO: Real work here  
                Thread.sleep(10);  
                  
                // 更新介面顯示  
                curNum += 1;  
                probar.setValue((int) (curNum*100/totalNum));  
                curTime = System.currentTimeMillis();  
                sb.setLength(0);  
                sb.append(curNum);  
                sb.append('/');  
                sb.append(totalNum);  
                sb.append(" cost time: ");  
                sb.append((curTime - startTime)/1000);  
                sb.append('s');  
                curTxtShow.setText(sb.toString());  
                  
            } catch (Exception e) {  
                e.printStackTrace();  
            }  
        }  
        System.out.println("Do Work End!");  
    }  
}