之前用c++作后台的时候碰到一个问题:返回一张图片数据。查了一些资料终于解决了。核心就是文件流怎么读写的问题,已经将问题简化为文件拷贝,代码如下:
#include <iostream>
#include <fstream>
using namespace std;
int main()
{
cout << "Content-Type:image/png\r\n\r\n";
fstream file;
file.open("1.png", ios::binary|ios::in|ios::ate); //打开时指针在文件尾
int length = file.tellg();
char* imgData = new char[length];
file.seekg(0);
file.read(imgData, length); //二进制只能用这个读
fstream file2;
file2.open("2.png", ios::binary|ios::out);
file2.write(imgData, length); //二进制只能用这个写
cout << "ok";
return 0;
}
结果:
原来是不存在2.png的
----------------------------------------华丽的分割线---------------------------------------------
事实上,以上代码非常辣鸡,存在以下问题:
- 使用了int, 不支持2G以上的文件
- 使用了new申请内存并且没有释放,可能造成程序崩溃
- 没有封装,不方便使用
现修改如下:
#include <iostream>
#include <string>
#include <fstream>
using namespace std;
/* 复制文件
* @参数 src - 源文件名
* @参数 dest - 目标文件名,如果目标文件已存在,则覆盖
* @返回 true | false 代表拷贝是否成功
*/
bool copyFile(string src, string dest) {
ifstream is(src, ios::binary);
if (is.fail()) {
return false;
}
ofstream os(dest, ios::binary);
if (os.fail()) {
return false;
}
is.seekg(0, ios::end);
long long length = is.tellg(); // C++ 支持的最大索引位置
is.seekg(0);
char buf[2048];
while (length > 0)
{
int bufSize = length >= 2048 ? 2048 : length;
is.read(buf, bufSize);
os.write(buf, bufSize);
length -= bufSize;
}
is.close();
os.close();
return true;
}
int main() {
if (copyFile("./ubuntu-19.10-live-server-amd64.iso", "./test_cp.iso")) {
cout << "复制文件成功" << endl;
}
else {
cout << "复制文件失败" << endl;
}
return 0;
}
拷贝大文件轻轻松松。
复制时间的思考
复制文件所需要的时间与磁盘本身、buf的大小都有很大关系。本文分别对大小为347.4M和4.7G的文件进行了拷贝实验。实验条件为:
设备 | 型号及参数 |
---|---|
CPU | Intel Core m3-6Y30 |
内存 | 4GB(4GB×1) LPDDR3(低功耗版)1866MHz |
硬盘 | 型号为 AMSUNG MZNTY128 100Q 容量为 128G SSD |
针对不同buf大小,得到拷贝用时如下。1-time和2-time分别对应347.4M和4.7G的文件。 可以发现,当buf大小增大到一定程度时对拷贝需要的时间趋于稳定。
实验代码如下:
#include <iostream>
#include <string>
#include <fstream>
#include<ctime>
using namespace std;
/* 复制文件
* @param src 源文件名
* @param dest 目标文件名,如果目标文件已存在,则覆盖
* @return true | false 代表拷贝是否成功
*/
bool copyFile(string src, string dest) {
const int BUF_SIZE = 32768;
ifstream is(src, ios::binary);
if (is.fail()) {
return false;
}
ofstream os(dest, ios::binary);
if (os.fail()) {
return false;
}
is.seekg(0, ios::end);
long long length = is.tellg(); // C++ 支持的最大索引位置
is.seekg(0);
char buf[BUF_SIZE];
while (length > 0)
{
int bufSize = length >= BUF_SIZE ? BUF_SIZE : length;
is.read(buf, bufSize);
os.write(buf, bufSize);
length -= bufSize;
}
is.close();
os.close();
return true;
}
int main(){
clock_t startTime = clock(), endTime;
//copyFile("wps-office_11.1.0.10702_amd64.deb", "test.deb");
copyFile("Windows 10 x64.iso", "test.iso");
endTime = clock();
cout << "The run time is:" <<(double)(endTime - startTime) / CLOCKS_PER_SEC << "s" << endl;
return 0;
}