1 | // Copyright (C) 2024 The Qt Company Ltd. |
2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 |
3 | |
4 | #include "qqmljscompilerstats_p.h" |
5 | |
6 | #include <QFile> |
7 | #include <QJsonArray> |
8 | #include <QJsonDocument> |
9 | #include <QJsonObject> |
10 | #include <QTextStream> |
11 | |
12 | QT_BEGIN_NAMESPACE |
13 | |
14 | namespace QQmlJS { |
15 | |
16 | using namespace Qt::StringLiterals; |
17 | |
18 | std::unique_ptr<AotStats> QQmlJSAotCompilerStats::s_instance = std::make_unique<AotStats>(); |
19 | QString QQmlJSAotCompilerStats::s_moduleId; |
20 | bool QQmlJSAotCompilerStats::s_recordAotStats = false; |
21 | |
22 | bool QQmlJS::AotStatsEntry::operator<(const AotStatsEntry &other) const |
23 | { |
24 | if (line == other.line) |
25 | return column < other.column; |
26 | return line < other.line; |
27 | } |
28 | |
29 | void AotStats::insert(const AotStats &other) |
30 | { |
31 | for (const auto &[moduleUri, moduleStats] : other.m_entries.asKeyValueRange()) { |
32 | m_entries[moduleUri].insert(hash: moduleStats); |
33 | } |
34 | } |
35 | |
36 | std::optional<QList<QString>> AotStats::readAllLines(const QString &path) |
37 | { |
38 | QFile aotstatsListFile(path); |
39 | if (!aotstatsListFile.open(flags: QIODevice::ReadOnly | QIODevice::Text)) { |
40 | qDebug().noquote() << u"Could not open \"%1\" for reading"_s .arg(a: aotstatsListFile.fileName()); |
41 | return std::nullopt; |
42 | } |
43 | |
44 | QStringList aotstatsFiles; |
45 | QTextStream stream(&aotstatsListFile); |
46 | while (!stream.atEnd()) |
47 | aotstatsFiles.append(t: stream.readLine()); |
48 | |
49 | return aotstatsFiles; |
50 | } |
51 | |
52 | std::optional<AotStats> AotStats::parseAotstatsFile(const QString &aotstatsPath) |
53 | { |
54 | QFile file(aotstatsPath); |
55 | if (!file.open(flags: QIODevice::ReadOnly | QIODevice::Text)) { |
56 | qDebug().noquote() << u"Could not open \"%1\""_s .arg(a: aotstatsPath); |
57 | return std::nullopt; |
58 | } |
59 | |
60 | return AotStats::fromJsonDocument(QJsonDocument::fromJson(json: file.readAll())); |
61 | } |
62 | |
63 | std::optional<AotStats> AotStats::aggregateAotstatsList(const QString &aotstatsListPath) |
64 | { |
65 | const auto aotstatsFiles = readAllLines(path: aotstatsListPath); |
66 | if (!aotstatsFiles.has_value()) |
67 | return std::nullopt; |
68 | |
69 | AotStats aggregated; |
70 | if (aotstatsFiles->empty()) |
71 | return aggregated; |
72 | |
73 | for (const auto &aotstatsFile : aotstatsFiles.value()) { |
74 | auto parsed = parseAotstatsFile(aotstatsPath: aotstatsFile); |
75 | if (!parsed.has_value()) |
76 | return std::nullopt; |
77 | aggregated.insert(other: parsed.value()); |
78 | } |
79 | |
80 | return aggregated; |
81 | } |
82 | |
83 | AotStats AotStats::fromJsonDocument(const QJsonDocument &document) |
84 | { |
85 | QJsonArray modulesArray = document.array(); |
86 | |
87 | QQmlJS::AotStats result; |
88 | for (const auto &modulesArrayEntry : modulesArray) { |
89 | const auto &moduleObject = modulesArrayEntry.toObject(); |
90 | QString moduleId = moduleObject[u"moduleId" _s].toString(); |
91 | const QJsonArray &filesArray = moduleObject[u"moduleFiles" _s].toArray(); |
92 | |
93 | QHash<QString, QList<AotStatsEntry>> files; |
94 | for (const auto &filesArrayEntry : filesArray) { |
95 | const QJsonObject &fileObject = filesArrayEntry.toObject(); |
96 | QString filepath = fileObject[u"filepath" _s].toString(); |
97 | const QJsonArray &statsArray = fileObject[u"entries" _s].toArray(); |
98 | |
99 | QList<AotStatsEntry> stats; |
100 | for (const auto &statsArrayEntry : statsArray) { |
101 | const auto &statsObject = statsArrayEntry.toObject(); |
102 | QQmlJS::AotStatsEntry stat; |
103 | auto micros = statsObject[u"durationMicroseconds" _s].toInteger(); |
104 | stat.codegenDuration = std::chrono::microseconds(micros); |
105 | stat.functionName = statsObject[u"functionName" _s].toString(); |
106 | stat.errorMessage = statsObject[u"errorMessage" _s].toString(); |
107 | stat.line = statsObject[u"line" _s].toInt(); |
108 | stat.column = statsObject[u"column" _s].toInt(); |
109 | stat.codegenSuccessful = statsObject[u"codegenSuccessfull" _s].toBool(); |
110 | stats.append(t: std::move(stat)); |
111 | } |
112 | |
113 | std::sort(first: stats.begin(), last: stats.end()); |
114 | files[filepath] = stats; |
115 | } |
116 | |
117 | result.m_entries[moduleId] = files; |
118 | } |
119 | |
120 | return result; |
121 | } |
122 | |
123 | QJsonDocument AotStats::toJsonDocument() const |
124 | { |
125 | QJsonArray modulesArray; |
126 | for (auto it1 = m_entries.begin(); it1 != m_entries.end(); ++it1) { |
127 | const QString moduleId = it1.key(); |
128 | const QHash<QString, QList<AotStatsEntry>> &files = it1.value(); |
129 | |
130 | QJsonArray filesArray; |
131 | for (auto it2 = files.begin(); it2 != files.end(); ++it2) { |
132 | const QString &filename = it2.key(); |
133 | const QList<AotStatsEntry> &stats = it2.value(); |
134 | |
135 | QJsonArray statsArray; |
136 | for (const auto &stat : stats) { |
137 | QJsonObject statObject; |
138 | auto micros = static_cast<qint64>(stat.codegenDuration.count()); |
139 | statObject.insert(key: u"durationMicroseconds" , value: micros); |
140 | statObject.insert(key: u"functionName" , value: stat.functionName); |
141 | statObject.insert(key: u"errorMessage" , value: stat.errorMessage); |
142 | statObject.insert(key: u"line" , value: stat.line); |
143 | statObject.insert(key: u"column" , value: stat.column); |
144 | statObject.insert(key: u"codegenSuccessfull" , value: stat.codegenSuccessful); |
145 | statsArray.append(value: statObject); |
146 | } |
147 | |
148 | QJsonObject o; |
149 | o.insert(key: u"filepath"_s , value: filename); |
150 | o.insert(key: u"entries"_s , value: statsArray); |
151 | filesArray.append(value: o); |
152 | } |
153 | |
154 | QJsonObject o; |
155 | o.insert(key: u"moduleId"_s , value: moduleId); |
156 | o.insert(key: u"moduleFiles"_s , value: filesArray); |
157 | modulesArray.append(value: o); |
158 | } |
159 | |
160 | return QJsonDocument(modulesArray); |
161 | } |
162 | |
163 | void AotStats::registerFile(const QString &moduleId, const QString &filepath) |
164 | { |
165 | m_entries[moduleId][filepath] = {}; |
166 | } |
167 | |
168 | void AotStats::addEntry(const QString &moduleId, const QString &filepath, |
169 | const AotStatsEntry &entry) |
170 | { |
171 | m_entries[moduleId][filepath].append(t: entry); |
172 | } |
173 | |
174 | bool AotStats::saveToDisk(const QString &filepath) const |
175 | { |
176 | QFile file(filepath); |
177 | if (!file.open(flags: QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate)) { |
178 | qDebug().noquote() << u"Could not open \"%1\""_s .arg(a: filepath); |
179 | return false; |
180 | } |
181 | |
182 | file.write(data: this->toJsonDocument().toJson(format: QJsonDocument::Indented)); |
183 | return true; |
184 | } |
185 | |
186 | void QQmlJSAotCompilerStats::registerFile(const QString &filepath) |
187 | { |
188 | QQmlJSAotCompilerStats::instance()->registerFile(moduleId: s_moduleId, filepath); |
189 | } |
190 | |
191 | void QQmlJSAotCompilerStats::addEntry(const QString &filepath, const QQmlJS::AotStatsEntry &entry) |
192 | { |
193 | auto *aotstats = QQmlJSAotCompilerStats::instance(); |
194 | aotstats->addEntry(moduleId: s_moduleId, filepath, entry); |
195 | } |
196 | |
197 | } // namespace QQmlJS |
198 | |
199 | QT_END_NAMESPACE |
200 | |