lesson3:使用java代码的方式对不能识别的协议进行压力测试

在我们的实际环境中，我们所使用的协议肯定不只是http的方式，对于rpc等调用协议，目前jmeter没有相应的sampler支持，这时就需要通过引入我们自己写的jar包的方式来解决这个问题。例如：当我们的服务方是采用netty+pb、thrift、dubbo等rpc方式时，本文采用thrift的方式来展示。

下载thrift的编译器：https://thrift.apache.org/download

thrfit服务代码：https://github.com/mantuliu/thriftServerDemo

jmeter的sdk代码：https://github.com/mantuliu/jMetterLessons

1.首先，我们先来实现一个非常简单的thrift服务，maven的pom文件如下：需要依赖thrift等开发包

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
  <modelVersion>4.0.0</modelVersion>
  <groupId>com.mantu</groupId>
  <artifactId>thriftServerDemo</artifactId>
  <packaging>jar</packaging>
  <version>1.0-SNAPSHOT</version>
  <name>thriftServerDemo</name>
  <url>http://maven.apache.org</url>
  <dependencies>
    <dependency>
      <groupId>junit</groupId>
      <artifactId>junit</artifactId>
      <version>3.8.1</version>
      <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.apache.thrift</groupId>
        <artifactId>libthrift</artifactId>
        <version>0.9.0</version>
    </dependency>
    <dependency>
        <groupId>log4j</groupId>
        <artifactId>log4j</artifactId>
        <version>1.2.16</version>
    </dependency>
    <dependency>
        <groupId>org.slf4j</groupId>
        <artifactId>slf4j-log4j12</artifactId>
        <version>1.6.1</version>
    </dependency>
    <dependency>
        <groupId>org.slf4j</groupId>
        <artifactId>slf4j-api</artifactId>
        <version>1.6.1</version>
    </dependency>
  </dependencies>
</project>

2.实现thrift的接口定义文件，服务有一个输入参数，完成一个sayHello的动作：

namespace java com.mantu

service  HelloWorldService {
  string sayHello(1:string username)
}

3.使用thrift的编译器编译接口文件后，得到java文件HelloWorldService

4.服务端实现HelloWorldService的实现类，代码如下：

package com.mantu;

import org.apache.thrift.TException;

/**
 * blog http://www.cnblogs.com/mantu/
 *
 * @author mantu
 *
 */
public class HelloWorldImpl implements HelloWorldService.Iface {

    public HelloWorldImpl() {
    }

    @Override
    public String sayHello(String username) throws TException {
        return "Hello," + username + " welcome to http://www.cnblogs.com/mantu/ ";
    }

}

5.在HelloServerDemo中启动thrift服务：

package com.mantu;

import org.apache.thrift.TProcessor;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.transport.TServerSocket;

/**
 * blog http://www.cnblogs.com/mantu/
 *
 * @author mantu
 *
 */
public class HelloServerDemo {
    public static final int SERVER_PORT = 8090;

    public void startServer() {
        try {
            System.out.println("thrift server start ....");

            TProcessor tprocessor = new HelloWorldService.Processor<HelloWorldService.Iface>(
                    new HelloWorldImpl());
            TServerSocket serverTransport = new TServerSocket(SERVER_PORT);
            TServer.Args tArgs = new TServer.Args(serverTransport);
            tArgs.processor(tprocessor);
            tArgs.protocolFactory(new TBinaryProtocol.Factory());
            TServer server = new TSimpleServer(tArgs);
            server.serve();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        HelloServerDemo server = new HelloServerDemo();
        server.startServer();
    }
}

6.启动thrift服务，监听端口为8090；

7.本文中的jmeter的sdk所使用的项目管理工具为maven，各位在使用的过程中，可能会发生相关的依赖包下载不到的情况，如出现此种情况，请在开发工具中直接引用jmeter相关的开发包，不要使用maven的方式；jMetterLessons的pom文件如下，引用的jmeter的相关包，及thrift的相关包：

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
  <modelVersion>4.0.0</modelVersion>
  <groupId>com.mantu</groupId>
  <artifactId>jMeterLessons</artifactId>
  <packaging>jar</packaging>
  <version>1.0-SNAPSHOT</version>
  <name>jMeterLessons</name>
  <url>http://maven.apache.org</url>
  <dependencies>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_java</artifactId>
        <version>3.0</version>
    </dependency>
     <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_core</artifactId>
        <version>3.0</version>
    </dependency>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_components</artifactId>
        <version>3.0</version>
    </dependency>
     <dependency>
        <groupId>org.apache.thrift</groupId>
        <artifactId>libthrift</artifactId>
        <version>0.9.0</version>
    </dependency>
  </dependencies>
</project>

8.sdk的主要代码实现参加类Lesson3，此类继承jmeter的AbstractJavaSamplerClient：

package com.mantu.jmeter;
 
import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import com.mantu.HelloWorldService;
 
 
/**
 * blog http://www.cnblogs.com/mantu/
 *
 * @author mantu
 *
 */
public class Lesson3 extends AbstractJavaSamplerClient{
     
    public static void main(String [] args){
         
    }
 
    @Override
    public SampleResult runTest(JavaSamplerContext arg0) {
        // TODO Auto-generated method stub
        String userName = arg0.getParameter("uName");
        SampleResult sr = new SampleResult();
        sr.setSampleLabel("thrift娴嬭瘯");
        try{
            sr.sampleStart();
            HelloClientDemo helloClient = new HelloClientDemo();
            helloClient.startClient(userName);
            sr.setResponseData("success");
            sr.setDataType(SampleResult.TEXT);
            sr.setSuccessful(true);
        }
        catch(Exception ex){
            sr.setSuccessful(false);
            ex.printStackTrace();
        }
        finally{
            sr.sampleEnd();
        }
        return sr;
    }
     
    class HelloClientDemo {
 
        public static final String SERVER_IP = "localhost";
        public static final int SERVER_PORT = 8090;
        public static final int TIMEOUT = 30000;
 
        /**
         *
         * @param userName
         */
        public void startClient(String userName) {
            TTransport transport = null;
            try {
                transport = new TSocket(SERVER_IP, SERVER_PORT, TIMEOUT);
                TProtocol protocol = new TBinaryProtocol(transport);
                HelloWorldService.Client client = new HelloWorldService.Client(
                        protocol);
                transport.open();
                System.out.println(client.sayHello(userName));
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                if (null != transport) {
                    transport.close();
                }
            }
        }
    }
}