forked from OSchip/llvm-project
50 lines
1.5 KiB
C++
50 lines
1.5 KiB
C++
//===-- TestTracer.h --------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Allows setting up a fake tracer for tests.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_TOOLS_EXTRA_UNITTESTS_CLANGD_SUPPORT_TESTTRACER_H
|
|
#define LLVM_CLANG_TOOLS_EXTRA_UNITTESTS_CLANGD_SUPPORT_TESTTRACER_H
|
|
|
|
#include "support/Trace.h"
|
|
#include "llvm/ADT/StringMap.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace clang {
|
|
namespace clangd {
|
|
namespace trace {
|
|
|
|
/// A RAII Tracer that can be used by tests.
|
|
class TestTracer : public EventTracer {
|
|
public:
|
|
TestTracer() : S(*this) {}
|
|
/// Stores all the measurements to be returned with take later on.
|
|
void record(const Metric &Metric, double Value,
|
|
llvm::StringRef Label) override;
|
|
|
|
/// Returns recorded measurements for \p Metric and clears them.
|
|
std::vector<double> takeMetric(llvm::StringRef Metric,
|
|
llvm::StringRef Label = "");
|
|
|
|
private:
|
|
std::mutex Mu;
|
|
/// Measurements recorded per metric per label.
|
|
llvm::StringMap<llvm::StringMap<std::vector<double>>> Measurements;
|
|
Session S;
|
|
};
|
|
|
|
} // namespace trace
|
|
} // namespace clangd
|
|
} // namespace clang
|
|
#endif
|