tableIO.cpp

#include "tableManager.h"
#include <memory>
vector<blockData> blockData::allData;
QMutex blockData::mutex;

vector<vector<string>> table::toStrVec(){
    int m=(this->allCol[0]->getAllData()).size()+1;
    int n=this->allCol.size();
    vector<vector<string>> tableframe(m,vector<string>(n,""));
    for(int i=0;i<n;i++){
        int sort=this->allCol[i]->getType();
        tableframe[0][i]=(this->allCol[i]->ID)+":"+to_string(sort);
    }
    int i=0;
    for (col* c : this->allCol)
    {   int j=1;
        for (Basic* b : c->getAllData())
        {
            tableframe[j][i]=b->toStr();
            j++;
        }
        i++;
    }
    return tableframe;
}

QString table::toStr()
{
    const vector<vector<string>>& tableframe=this->toStrVec();
    string data;
    int m=tableframe.size();
    int n=tableframe[0].size();
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            if(j==n-1){
                data.append(tableframe[i][j]+"\n");
            }
            else{
                data.append(tableframe[i][j]+",");
            }
        }
    }
    return QString::fromStdString(data);
}


vector<string> table::toStr(const int& fileLen){
    const vector<vector<string>>& tableframe=this->toStrVec();
    int m=tableframe.size();
    int n=tableframe[0].size();
    string title="";
    for(int i=0;i<n;++i){
        if(i==n-1){
            title.append(tableframe[0][i]+"\n");
            continue;
        }
        title.append(tableframe[0][i]+",");
    }
    int totalNum=(m-1)/fileLen+((m-1)%fileLen==0? 0:1);
    if(totalNum==0){
        return {title};
    }
    vector<string> result(totalNum,"");
    for(int i=1;i<m;i++){
        int tmpNum=(i-1)/fileLen;
        string& data=result[tmpNum];
        if((i-1)%fileLen==0){
            data.append(title);
        }
        for(int j=0;j<n;j++){
            if(j==n-1){
                data.append(tableframe[i][j]+"\n");
            }
            else{
                data.append(tableframe[i][j]+",");
            }
        }
    }
    return result;
}

void table:: saveFile(const string& path) //将整个表的内容按约定格式写入空文件
{
    const vector<string>& allData=this->toStr(IO::singleFileLen);
    int num=1;
    for(const string& data:allData){
        const string& tmp=IO::path_to_splitpath(path,num);
        const vector<vector<int>>& len_data=IO::strdata_to_lendata(data);
        multiSave* ms= new multiSave(tmp,data,len_data);
        ms->setAutoDelete(true);
        QThreadPool::globalInstance()->start(ms);
        ++num;
    }
    IO::del_table_blocks(path,num);
    this->allRecord.clear();
}

table* table::loadFile(const string& path,int mark) //按约定格式从文件中读取表
{
    vector<col*> cols;
    vector<int> blocksLen;
    const int& num=IO::table_blocks_num(path);
    if(num==0){
        throw string("No corresponding table was found");
    }
    for(int i=1;i<=num;++i){
        multiRead* mr=new multiRead (IO::path_to_splitpath(path,i),i);
        mr->setAutoDelete(true);
        QThreadPool::globalInstance()->start(mr);
    }
    while (true) {
        if(blockData::allData.size()==num){
            break;
        }
    }
    sort(blockData::allData.begin(),blockData::allData.end());
    for(const blockData& tmpData:blockData::allData){
        if(tmpData.num==1){
            cols=IO::get_empty_table_cols(tmpData.data);
        }
        const int& dataLen=IO::put_single_block_data(cols,tmpData.data);
        blocksLen.push_back(dataLen);
    }
    blockData::allData=vector<blockData>();
    return new table(IO::path_to_name(path),cols,blocksLen);
}

void table::updateFile(const string& path) //根据table.allRecord更新文件内容
{
    if(this->allRecord.empty()){
        return;
    }
    map<int,fstream> fileStore;
    map<int,vector<vector<int>>> lenStore;
    for(record rcd:this->allRecord){
        switch (rcd.type) {
        case ADD:{
            IO::Add(rcd.targetTuple,fileStore,lenStore,path,this->allBlocksLen);
            break;
        }
        case DEL:{
            IO::Del(rcd.opSub,fileStore,lenStore,path,this->allBlocksLen);
            break;
        }
        case MOD:{
            IO::Del(rcd.opSub,fileStore,lenStore,path,this->allBlocksLen);
            IO::Add(rcd.targetTuple,fileStore,lenStore,path,this->allBlocksLen);
            break;
        }
        }
    }
    for(auto iter=fileStore.begin();iter!=fileStore.end();iter++){
        iter->second.flush();
    }
    for(auto iter=lenStore.begin();iter!=lenStore.end();iter++){
        IO::write_to_len_file(IO::path_to_splitpath(path,iter->first),iter->second);
    }
    //全部写入后清除record记录
    this->allRecord.clear();
}