在LoadRunner向远程Linux/Unix执行命令行并收集性能数据

前面介绍过在LoadRunner的Java协议实现“ 使用SSH连接Linux”，当然连接之后的故事由你主导。
今天要讲的，是一个非Java版本。是对“ 在LoadRunner中执行命令行程序之：popen()取代system()”的一个升华。
     下面的脚本，是在LoadRunner里连接Linux/Unix远程服务器，收集其磁盘IO的负载到测试结果中。
涉及到三个知识点：
    1、LoadRunner自带“PuTTY Link”的使用，路径为“%LR_PATH%\bin\plink.exe”；
    2、Linux/Unix的磁盘监控指令，读者也可以扩展为其它任何实用指令；
    3、LoadRunner自带函数lr_user_data_point的使用，保存自定义数据到测试结果。
脚本贴出如下：

1. #define BUFFER_SIZE 20480 // 初始给它 20 KB
   
2. extern char* strtok(char *token, const char *delimiter); // 显示申明
   
3. Action(){
   
4.     long fp; // 数据流
   
5.     int count; // 用于保存流长度
   
6.     char buffer[BUFFER_SIZE]; // 给数据流分配内存空间
   
7.     char * row_token; // 记录每一行的地址
   
8.     char field_name[100]; // 第一列的名称
   
9.     int field_value; // 保存系列的值
   
10.     char lrudp_name[100]; // 保存LR自定义指标值
    
11.     int rc; // 保存返回值
    
12. 
    
13.     lr_start_transaction("DiskIO");// Linux采样方式: plink -ssh -l username -pw password hostname command
    
14.     lr_save_string("higkoo", "UserName");
    
15.     lr_save_string("123456", "Password");
    
16.     lr_save_string("192.168.10.31", "Server");
    
17.     lr_save_string("iostat -xc | awk 'NR >2 {print $1, $10}'", "Command"); // 使用iostat拿到磁盘IO的状态信息
    
18.     lr_save_string(lr_eval_string("\"%LR_PATH%\\bin\\plink\" -ssh -l {UserName} -pw {Password} {Server} \"{Command}\" 2>&1 "), "Result"); // 使用plink连接远程Linux服务器并拿到执行结果
    
19.     fp = popen(lr_eval_string("{Result}"), "r");
    
20.     if (fp == NULL) {
    
21.         lr_error_message("执行命令失败");
    
22.         return -1;
    
23.     }
    
24. 
    
25.     count = fread(buffer, sizeof(char), BUFFER_SIZE, fp); // 读取结果
    
26.     if (feof(fp) == 0) {
    
27.         lr_error_message("返回结果太大，请给数据流分配更大内存空间，谢谢！");
    
28.         return -1;
    
29.     }
    
30.     if (ferror(fp)) {
    
31.         lr_error_message ("监控指令返回错误");
    
32.         return -1;
    
33.     }
    
34.     if (count == 0) {
    
35.         lr_error_message("监控指令返回结果为熔");
    
36.         return -1;
    
37.     }
    
38.     buffer[count] = NULL;
    
39. 
    
40.     row_token = (char*) strtok(buffer, "\n"); // 按换行符分割
    
41.     if (row_token == NULL) {
    
42.         lr_error_message ("未发现有效数据");
    
43.         return -1;
    
44.     }
    
45.     while (row_token != NULL) { // 开始读取数据
    
46.         rc = sscanf(row_token, "%s %d", field_name, &field_value); //分割名称与值
    
47.         if (rc != 2) {
    
48.             lr_error_message("Incorrect number of items read from the row.");
    
49.             return -1;
    
50.         }
    
51. 
    
52.         sprintf(lrudp_name, "disk_busy_%s", field_name);// 自定义数据的名称
    
53.         lr_user_data_point(lrudp_name, field_value);// 保存到LR自定义数据
    
54. 
    
55.         row_token = (char*) strtok(NULL, "\n");
    
56.     }
    
57. 
    
58.     pclose(fp);
    
59. 
    
60.     lr_end_transaction("DiskIO", LR_AUTO);
    
61. }

复制代码

功能实现了，后面的故事你来讲~~