作业要求:
存储一张表,能对表进行添加,搜索和索引。且支持多线程。
表中有100个属性,每个属性都是int64_t类型。
本节实现了对txt文件的读取,写入与范围查询。
代码实现:
数据结构:
每行数据有一个id,以及100个属性。
/* data */
int64_t id; //数据的ID
//数据的100个属性
int64_t property[PRO_NUM];
生成随机数,添加新数据进入txt文件:
使用文件流(#include <fstream>)提供对文件的流式访问。
对每个属性都生成0-100的随机数。
bool AddData(int64_t iid){
int i = 0;
//生成随机数并存入
id = iid;
for(; i < PRO_NUM; i++){ //PRO_NUM为100
property[i] = rand() % 100;
}
//将生成数存入文件中
//打开本地的表文件
ofstream outfile;
outfile.open("data_storage.txt", ios::binary | ios::app | ios::in | ios::out);
//打开文件出错判断
if(!outfile){
cout << "Open file error" << endl;
return false;
}
//存储数据进本地文件中
outfile << id << " ";
for(int i = 0; i < PRO_NUM; i++){
outfile << property[i] << " ";
}
outfile << "\n";
cout << "Write success" << endl;
outfile.close();
return true;
}
实现对txt文件中存储数据的某个属性的范围查询,返回符合要求的行数:
vector<int> FindData(int property_num, int64_t min, int64_t max){
vector<int> row;
vector<int> ret_row; //创建需要返回的行向量vector
//打开本地的表文件
ifstream infile;
infile.open("data_storage.txt", ios::in); //打开模式
int64_t number;
//将txt中的数据存入row中
while(infile >> number){
row.push_back(number);
}
cout << "row.size: " << row.size() << endl;
//判断row中符合要求的行,放入ret_row
for(int i = property_num; i < row.size(); i = i + PRO_NUM + 1){
if(row[i] >= min && row[i] <= max){
ret_row.push_back(1 + i / (PRO_NUM + 1));
}
}
return ret_row;
}
存五行来测试一下功能(查询的第二个属性,范围50-100的行):
看一下生成的txt文件:
没问题。