#include <algorithm>
#include <iostream>
#include <ostream>
#include <bsoncxx/json.hpp>
#include <bsoncxx/stdx/make_unique.hpp>
#include <mongocxx/client.hpp>
#include <mongocxx/gridfs/bucket.hpp>
#include <mongocxx/instance.hpp>
#include <mongocxx/uri.hpp>
#include <bsoncxx/types/bson_value/view.hpp>
#include <fstream>
#include <bsoncxx/array/view.hpp>
#include <bsoncxx/builder/basic/array.hpp>
#include <bsoncxx/builder/basic/document.hpp>
#include <bsoncxx/builder/basic/kvp.hpp>
#include <bsoncxx/document/value.hpp>
#include <bsoncxx/document/view.hpp>
#include <bsoncxx/json.hpp>
#include <bsoncxx/stdx/string_view.hpp>
#include <bsoncxx/string/to_string.hpp>
#include <bsoncxx/types.hpp>
#include <bsoncxx/types/bson_value/view.hpp>
#include <bsoncxx/builder/stream/array.hpp>
#include <bsoncxx/builder/stream/document.hpp>
#include <bsoncxx/builder/stream/helpers.hpp>
#include <bsoncxx/builder/basic/document.hpp>
using namespace mongocxx;
using namespace bsoncxx;
using bsoncxx::stdx::make_unique;
using namespace bsoncxx;
using namespace builder::stream;
int main() {
std::cout << "start" << std::endl;
// 初始化连接
mongocxx::instance inst{};
mongocxx::client conn{ mongocxx::uri{/*数据库地址用户名密码等信息*/}};
auto db = conn["test"];
auto bucket = db.gridfs_bucket();
mongocxx::gridfs::uploader up = bucket.open_upload_stream("img");
std::ifstream fp;
//以读取方式打开jpg文件
fp.open("D:/2.png", std::ios::binary | std::ios::in);
//定位到文件末尾
fp.seekg(0, fp.end);
//获得文件总长度
size_t allLength = fp.tellg();
//将指针定位到文件首
fp.seekg(0, fp.beg);
char* buff = new char[allLength + 10]{ 0 };
fp.read(buff, allLength);
up.write(reinterpret_cast<uint8_t*>(const_cast<char*>(buff)), allLength);
delete[]buff;
buff = nullptr;
auto result = up.close();
// 文件唯一ID
auto kk = result.id();
/**************下载示例***************/
// 通过文件的ID读取文件
std::string idStr = "620f1faaf06f0000de000e52";
// 构造oid构造函数参数
bsoncxx::stdx::string_view fileIdSv = idStr;
// 构造BSON格式ID
auto bsonId = oid(fileIdSv);
builder::stream::document build_doc;
// 构建文档
build_doc << "id" << bsonId;
// 获取文档视图
auto doc = build_doc.view();
// 通过获取文档的值来转化为bsoncxx::types::bson_value::view类型
auto downloader = bucket.open_download_stream(doc["id"].get_value());
// 文件长度
auto file_length = downloader.file_length();
// 读取字节计数
auto bytes_counter = 0;
// 需要的缓冲区大小
auto buffer_size = std::min(file_length, static_cast<std::int64_t>(downloader.chunk_size()));
// 创建缓冲区
auto buffer = make_unique<std::uint8_t[]>(static_cast<std::size_t>(buffer_size));
// 文件存储位置
std::fstream fq;
fq.open("E:/Desktop/3.png", std::ios::binary | std::ios::out);
// 循环接受下载并写入
while (auto length_read =
downloader.read(buffer.get(), static_cast<std::size_t>(buffer_size))) {
bytes_counter += static_cast<std::int32_t>(length_read);
fq.write(reinterpret_cast<char*>(const_cast<uint8_t*>(buffer.get())), length_read);
}
fq.close();
std::cout << "总共接收字节数" << bytes_counter << std::endl;
return 0;
}
C++使用mongo-cxx-driver进行文件GridFS上传下载
最新推荐文章于 2022-07-31 12:09:32 发布