XML文件动态生成脚本需求咨询（适配负载测试场景）

绝对有现成的方案，而且用Python写起来超简单——毕竟Python处理这类批量生成任务简直是拿手好戏。我给你一套直接能用的脚本，还能满足所有配置需求，咱们一步步来：

这个脚本支持通过命令行配置生成数量、常量字段值、单文件的记录数（以此控制文件大小），还能自动创建输出目录。

import random
import os
import argparse

def generate_16_digit_number():
    # 生成首位不为0的16位随机数字
    return f"{random.randint(1,9)}{random.randint(0,999999999999999):015d}"

def create_xml_file(output_path, constant_value, num_entries=1):
    # 构建XML内容，通过重复record节点控制文件大小
    xml_content = '<?xml version="1.0" encoding="UTF-8"?>\n'
    xml_content += '<data>\n'
    for _ in range(num_entries):
        xml_content += f'  <record>\n'
        xml_content += f'    <constant_field>{constant_value}</constant_field>\n'
        xml_content += f'    <dynamic_field>{generate_16_digit_number()}</dynamic_field>\n'
        xml_content += f'  </record>\n'
    xml_content += '</data>'
    
    # 写入文件
    with open(output_path, 'w', encoding='utf-8') as f:
        f.write(xml_content)

def main():
    parser = argparse.ArgumentParser(description='Generate XML files for load testing')
    parser.add_argument('--count', type=int, required=True, help='Number of XML files to generate')
    parser.add_argument('--constant', type=str, required=True, help='Fixed value for the constant field')
    parser.add_argument('--entries-per-file', type=int, default=1, help='Number of record entries per file (controls file size)')
    parser.add_argument('--output-dir', type=str, default='load_test_xmls', help='Directory to save generated files')
    
    args = parser.parse_args()
    
    # 创建输出目录（不存在则自动创建）
    os.makedirs(args.output_dir, exist_ok=True)
    
    # 批量生成文件
    for i in range(args.count):
        file_path = os.path.join(args.output_dir, f'test_file_{i+1}.xml')
        create_xml_file(file_path, args.constant, args.entries_per_file)
        print(f"Generated {file_path}")

if __name__ == '__main__':
    main()

使用方式 #

在命令行运行脚本，比如：

python xml_generator.py --count 15 --constant "LOAD_TEST_CONSTANT" --entries-per-file 200 --output-dir my_test_data

这条命令会生成15个XML文件，每个文件包含200条<record>节点，存放在my_test_data目录下。

如果你不想用Python，还有这些方案可选：

1. Bash脚本（Linux/macOS环境） #

适合没有Python环境的场景，用awk辅助生成16位随机数：

#!/bin/bash

COUNT=$1
CONSTANT=$2
ENTRIES_PER_FILE=$3
OUTPUT_DIR="load_test_xmls"

mkdir -p $OUTPUT_DIR

for ((i=1; i<=COUNT; i++)); do
    FILE="${OUTPUT_DIR}/test_file_${i}.xml"
    echo '<?xml version="1.0" encoding="UTF-8"?>' > $FILE
    echo '<data>' >> $FILE
    for ((j=1; j<=ENTRIES_PER_FILE; j++)); do
        # 用awk生成16位随机数
        DYNAMIC=$(awk -v min=1000000000000000 -v max=9999999999999999 'BEGIN{srand(); print int(min+rand()*(max-min+1))}')
        echo "  <record>" >> $FILE
        echo "    <constant_field>${CONSTANT}</constant_field>" >> $FILE
        echo "    <dynamic_field>${DYNAMIC}</dynamic_field>" >> $FILE
        echo "  </record>" >> $FILE
    done
    echo '</data>' >> $FILE
    echo "Generated $FILE"
done

运行方式：bash xml_generator.sh 10 "MY_FIXED_VAL" 50

2. JMeter内置Groovy脚本（直接在负载测试中生成） #

如果你的负载测试用JMeter，完全不用提前生成文件——直接在HTTP请求的Body Data里用Groovy动态生成XML，省掉文件IO开销：

def constant = "LOAD_TEST_CONSTANT"
// 生成16位随机数
def dynamic = String.format("%016d", new Random().nextInt(9999999999999999) + 1000000000000000)

return """<?xml version="1.0" encoding="UTF-8"?>
<data>
  <record>
    <constant_field>${constant}</constant_field>
    <dynamic_field>${dynamic}</dynamic_field>
  </record>
</data>"""

内容的提问来源于stack exchange，提问作者arvind_cool