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
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
* 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/domain/event.h"
#include "common/domain/folder.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 storage(Sink::storageLocation(), resource, Sink::Storage::ReadOnly);
auto transaction = storage.createTransaction(Sink::Storage::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(Sink::storageLocation(), folder, Sink::Storage::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;
query.liveQuery = false;
auto model = SinkshUtils::loadModel("resource", query);
//SUUUPER ugly, but can't think of a better way with 2 glasses of wine in me on Christmas day
static QStringList resources;
resources.clear();
QObject::connect(model.data(), &QAbstractItemModel::rowsInserted, [model](const QModelIndex &index, int start, int end) mutable {
for (int i = start; i <= end; i++) {
auto object = model->data(model->index(i, 0, index), Sink::Store::DomainObjectBaseRole).value<Sink::ApplicationDomain::ApplicationDomainType::Ptr>();
resources << object->identifier();
}
});
QObject::connect(model.data(), &QAbstractItemModel::dataChanged, [model, state](const QModelIndex &, const QModelIndex &, const QVector<int> &roles) {
if (roles.contains(Sink::Store::ChildrenFetchedRole)) {
statResources(resources, state);
state.commandFinished();
}
});
if (!model->data(QModelIndex(), Sink::Store::ChildrenFetchedRole).toBool()) {
return true;
}
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::EventDriven);
state.completer = &SinkshUtils::resourceCompleter;
return Syntax::List() << state;
}
REGISTER_SYNTAX(SinkStat)
}
|
