summaryrefslogtreecommitdiffstats
path: root/sinksh/syntax_modules/sink_stat.cpp
blob: f263bbc7fe2a83138b8fac15da17a42f64c3db80 (plain)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * Copyright (C) 2014 Aaron Seigo <aseigo@kde.org>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the
 *   Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
 */

#include <QDebug>
#include <QObject> // tr()
#include <QTimer>
#include <QDir>

#include "common/resource.h"
#include "common/storage.h"
#include "common/resourceconfig.h"
#include "common/log.h"
#include "common/storage.h"
#include "common/definitions.h"

#include "sinksh_utils.h"
#include "state.h"
#include "syntaxtree.h"

namespace SinkStat
{

void statResource(const QString &resource, const State &state)
{
    qint64 total = 0;
    Sink::Storage::DataStore storage(Sink::storageLocation(), resource, Sink::Storage::DataStore::ReadOnly);
    auto transaction = storage.createTransaction(Sink::Storage::DataStore::ReadOnly);

    QList<QByteArray> databases = transaction.getDatabaseNames();
    for (const auto &databaseName : databases) {
        auto db = transaction.openDatabase(databaseName);
        qint64 size = db.getSize() / 1024;
        state.printLine(QObject::tr("%1:\t%2 [kb]").arg(QString(databaseName)).arg(size), 1);
        total += size;
    }
    state.printLine();
    state.printLine(QObject::tr("Calculated named database sizes total of main database: %1 [kb]").arg(total), 1);
    state.printLine(QObject::tr("Write amplification of main database: %1").arg(float(storage.diskUsage() / 1024)/float(total)), 1);
    int diskUsage = 0;

    QDir dir(Sink::storageLocation());
    for (const auto &folder : dir.entryList(QStringList() << resource + "*")) {
        auto size = Sink::Storage::DataStore(Sink::storageLocation(), folder, Sink::Storage::DataStore::ReadOnly).diskUsage();
        diskUsage += size;
        state.printLine(QObject::tr("...Accumulating %1: %2 [kb]").arg(folder).arg(size / 1024), 1);
    }
    auto size = diskUsage / 1024;
    state.printLine(QObject::tr("Actual database file sizes total: %1 [kb]").arg(size), 1);

    QDir dataDir{Sink::resourceStorageLocation(resource.toLatin1()) + "/blob/"};
    Q_ASSERT(dataDir.exists());
    qint64 dataSize = 0;
    for (const auto &e : dataDir.entryInfoList(QDir::Files | QDir::NoSymLinks | QDir::NoDotAndDotDot)) {
        dataSize += e.size();
    }
    state.printLine(QObject::tr("Total BLOB size [kb]: %1").arg(dataSize / 1024), 1);
    state.printLine();
}

bool statAllResources(State &state)
{
    Sink::Query query;
    for (const auto &r : SinkshUtils::getStore("resource").read(query)) {
        statResource(SinkshUtils::parseUid(r.identifier()), state);
    }
    return false;
}

bool stat(const QStringList &args, State &state)
{
    if (args.isEmpty()) {
        return statAllResources(state);
    }

    for (const auto &r : args) {
        statResource(SinkshUtils::parseUid(r.toUtf8()), state);
    }
    return false;
}

Syntax::List syntax()
{
    Syntax state("stat", QObject::tr("Shows database usage for the resources requested"), &SinkStat::stat, Syntax::NotInteractive);
    state.completer = &SinkshUtils::resourceCompleter;

    return Syntax::List() << state;
}

REGISTER_SYNTAX(SinkStat)

}