EasyGo/IndexDatabase.h at main · magicdmer/EasyGo · 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
#ifndef INDEXDATABASE_H
#define INDEXDATABASE_H

#include <QObject>
#include <QSqlDatabase>
#include <QSqlError>
#include <QSqlQuery>
#include <QVector>

struct ProgramInfo{
    QString pyname;
    QString name;
    QString path;
};

class IndexDatabase
{
public:
    IndexDatabase();
    IndexDatabase(QString name);
    IndexDatabase(QString connecName, QString dbName);
    ~IndexDatabase();

public:
    bool load();
    bool beginInsert()
    {
        if (!m_database.transaction())
        {
            return false;
        }

        return true;
    }
    bool insert(ProgramInfo& info);
    bool endInsert()
    {
        if (!m_database.commit())
        {
            return false;
        }

        return true;
    }
    bool insert(QVector<ProgramInfo> &infos);
    bool query(QString query,QVector<ProgramInfo>& vecInfo);
    bool clear();

private:
    QSqlDatabase m_database;
};

IndexDatabase* GetIndexDatabase();

#endif // INDEXDATABASE_H