qileilove

package d706;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.TimeZone;

/*
 * 使用JMeter測試mysql數據庫性能，插入數據的程序。
 * 本程序將在可裝入表中插入 10,000條記錄。然后編譯并執行這段代碼，
 * 用測試數據填充可裝入數據表。
 *
 * 建表語句：create table TEST_DB(id int auto_increment primary key, name varchar(20), sex char(1) );  
 *
 * 表TEST_DB 增加列test: alter table TEST_DB add(sex char(10));
 * 表TEST_DB 修改列test: alter table TEST_DB modify sex char(20) not null;
 * 表TEST_DB 刪除列test: alter table TEST_DB drop column sex;
 *
 */
public class Test_DB_Insert {
 
 public static void main(String[] args){
  int numOfTestRecords;                  // 插入數據量;
  Connection con = null;                 // 連接數據庫;
  PreparedStatement statement = null;    // 獲取數據庫操作對象;
  
    try {
     //注冊驅動;
     Class.forName("com.mysql.jdbc.Driver");
     //連接數據庫的信息;
     String dbName = "test";
     String url = "jdbc:mysql://127.0.0.1:3306/" + dbName;
     String userName = "root";
     String pwd = "root";
     
     //獲取數據庫連接;
     con = DriverManager.getConnection(url,userName, pwd);
     
     con.setAutoCommit(false);  //關閉事務自動提交
     
      //記錄執行時間
      SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss:SS"); 
      TimeZone t = sdf.getTimeZone(); 
      t.setRawOffset(0); 
      sdf.setTimeZone(t); 
      Long startTime = System.currentTimeMillis(); 
     
     //創建數據庫操作對象;
     statement = con.prepareStatement("INSERT INTO TEST_DB(name,sex) VALUES(?,?)");
       
     numOfTestRecords = 100; //插入的測試數據量;
     
      for(int i = 0; i<numOfTestRecords; i++) {  //循環
       statement.setString(1,"DBTest-" + i);
       statement.setString(2,"" + i%2);  //0表示男;1表示女.
       statement.addBatch();  // 把一個SQL命令加入命令列表 
       //statement.executeUpdate(); //執行SQL;
      }
     
      statement.executeBatch(); //執行批量更新
      con.commit();//語句執行完畢，提交事務
     
      int[] ref = statement.executeBatch();              
      //if(ref[numOfTestRecords-1] == 0){System.out.println("插入數據操作完成.");} //
      System.out.println("插入數據操作完成.");
      Long endTime = System.currentTimeMillis(); 
         System.out.println("用時：" + sdf.format(new Date(endTime - startTime))); //
     
    }catch(Exception e) {
       System.out.println("An error has occurred: " + e.toString());
       e.printStackTrace();
       }finally{
        
        if(statement != null){ //關閉數據庫操作對象
         try{
          statement.close();
         }catch(SQLException se){
          se.printStackTrace();
         }
        }

        if(con != null){  //關閉數據庫連接
         try{
          if(con.isClosed()){con.close();}  //
         }catch(SQLException se){
          se.printStackTrace();
         }
        }
       }      
      
 }
}