fort/src/ui/db/databasemanager.h

#ifndef DATABASEMANAGER_H
#define DATABASEMANAGER_H

#include <QObject>
#include <QHash>
#include <QStringList>
#include <QVector>

QT_FORWARD_DECLARE_CLASS(SqliteDb)
QT_FORWARD_DECLARE_CLASS(SqliteStmt)

class DatabaseManager : public QObject
{
    Q_OBJECT

public:
    explicit DatabaseManager(const QString &filePath,
                             QObject *parent = nullptr);
    virtual ~DatabaseManager();

    bool initialize();

    SqliteDb *sqliteDb() const { return m_sqliteDb; }

    void addApp(const QString &appPath, bool &isNew);
    void addTraffic(quint16 procCount, const quint8 *procBits,
                    const quint32 *trafBytes);

    void getAppList(QStringList &list);

signals:

public slots:
    qint64 getAppId(const QString &appPath);

    qint32 getMinTrafTime(const char *sql, qint64 appId = 0);

private:
    bool createTables();

    qint64 createAppId(const QString &appPath);

    SqliteStmt *getSqliteStmt(const char *sql);

    void insertTraffic(SqliteStmt *stmt, qint64 appId = 0);
    void updateTraffic(SqliteStmt *stmt, quint32 inBytes,
                       quint32 outBytes, qint64 appId = 0);

private:
    qint32 m_lastTrafHour;
    qint32 m_lastTrafDay;
    qint32 m_lastTrafMonth;

    QString m_filePath;

    SqliteDb *m_sqliteDb;

    QHash<const char *, SqliteStmt *> m_sqliteStmts;

    QVector<qint64> m_appIds;
};

#endif // DATABASEMANAGER_H