性能测试数据生成—java

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

 

public class JDBCTest {
public static void main(String[] args) throws IOException {
    
String user_id1 = "";//保存user_id
File file = new File("d:\\initial_id.txt");
FileWriter os = new FileWriter(file);
BufferedWriter initial_bw = new BufferedWriter(os);//为initial_id创建一个文件

File file1 = new File("d:\\user_id1.txt");
FileWriter os1 = new FileWriter(file1);
BufferedWriter user_bw = new BufferedWriter(os1);//为user_id1创建一个文件

try {
Class.forName("com.mysql.jdbc.Driver");// 加载驱动程序

String url = "jdbc:mysql://10.10.10.12:#####/#####";// URL指向要访问的数据库名
String user = "root";// MySQL配置时的用户名
String password = "#############";// MySQL配置时的密码
Connection connection = DriverManager.getConnection(url, user,password);// 连续数据库
if (!connection.isClosed()){
System.out.println("Succeeded connecting to the Database!");
}
Statement statement = connection.createStatement();// statement用来执行SQL语句

System.out.println("user_id1" + "\t" + "initial_id");
initial_bw.append("initial_id");
initial_bw.newLine();//打印标题

user_bw.append("user_id1");
user_bw.newLine();//打印标题

for(int i=0;i<400;i++){

String sqlname="user_message_"+i;

String sql = "select user_id1,initial_id from "+sqlname+"  group by initial_id order by user_id1 limit 250";// 要执行的SQL语句 www.2cto.com
ResultSet rs = statement.executeQuery(sql);// 执行SQL语句并返回结果集

 

while (rs.next()) {

   //System.out.println(rs.getString("initial_id") + "\t"+ rs.getString("user_id1"));

if(!user_id1.equals(rs.getString("user_id1"))&user_id1.equals(""))
{
user_bw.append(rs.getString("user_id1"));
    user_bw.newLine();

       initial_bw.append(rs.getString("initial_id"));

}
  
else if (!user_id1.equals(rs.getString("user_id1"))) {
   
    user_bw.append(rs.getString("user_id1"));
    user_bw.newLine();
   
    initial_bw.newLine();
    initial_bw.append(rs.getString("initial_id"));

}
   else
   {
    initial_bw.append(","+rs.getString("initial_id"));
   }

   user_id1 = rs.getString("user_id1");
  
}
rs.close();//关闭果集
}
System.out.println("恭喜您！数据生成结束");
connection.close();//数据库连接
initial_bw.flush();
user_bw.flush();

} catch (ClassNotFoundException e) {
// 如果连接数据库失败就会报异常
System.out.println("Sorry,can`t find the Driver!");
e.printStackTrace();
} catch (SQLException e) {
e.printStackTrace();
} catch (Exception e) {
e.printStackTrace();
}

}

}