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();
         }
        }
       }      
      
 }
}