nGrinder TestRunnerInsertMysqlMore2.groovy

package com.iteye.lindows;

import junit.framework.Assert
import net.grinder.script.GTest
import net.grinder.scriptengine.groovy.junit.GrinderRunner
import net.grinder.scriptengine.groovy.junit.annotation.AfterThread
import net.grinder.scriptengine.groovy.junit.annotation.BeforeProcess
import net.grinder.scriptengine.groovy.junit.annotation.BeforeThread
import org.junit.Test
import org.junit.runner.RunWith

import java.sql.Connection
import java.sql.DriverManager
import java.sql.Statement

import static net.grinder.script.Grinder.grinder
import static org.junit.Assert.assertTrue

/**
 * java代码示例，连接数据库进行查询
 *
 * @author Lindows
 */
@RunWith(GrinderRunner)
class TestRunnerInsert {
    public static GTest insertTable
    public static Connection conn;
    public static Statement stmt; //创建Statement对象
    public static List<String> dbLineList; // 准备参数化数据对象
    public static fileRownNumber;

    @BeforeProcess
    public static void beforeProcess() {
        insertTable = new GTest(1, "插入表数据")
        dbLineList = new File("C:/Users/Lindows/IdeaProjects/TestGroovy/src/resources/database.txt").readLines("UTF-8")
        System.out.println("tangxje:  " + dbLineList.size())
        fileRownNumber = new Random().nextInt(dbLineList.size()-1)+1 //文件database.txt，含titile行0,共10行。
        // 方法二,直接获取(文件行数如:10行-1)=9填入下面，nextInt从0~8取值
        //fileRownNumber = new Random().nextInt(9)
        grinder.logger.info("tangxje ::"  + fileRownNumber)
        try {
            //调用Class.forName()方法加载驱动程序
            Class.forName("com.mysql.jdbc.Driver");
            grinder.logger.info("成功加载MySQL驱动！");
            /*
             String url="jdbc:mysql://10.244.73.65:3306/performance_test";    //JDBC的URL
             String username = "fabu";
             String passwd = "SuningRds@123";
            */
            String url = dbLineList.get(fileRownNumber).split(",")[0];
            String username = dbLineList.get(fileRownNumber).split(",")[1];
            String passwd = dbLineList.get(fileRownNumber).split(",")[2];
            grinder.logger.debug(url)

            conn = DriverManager.getConnection(url, username, passwd);
            stmt = conn.createStatement(); //创建Statement对象
            grinder.logger.info("成功创建stmt！");
        } catch (Exception e) {
            e.printStackTrace()
        }
    }

    @BeforeThread
    public void beforeThread() {
        insertTable.record(this, "insertTable")
        grinder.statistics.delayReports=true
    }

    @Test
    public void insertTable() {
        try{
            grinder.logger.info("成功连接到数据库！");
            StringBuffer sql = new StringBuffer()
            sql.append("insert into tab_001(column_int,column_double,column_decimal,column_varchar_name,column_varchar_address,column_text,column_timestamp_create_time,column_timestamp_update_time) values (1000,300.25,600.98,‘jack‘,‘")
                    .append("China BeiJing")
                    .append(new Random().nextInt(99999999))
                    .append("‘, ‘work in SuNing for 3 years‘,‘2017-06-12 18:00:00‘,‘2017-06-13 15:00:00‘)")
            grinder.logger.info(sql.toString())
            Thread.sleep(new Random().nextInt(10)) //这里可以设置思考时间10ms
            assertTrue(!stmt.execute(sql.toString()))//执行sql insert，!stmt.execute(sql)该写法只于insert true确认
            //assertTrue(stmt.execute(sql));//执行sql query , !stmt.execute(sql)该写法只适用于query true确认
        }catch(Exception e) {
            e.printStackTrace();
        }
    }

    @AfterThread
    public void afterThread() {
        stmt.close();
        conn.close();
    }
}