1 //===-- StreamTee.h ------------------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef liblldb_StreamTee_h_
11 #define liblldb_StreamTee_h_
17 #include "lldb/Utility/Stream.h"
19 namespace lldb_private {
21 class StreamTee : public Stream {
23 StreamTee() : Stream(), m_streams_mutex(), m_streams() {}
25 StreamTee(lldb::StreamSP &stream_sp)
26 : Stream(), m_streams_mutex(), m_streams() {
27 // No need to lock mutex during construction
29 m_streams.push_back(stream_sp);
32 StreamTee(lldb::StreamSP &stream_sp, lldb::StreamSP &stream_2_sp)
33 : Stream(), m_streams_mutex(), m_streams() {
34 // No need to lock mutex during construction
36 m_streams.push_back(stream_sp);
38 m_streams.push_back(stream_2_sp);
41 StreamTee(const StreamTee &rhs)
42 : Stream(rhs), m_streams_mutex(), m_streams() {
43 // Don't copy until we lock down "rhs"
44 std::lock_guard<std::recursive_mutex> guard(rhs.m_streams_mutex);
45 m_streams = rhs.m_streams;
48 ~StreamTee() override {}
50 StreamTee &operator=(const StreamTee &rhs) {
52 Stream::operator=(rhs);
53 std::lock_guard<std::recursive_mutex> lhs_locker(m_streams_mutex);
54 std::lock_guard<std::recursive_mutex> rhs_locker(rhs.m_streams_mutex);
55 m_streams = rhs.m_streams;
60 void Flush() override {
61 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
62 collection::iterator pos, end;
63 for (pos = m_streams.begin(), end = m_streams.end(); pos != end; ++pos) {
64 // Allow for our collection to contain NULL streams. This allows
65 // the StreamTee to be used with hard coded indexes for clients
66 // that might want N total streams with only a few that are set
68 Stream *strm = pos->get();
74 size_t Write(const void *s, size_t length) override {
75 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
76 if (m_streams.empty())
79 size_t min_bytes_written = SIZE_MAX;
80 collection::iterator pos, end;
81 for (pos = m_streams.begin(), end = m_streams.end(); pos != end; ++pos) {
82 // Allow for our collection to contain NULL streams. This allows
83 // the StreamTee to be used with hard coded indexes for clients
84 // that might want N total streams with only a few that are set
86 Stream *strm = pos->get();
88 const size_t bytes_written = strm->Write(s, length);
89 if (min_bytes_written > bytes_written)
90 min_bytes_written = bytes_written;
93 if (min_bytes_written == SIZE_MAX)
95 return min_bytes_written;
98 size_t AppendStream(const lldb::StreamSP &stream_sp) {
99 size_t new_idx = m_streams.size();
100 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
101 m_streams.push_back(stream_sp);
105 size_t GetNumStreams() const {
108 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
109 result = m_streams.size();
114 lldb::StreamSP GetStreamAtIndex(uint32_t idx) {
115 lldb::StreamSP stream_sp;
116 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
117 if (idx < m_streams.size())
118 stream_sp = m_streams[idx];
122 void SetStreamAtIndex(uint32_t idx, const lldb::StreamSP &stream_sp) {
123 std::lock_guard<std::recursive_mutex> guard(m_streams_mutex);
124 // Resize our stream vector as necessary to fit as many streams
125 // as needed. This also allows this class to be used with hard
126 // coded indexes that can be used contain many streams, not all
127 // of which are valid.
128 if (idx >= m_streams.size())
129 m_streams.resize(idx + 1);
130 m_streams[idx] = stream_sp;
134 typedef std::vector<lldb::StreamSP> collection;
135 mutable std::recursive_mutex m_streams_mutex;
136 collection m_streams;
139 } // namespace lldb_private
141 #endif // liblldb_StreamTee_h_