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
|
/*
* 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 statResources(const QStringList &resources, const State &state)
{
qint64 total = 0;
for (const auto &resource : resources) {
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) {
state.printLine(QObject::tr("Database: %1").arg(QString(databaseName)), 1);
auto db = transaction.openDatabase(databaseName);
qint64 size = db.getSize() / 1024;
state.printLine(QObject::tr("Size [kb]: %1").arg(size), 1);
total += size;
}
int diskUsage = 0;
QDir dir(Sink::storageLocation());
for (const auto &folder : dir.entryList(QStringList() << resource + "*")) {
diskUsage += Sink::Storage::DataStore(Sink::storageLocation(), folder, Sink::Storage::DataStore::ReadOnly).diskUsage();
}
auto size = diskUsage / 1024;
state.printLine(QObject::tr("Disk usage [kb]: %1").arg(size), 1);
}
state.printLine(QObject::tr("Total [kb]: %1").arg(total));
}
bool statAllResources(State &state)
{
Sink::Query query;
QStringList resources;
for (const auto &r : SinkshUtils::getStore("resource").read(query)) {
resources << r.identifier();
}
statResources(resources, state);
return false;
}
bool stat(const QStringList &args, State &state)
{
if (args.isEmpty()) {
return statAllResources(state);
}
statResources(args, 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)
}
|
