0

我正在编写一个 Jmeter 测试计划以将文件分块上传到服务器。我对Java知之甚少。

我在带有Bean Shell Pre-Processor的HTTP Request Sampler上使用了while 控制器。我编写了一个简短的脚本来从文件中获取字节,现在我面临的问题是:HTTP 请求采样器在文件上传部分获取文件路径。有没有办法在 Bean Shell 预处理器的内存中创建一个文件,然后在文件路径字段中使用该内存文件变量。

我认为理论上是,这是可能的。因为每当我们上传文件时,我们首先将其放入内存,然后发送到服务器。那么,我们是否可以从字节(1 MB 的块)在内存中创建一个文件,然后将其作为文件上传发送。这是我在 Bean Shell Pre-Processor 中编写的代码

Integer maxChunkSize    = new Integer(vars.get("FILE_MAX_SIZE"));
String uploadFilePath   = vars.get("UPLOAD_FILE");
uploadFileSize      = new File(uploadFilePath).length();
InputStream uploadFile  = new BufferedInputStream(new FileInputStream(uploadFilePath));
int offset      = whileCounter * maxChunkSize;
ByteArrayOutputStream bos = new ByteArrayOutputStream();
byte[] bytes;
int size;

/* creating bytes array to hold file data*/
if (offset < uploadFileSize) {
    if (((int) offset + (int) maxChunkSize) < uploadFileSize) {
        bytes = new byte[ (int) maxChunkSize];
        size = maxChunkSize;
    } else {
        size = (int) (uploadFileSize - offset);
        bytes = new byte[ (int) size];
        vars.put("WHILE_LOOP", "0");
    }
}

/* printing results for debugging */
/*
log.info(" ============================================================== ");
log.info("While counter " + whileCounter.toString() );
log.info("While loop " + vars.get("WHILE_LOOP").toString() );
log.info("The file to upload is : " + uploadFilePath);
log.info("Maximum Chunk size is : " + maxChunkSize.toString());
log.info("Current Offset is : " + offset.toString());
log.info("The file size is " + uploadFileSize.toString());
log.info(" ============================================================== ");
*/


/* here it is giving method invocation on loop counter 2, so skip method is used */
uploadFile.skip(offset);
int bytesRead = uploadFile.read(bytes, 0, size);


/* write to byte output stream to read as a file */
bos.write(bytes, 0, bytesRead); 


/* params for next iteration */
uploadFile.close();
whileCounter++;
vars.put("WHILE_COUNTER", whileCounter.toString() );

预期:通过 JMeter 分块上传文件的替代方法,或者创建一个内存变量,作为 JMeter HTTP 请求采样器->文件上传部分中文件上传路径的文件。

4

1 回答 1

0

我认为您无法使用HTTP 请求采样器实现此目的,我会考虑的选项包括:

  1. 使用 Java 代码实现分块文件上传,而不依赖于 JMeter 的 HTTP 请求采样器,例如

    import org.apache.http.client.methods.HttpPost
    import org.apache.http.entity.InputStreamEntity
    import org.apache.http.impl.client.CloseableHttpClient
    import org.apache.http.impl.client.HttpClients
    
    CloseableHttpClient client = HttpClients.createDefault();
    
    HttpPost post = new HttpPost("http://example.com");
    File file = new File("/path/to/your/file");
    
    InputStreamEntity reqEntity = new InputStreamEntity(new FileInputStream(file), -1, "Your file mime type");
    
    //Alternatively:
    //FileEntity reqEntity = new FileEntity(file, "Your file mime type");
    
    reqEntity.setChunked(true);
    
    client.execute(post)
    
  2. 选择HTTP 原始请求采样器,它只是通过网络发送分块数据(块的大小可通过kg.apc.jmeter.samplers.FileReadChunkSize属性控制),因此您可以使用像Wireshark之类的嗅探器工具记录关联的请求主体数据,然后以所需的块大小重放它. 如果需要,请查看JMeter HTTP 原始请求采样器 - 何时以及如何使用它文章以获取更多信息。

  3. 另请注意,自 JMeter 3.1 起,建议使用 JSR223 测试元素和 Groovy 语言编写脚本,因此请考虑在下一次有机会时迁移到 Groovy。

于 2019-01-31T05:33:40.430 回答