miniSQL/API.cpp at master · 0xKira/miniSQL · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <iostream>
#include "API.h"

using namespace std;

void API::createTable(TableStruct &table) {
    cm.addTable(table);
    bm.createTable(table.tableName);
}

void API::dropTable(string tableName) {
    cm.deleteTable(tableName);
    bm.dropTable(tableName);
}

void API::createIndex(string tableName, string indexName, string attrName) {
    cm.addIndex(tableName, indexName, attrName);
    TableStruct &table = cm.getTable(tableName);
    int i;
    for (i = 0; i < table.attrs.size(); i++)
        if (table.attrs[i].attrName == attrName) break;
    if (i == table.attrs.size())
        cerr << "no such attribute" << endl;
    else
        im.createIndex(table.attrs[i], indexName, table);
}

void API::dropIndex(string indexName) {
    im.dropIndex(indexName);
    string filename("./data/" + indexName + ".index");
    remove(filename.c_str());
}

vector<Tuple> &API::select(TableStruct &table, vector<Condition> &condTable) {
    vector<Tuple> *result = new vector<Tuple>;
    bool flag;
    if (table.hasIndex) {
        flag = rm.selectFromTable(table, condTable, *result);
//        if (condTable.size() == 1 && condTable[0].flag == EQ) {
//            int temp = im.exactSearchWithIndex(im.names[table.tableName], condTable[0].d);
//            vector<int> tempInt;
//            tempInt.push_back(temp);
//            rm.selectFromTableWithIndex(table, condTable, tempInt, *result);
//        } else {
//            vector<int> tempInt;
//            tempInt = im.rangeSearchWithIndex(im.names[table.tableName], condTable[0].d, condTable[condTable.size()].d);
//            rm.selectFromTableWithIndex(table, condTable, tempInt, *result);
//        }
    } else {
        flag = rm.selectFromTable(table, condTable, *result);
    }
    if (!flag) printf("Can't find\n");
    return *result;
}

void API::insertData(TableStruct &table, Tuple &tuple) {
    rm.insertIntoTable(table, tuple);
}

void API::deleteData(TableStruct &table, vector<Condition> &condTable) {
    if (table.hasIndex) {
        rm.deleteFromTable(table, condTable);
//        for (int i = 0; i < condTable.size(); i++) {
//            im.deleteOld(im.names[table.tableName], condTable[0].d);
//        }
    } else {
        rm.deleteFromTable(table, condTable);
    }
    cm.writeback(table);
}