dbs.h~

来自「简单的基于文件的数据库小程序」· H~ 代码 · 共 95 行

#ifndef _DBS_H_
#define _DBS_H_

#include <vector>
#include "Tool.h"
#include "db.h"

using namespace std; 

class dbs
{
public:
    std::vector<db*> databases;
    dbs()
    {
    databases = std::vector<db*> (0);
    vector<char*> buffer;
    if(hasFile("root")) {
        int fp=openFile("root");
        getNextLine(fp);
        getNextLine(fp);
	char* temp=getNextLine(fp);
        while(temp!=NULL)
	{
	    //char* temp=getNextLine(fd);
	   
	    if(strcmp(temp,"")!=0){
		
		buffer.push_back(temp);
		
		}
		
	    
	    temp = getNextLine(fp);
	    
	    
                     
        }
        
      closeFile(fp);
      for(uint i=0;i<buffer.size();i++){
			databases.push_back(new db(buffer[i]));
	}
      }
    else return;
    }
    
    
    void add(const char* name)
    {
    databases.push_back(new db(name));
    addLineToFile("root",name);
	checkFileExist(name);
    }
    
    
    void del(const char* name)
    {
	deleteOneLine("root",name);
	 int fd = openFile(name);
	 char* buffer= new char[256];
	 while((buffer = getNextLine(fd))!=NULL)
	 {
	     //std::cout<<buffer<<endl;
	     if(buffer!="")
	     deleteFile(buffer);
	     
	 }
	 closeFile(fd);
	
	 
	deleteFile(name);
	
	vector<db *>::iterator it=databases.begin();
	for(;it!=databases.end()+1;++it){
	    //cout<<(*it)->dbName<<"*****"<<endl;
	    if(strcmp((*it)->dbName,name)==0){
		it=databases.erase(it);
		//cout<<"fffff"<<endl;
		databases.resize(0);
		break;
	    }
        }
	
	delete buffer;
    }
        
  virtual ~dbs()
    {
    
    }
    
};
#endif