Files
carbon-lang/toolchain/diagnostics/sorting_diagnostic_consumer.h
T
Jon Ross-Perkins 93e40289dd Collapse diagnostic errors and notes into a single vector. (#3805)
On #3792 it was requested to put context-related diagnostic messages
prior to the diagnostic error. In order to support that, remove the
distinction that an error needs to come first on DiagnosticMessage.
2024-03-21 23:52:51 +00:00

62 lines
2.3 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#ifndef CARBON_TOOLCHAIN_DIAGNOSTICS_SORTING_DIAGNOSTIC_CONSUMER_H_
#define CARBON_TOOLCHAIN_DIAGNOSTICS_SORTING_DIAGNOSTIC_CONSUMER_H_
#include "common/check.h"
#include "llvm/ADT/STLExtras.h"
#include "toolchain/diagnostics/diagnostic_emitter.h"
namespace Carbon {
// Buffers incoming diagnostics for printing and sorting.
class SortingDiagnosticConsumer : public DiagnosticConsumer {
public:
explicit SortingDiagnosticConsumer(DiagnosticConsumer& next_consumer)
: next_consumer_(&next_consumer) {}
~SortingDiagnosticConsumer() override {
// We choose not to automatically flush diagnostics here, because they are
// likely to refer to data that gets destroyed before the diagnostics
// consumer is destroyed, because the diagnostics consumer is typically
// created before the objects that diagnostics refer into are created.
CARBON_CHECK(diagnostics_.empty())
<< "Must flush diagnostics consumer before destroying it";
}
// Buffers the diagnostic.
auto HandleDiagnostic(Diagnostic diagnostic) -> void override {
diagnostics_.push_back(std::move(diagnostic));
}
// Sorts and flushes buffered diagnostics.
void Flush() override {
llvm::stable_sort(diagnostics_,
[](const Diagnostic& lhs, const Diagnostic& rhs) {
const auto& lhs_loc = lhs.messages[0].location;
const auto& rhs_loc = rhs.messages[0].location;
return std::tie(lhs_loc.filename, lhs_loc.line_number,
lhs_loc.column_number) <
std::tie(rhs_loc.filename, rhs_loc.line_number,
rhs_loc.column_number);
});
for (auto& diag : diagnostics_) {
next_consumer_->HandleDiagnostic(std::move(diag));
}
diagnostics_.clear();
}
private:
// A Diagnostic is undesirably large for inline storage by SmallVector, so we
// specify 0.
llvm::SmallVector<Diagnostic, 0> diagnostics_;
DiagnosticConsumer* next_consumer_;
};
} // namespace Carbon
#endif // CARBON_TOOLCHAIN_DIAGNOSTICS_SORTING_DIAGNOSTIC_CONSUMER_H_