jmeter中实现java请求实战日志

view code

public class JdbcInsert implements JavaSamplerClient {

    // 全局变量

    PreparedStatement pstmt;

    Connection conn;                                                      

    @Override

    public Arguments getDefaultParameters() {

        // 主要用于把jmeter界面中的参数和java脚本关联起来

        Arguments arguments = new Arguments();

        arguments.addArgument("uuid",UUID.randomUUID().toString());

        arguments.addArgument("username","张三");

        arguments.addArgument("pwd","123");

        arguments.addArgument("jdbcurl","jdbc:mysql://localhost:3306/xxxx"

        return arguments;

    }                                                                     

    // 执行测试操作，类似于LR里的Action函数，执行N次

    @Override

    public SampleResult runTest(JavaSamplerContext context) {

        int rows = 0;

        // 框架要求可能出现问题的代码，不能抛出异常，只能自己处理，把有问题的代码放到try里面，解决方案放到catch里

        try {

            pstmt.setString(1,context.getParameter("uuid"));

            pstmt.setString(2, context.getParameter("username"));

            pstmt.setString(3, context.getParameter("pwd"));

            // 执行sql多次

            rows = pstmt.executeUpdate();

        } catch (SQLException e) {

            // 当出现异常时打印异常

            e.printStackTrace();

        }

        System.out.println(rows);

        return null;

    }                                                                     

    // 初始化操作，类似于LR中的init函数，执行1次

    @Override

    public void setupTest(JavaSamplerContext context) {

        String url =context.getParameter("jdbcurl");

        String username = "root";

        String password = "123123";

        String sql = "insert into t_user (uuid,username,pwd) values(?,?,?)

        // 注册驱动，告诉java要链接哪个数据库

        try {

            Class.forName("com.mysql.jdbc.Driver");

            // 建立mysql数据库链接

            conn = (Connection) DriverManager.getConnection(url, username,

            // 把sql进行编译

            pstmt = (PreparedStatement) conn.prepareStatement(sql);

        } catch (Exception e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }                                                                 

    }                                                                     

    // 结束操作，类似于LR里的end函数，执行1次

    @Override

    public void teardownTest(JavaSamplerContext context) {

        try {

            conn.close();

        } catch (SQLException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }                                                                     

    public static void main(String[] args) {

        JdbcInsert jdbcInsert = new JdbcInsert();

        //先创建一个参数对象Argument

        Arguments arguments = jdbcInsert.getDefaultParameters();

        //再创建一个上下文对象JavaSamplerContext,并且把Argument对象传入

        JavaSamplerContext context = new JavaSamplerContext(arguments);

        jdbcInsert.setupTest(context);

        jdbcInsert.runTest(context);

        jdbcInsert.teardownTest(context);                                 

    }                                                                     

}