winsocket客户端编程以及jmeter外部调用

  1 public class SocketClient {
  2 OutputStream clientout = null;
  3 InputStream clienIn = null;
  4 byte[] b = new byte[1024];
  5 // String sendstr="From client 111";
  6 static String hex = "";
  7 Socket socket = null;
  8 static String hexStr = "0123456789ABCDEF";
  9 long diff1, diff2, start1, start2, end1, end2;
 10 SocketAddress socketAddress = null;
 11 
 12 /**
 13 * 连接、获取流
 14 */
 15 public String CallClient(String serverIp, int port, String fileName,
 16 int timeout1, int timeout2, String successCode) throws IOException {
 17 try {
 18 socketAddress = new InetSocketAddress(serverIp, port);
 19 long start1 = System.currentTimeMillis();
 20 socket = new Socket();
 21 socket.connect(socketAddress, timeout1);
 22 long end1 = System.currentTimeMillis();
 23 diff1 = end1 - start1;
 24 } catch (BindException e) {
 25 //System.out.println("IP地址或端口绑定异常");
 26 socket.close();
 27 return "IP地址或端口绑定异常！";
 28 } catch (UnknownHostException e) {
 29 socket.close();
 30 return "未识别主机地址！";
 31 } catch (SocketTimeoutException e) {
 32 socket.close();
 33 return "连接超时！";
 34 } catch (ConnectException e) {
 35 socket.close();
 36 return "拒绝连接！";
 37 } catch (Exception e) {
 38 socket.close();
 39 return "连接失败！";
 40 }
 41 
 42 try {
 43 System.out.println("连接建立:端口" + socket.getLocalPort());
 44 clientout = socket.getOutputStream();
 45 clienIn = socket.getInputStream();
 46 hex = file(fileName);
 47 byte[] buffer = HexStringToBinary(hex);
 48 clientout.write(buffer);
 49 // for (byte b : buffer) {
 50 // System.out.print(b);
 51 // }
 52 clientout.flush();
 53 } catch (UnknownHostException e) {
 54 e.printStackTrace();
 55 } catch (IOException e) {
 56 e.printStackTrace();
 57 }
 58 
 59 /**
 60 * 返回response
 61 */
 62 Reader reader = new InputStreamReader(clienIn);
 63 char chars[] = new char[64];
 64 int len;
 65 StringBuffer sb = new StringBuffer();
 66 try {
 67 socket.setSoTimeout(timeout2);
 68 while ((len = reader.read(chars)) != -1) {
 69 sb.append(new String(chars, 0, len));
 70 }
 71 } catch (SocketTimeoutException e) {
 72 socket.close();
 73 return "Socket is closed";
 74 }
 75 catch (IllegalArgumentException e) {
 76 socket.close();
 77 return "timeout can‘t be negative";
 78 }
 79 clienIn.close();
 80 clientout.close();
 81 socket.close();
 82 String resString = sb.toString();
 83 if (resString.contains(successCode)) {
 84 System.out.println("true");
 85 return successCode;
 86 } else {
 87 System.err.println("false");
 88 return resString;
 89 }
 90 }
 91 
 92 /**
 93 * 数组转换成十六进制字符串
 94 * 
 95 * @param byte[]
 96 * @return HexString
 97 */
 98 public static final String bytesToHexString(byte[] bArray) {
 99 StringBuffer sb = new StringBuffer(bArray.length);
100 String sTemp;
101 for (int i = 0; i < bArray.length; i++) {
102 sTemp = Integer.toHexString(0xFF & bArray[i]);
103 if (sTemp.length() < 2)
104 sb.append(0);
105 sb.append(sTemp.toUpperCase());
106 }
107 return sb.toString();
108 }
109 
110 /**
111 * 报文处理
112 * 
113 * @param hexString
114 * @return 将十六进制字符转换为字节数组
115 */
116 public static byte[] HexStringToBinary(String hexString) {
117 // hexString的长度对2取整，作为bytes的长度
118 //    System.out.println("发送报文:" + hexString);
119 int len = hexString.length() / 2;
120 byte[] bytes = new byte[len];
121 byte high = 0;// 字节高四位
122 byte low = 0; // 字节低四位
123 for (int i = 0; i < len; i++) {
124 // 右移四位得到高位
125 high = (byte) ((hexStr.indexOf(hexString.charAt(2 * i))) << 4);
126 low = (byte) hexStr.indexOf(hexString.charAt(2 * i + 1));
127 bytes[i] = (byte) (high | low);// 高地位做或运算
128 }
129 return bytes;
130 }
131 
132 /**
133 * 从文件中获取报文text
134 * 
135 * @return
136 */
137 public String file(String filename) {
138 File file = new File(filename);
139 BufferedReader reader = null;
140 StringBuffer sb = new StringBuffer();
141 try {
142 reader = new BufferedReader(new FileReader(file));
143 String tempString = null;
144 // 一次读入一行，直到读入null为文件结束
145 while ((tempString = reader.readLine()) != null) {
146 // 显示行号
147 sb = sb.append(tempString);
148 }
149 reader.close();
150 } catch (FileNotFoundException e) {
151 e.printStackTrace();
152 } catch (IOException e) {
153 e.printStackTrace();
154 }
155 return sb.toString().replace("\t", "").replace("\n", "").replace(" ", "");
156 }
157 }
158 
159  
160 
161 ****************************************************************************
162 
163 jmeter调用部分
164 
165 /getDefaultParameters方法：主要用于设置传入的参数 
166 //setupTest方法：为初始化方法,用于初始化性能测试时的每个线程 
167 //runTest方法：为性能测试时的线程运行体 
168 //teardownTest方法：为测试结束方法,用于结束性能测试中的每个线程
169 
170 public class jmeter extends AbstractJavaSamplerClient {
171 
172 private String serverIp;
173 private int serverPort;
174 private String file;
175 private int timeout1;//连接超时
176 private int timeout2;//接收超时
177 private String successCode;
178 @Override
179 //runTest方法：为性能测试时的线程运行体 
180 public SampleResult runTest(JavaSamplerContext sc){
181 serverIp=sc.getParameter("serverIp"); 
182 serverPort=Integer.valueOf(sc.getParameter("serverPort"));
183 file=sc.getParameter("file"); 
184 timeout1=Integer.valueOf(sc.getParameter("timeout1"));
185 timeout2=Integer.valueOf(sc.getParameter("timeout2"));
186 successCode=sc.getParameter("successCode"); 
187 
188 SampleResult sr=new SampleResult();
189 sr.setSampleLabel("Socket Test");
190 try{
191 sr.sampleStart();//jmeter开始统计响应时间标记
192 String receive;
193 SocketClient c=new SocketClient(); 
194 receive=c.CallClient(serverIp,serverPort,file,timeout1,timeout2,successCode);
195 // System.out.println(receive);
196 if (receive.equals(successCode)) {
197 sr.setResponseMessage("返回:"+receive.toString());
198 //System.out.println("response："+receive.toString());
199 sr.setDataType(SampleResult.TEXT);
200 sr.setSuccessful(true);
201 }
202 else {
203 sr.setResponseMessage("报错报文："+receive.toString());
204 sr.setSuccessful(false);
205 }
206 }
207 catch(Exception e){
208 e.printStackTrace();
209 sr.setSuccessful(false);
210 }
211 finally{
212 sr.sampleEnd();//jmeter计时结束
213 }
214 return sr;
215 }
216 @Override
217 //getDefaultParameters方法：主要用于设置传入的参数 
218 public Arguments getDefaultParameters(){
219 Arguments params=new Arguments();
220 params.addArgument("serverIp", "127.0.0.1");//设置参数，并赋予默认值1 
221 params.addArgument("serverPort", "8009");//设置参数，并赋予默认值2
222 params.addArgument("file", "D:/EL04.tlt");
223 params.addArgument("timeout1", "10000");
224 params.addArgument("timeout2", "10000");
225 params.addArgument("successCode", "AAAAAA");
226 return params;
227 }
228 @Override
229 public void setupTest(JavaSamplerContext context){
230 super.setupTest(context);
231 }
232 @Override
233 public void teardownTest(JavaSamplerContext context){
234 super.teardownTest(context);
235 }
236 
237 /***
238 * 测试接口 
239 * @param args
240 */
241 public static void main(String[] args) 
242 { 
243 Arguments params = new Arguments(); 
244 params.addArgument("serverIp", "127.0.0.1");//设置参数，并赋予默认值1 
245 params.addArgument("serverPort", "8002");//设置参数，并赋予默认值2
246 params.addArgument("file", "D:/EL04.tlt");
247 params.addArgument("timeout1", "10000");
248 params.addArgument("timeout2", "10000");
249 params.addArgument("successCode", "AAAAAA");
250 JavaSamplerContext arg0 = new JavaSamplerContext(params); 
251 jmeter test = new jmeter(); 
252 test.setupTest(arg0); 
253 test.runTest(arg0);
254 test.teardownTest(arg0); 
255 }
256 
257 }