loadrunner 脚本开发-文件读写操作
函数说明
函数原型：
size_t fwrite( const void *buffer, size_t size, size_t count, FILE *file_pointer );
参数说明：
buffer:包含数据的缓冲区
size:缓冲区buffer的大小(字节为单位)
count:缓冲区中写入多少个size大小数据(The number of bytes to store in the buffer)
file_pointer:指向存放数据文件的文件指针
size_t: 际写入的数据块数目(每个数据块大小为size)
函数原型：
size_t fread( void *buffer, size_t size, size_t count, FILE *file_pointer );
参数说明：
同fwrite
函数原型：
int ferror( FILE *file_pointer );
功能：检查文件i/o错误
参数说明：
file_pointer:指向文件的文件指针
函数原型
int feof( FILE *file_pointer );
功能：检测流上的文件结束符
参数说明：
file_pointer:指向文件的文件指针
以下通过例子说明如何进行数据存取操作
例子1：
从d盘的readme.txt中读取文件内容并显示读取字节数
代码：
Action()
{
int count, total = 0;
char buffer[1000];
long file_stream;
char * filename = "d:\\readme.txt";
if ((file_stream = fopen(filename, "r")) == NULL )
{
lr_error_message ("Cannot open %s", filename);
return -1;
}
// Read until end of file
while (!feof(file_stream))
{
// Read 1000 bytes while maintaining a running count
count = fread(buffer, sizeof(char), 1000, file_stream);
lr_output_message ("= bytes read", count);
if (ferror(file_stream)) {
lr_output_message ("Error reading file %s", filename);
break;
}
total += count; // Add up actual bytes read
}
// Display final total
lr_output_message ("Total number of bytes read = %d", total );
// Close the file stream
if (fclose(file_stream))
{
lr_error_message ("Error closing file %s", filename);
}
return 0;
}
结果：
例子2：
将每个学生的数据(学号和成绩)保存到某个文件，然后从该文件读取存放的学生数据记录并展示
代码：
Action()
{
// 定义保存学生数据的文件名
#ifdef unix
char * filename = "/tmp/marks.txt";
#else
char * filename = "d:\\marks.txt";
#endif
// 保存学生数据(学号和成绩)
struct student_data
{
int student_id;
unsigned char marks[10];
};
struct student_data sd1; // 定义结构体，用于定义学生数据
struct student_data sd2; // 用于存放从文件中读取的学生数据
long file_handle = NULL; // 文件句柄
int i = 0;
int count = 0; // 统计每次读取学生记录个数
int total = 0; // 用于统计总的学生记录个数
// 打开、创建数据文件
if ((file_handle = fopen(filename, "w+")) == NULL)
{
lr_output_message ("Unable to create %s", filename);
return -1;
}
// 设置学号
sd1.student_id = 1001;
// 设置成绩(假设10门课程)
for (i = 0; i < 10; i++)
{
sd1.marks[i] = (unsigned char)(85 + i);
}
// 把学生记录写入数据文件
i = fwrite(&sd1, sizeof(struct student_data), 1, file_handle);
if (i > 0) // fwrite调用不成功则返回0
{
lr_output_message ("Successfully wrote %d record", i);
}
fclose(file_handle); // 关闭文件
if ((file_handle = fopen(filename, "r")) == NULL ) // 如果打开文件失败
{
lr_error_message ("Cannot open %s", filename);
return -1;
}
// 读取全部文件内容
while (!feof(file_handle))
{
count = fread(&sd2, sizeof(struct student_data), 1, file_handle);
lr_output_message ("%d 个学生数据记录", count);
// 检查文件I/O错误
if (ferror(file_handle))
{
lr_output_message ("Error reading file %s", filename);
break;
}
total += count; // 计算实际读取的总字节数
}
// 展示最后的数据
lr_output_message("学生的学号为：%d,成绩如下：", sd2.student_id);
for(i=0; i<10; i++)
{
lr_output_message("%u", sd2.marks[i]);
}
lr_output_message ("Total number of student record = %d", total);
// 关键文件流
if (fclose(file_handle))
{
lr_error_message ("Error closing file %s", filename);
}
return 0;
}
运行结果：
