1 //===- FDRRecordConsumer.h - XRay Flight Data Recorder Mode Records -------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 #ifndef LLVM_XRAY_FDRRECORDCONSUMER_H 9 #define LLVM_XRAY_FDRRECORDCONSUMER_H 10 11 #include "llvm/Support/Error.h" 12 #include "llvm/XRay/FDRRecords.h" 13 #include <algorithm> 14 #include <memory> 15 #include <vector> 16 17 namespace llvm { 18 namespace xray { 19 20 class RecordConsumer { 21 public: 22 virtual Error consume(std::unique_ptr<Record> R) = 0; 23 virtual ~RecordConsumer() = default; 24 }; 25 26 // This consumer will collect all the records into a vector of records, in 27 // arrival order. 28 class LogBuilderConsumer : public RecordConsumer { 29 std::vector<std::unique_ptr<Record>> &Records; 30 31 public: LogBuilderConsumer(std::vector<std::unique_ptr<Record>> & R)32 explicit LogBuilderConsumer(std::vector<std::unique_ptr<Record>> &R) 33 : Records(R) {} 34 35 Error consume(std::unique_ptr<Record> R) override; 36 }; 37 38 // A PipelineConsumer applies a set of visitors to every consumed Record, in the 39 // order by which the visitors are added to the pipeline in the order of 40 // appearance. 41 class PipelineConsumer : public RecordConsumer { 42 std::vector<RecordVisitor *> Visitors; 43 44 public: PipelineConsumer(std::initializer_list<RecordVisitor * > V)45 PipelineConsumer(std::initializer_list<RecordVisitor *> V) : Visitors(V) {} 46 47 Error consume(std::unique_ptr<Record> R) override; 48 }; 49 50 } // namespace xray 51 } // namespace llvm 52 53 #endif // LLVM_XRAY_FDRRECORDCONSUMER_H 54