Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(98)

Side by Side Diff: webrtc/logging/rtc_event_log/rtc_event_log.cc

Issue 3007473002: Make RtcEventLogImpl to use a TaskQueue instead of a helper-thread (Closed)
Patch Set: s/ph/pb Created 3 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source 5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include "webrtc/logging/rtc_event_log/rtc_event_log.h" 11 #include "webrtc/logging/rtc_event_log/rtc_event_log.h"
12 12
13 #include <atomic>
14 #include <deque>
15 #include <functional>
13 #include <limits> 16 #include <limits>
17 #include <memory>
14 #include <utility> 18 #include <utility>
15 #include <vector> 19 #include <vector>
16 20
17 #include "webrtc/logging/rtc_event_log/rtc_event_log_helper_thread.h"
18 #include "webrtc/modules/audio_coding/audio_network_adaptor/include/audio_networ k_adaptor.h" 21 #include "webrtc/modules/audio_coding/audio_network_adaptor/include/audio_networ k_adaptor.h"
19 #include "webrtc/modules/remote_bitrate_estimator/include/bwe_defines.h" 22 #include "webrtc/modules/remote_bitrate_estimator/include/bwe_defines.h"
20 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" 23 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h"
21 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" 24 #include "webrtc/modules/rtp_rtcp/source/byte_io.h"
22 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/app.h" 25 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/app.h"
23 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/bye.h" 26 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/bye.h"
24 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/common_header.h" 27 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/common_header.h"
25 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/extended_jitter_report.h" 28 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/extended_jitter_report.h"
26 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/extended_reports.h" 29 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/extended_reports.h"
27 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/psfb.h" 30 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/psfb.h"
28 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/receiver_report.h" 31 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/receiver_report.h"
29 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/rtpfb.h" 32 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/rtpfb.h"
30 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sdes.h" 33 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sdes.h"
31 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sender_report.h" 34 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/sender_report.h"
32 #include "webrtc/rtc_base/atomicops.h"
33 #include "webrtc/rtc_base/checks.h" 35 #include "webrtc/rtc_base/checks.h"
34 #include "webrtc/rtc_base/constructormagic.h" 36 #include "webrtc/rtc_base/constructormagic.h"
35 #include "webrtc/rtc_base/event.h" 37 #include "webrtc/rtc_base/event.h"
38 #include "webrtc/rtc_base/ignore_wundef.h"
36 #include "webrtc/rtc_base/logging.h" 39 #include "webrtc/rtc_base/logging.h"
37 #include "webrtc/rtc_base/protobuf_utils.h" 40 #include "webrtc/rtc_base/protobuf_utils.h"
38 #include "webrtc/rtc_base/swap_queue.h" 41 #include "webrtc/rtc_base/ptr_util.h"
39 #include "webrtc/rtc_base/thread_checker.h" 42 #include "webrtc/rtc_base/sequenced_task_checker.h"
43 #include "webrtc/rtc_base/task_queue.h"
44 #include "webrtc/rtc_base/thread_annotations.h"
40 #include "webrtc/rtc_base/timeutils.h" 45 #include "webrtc/rtc_base/timeutils.h"
41 #include "webrtc/system_wrappers/include/file_wrapper.h" 46 #include "webrtc/system_wrappers/include/file_wrapper.h"
47 #include "webrtc/typedefs.h"
42 48
43 #ifdef ENABLE_RTC_EVENT_LOG 49 #ifdef ENABLE_RTC_EVENT_LOG
44 // *.pb.h files are generated at build-time by the protobuf compiler. 50 // *.pb.h files are generated at build-time by the protobuf compiler.
51 RTC_PUSH_IGNORING_WUNDEF()
45 #ifdef WEBRTC_ANDROID_PLATFORM_BUILD 52 #ifdef WEBRTC_ANDROID_PLATFORM_BUILD
46 #include "external/webrtc/webrtc/logging/rtc_event_log/rtc_event_log.pb.h" 53 #include "external/webrtc/webrtc/logging/rtc_event_log/rtc_event_log.pb.h"
47 #else 54 #else
48 #include "webrtc/logging/rtc_event_log/rtc_event_log.pb.h" 55 #include "webrtc/logging/rtc_event_log/rtc_event_log.pb.h"
49 #endif 56 #endif
57 RTC_POP_IGNORING_WUNDEF()
50 #endif 58 #endif
51 59
52 namespace webrtc { 60 namespace webrtc {
53 61
54 #ifdef ENABLE_RTC_EVENT_LOG 62 #ifdef ENABLE_RTC_EVENT_LOG
55 63
64 namespace {
65 const int kEventsInHistory = 10000;
66
67 bool IsConfigEvent(const rtclog::Event& event) {
68 rtclog::Event_EventType event_type = event.type();
69 return event_type == rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT ||
70 event_type == rtclog::Event::VIDEO_SENDER_CONFIG_EVENT ||
71 event_type == rtclog::Event::AUDIO_RECEIVER_CONFIG_EVENT ||
72 event_type == rtclog::Event::AUDIO_SENDER_CONFIG_EVENT;
73 }
74
75 // This class exists because C++11 doesn't allow transferring a unique_ptr to
76 // a lambda (a copy constructor is required).
77 class RtcEventLogTask final : public rtc::QueuedTask {
nisse-webrtc 2017/08/30 12:18:35 You could templateize this and the below class, on
eladalon 2017/08/30 13:01:18 Good suggestion - I think it would actually, oddly
78 public:
79 RtcEventLogTask(std::unique_ptr<rtclog::Event> event,
80 std::function<void(std::unique_ptr<rtclog::Event>)> handler)
81 : event_(std::move(event)), handler_(handler) {}
82
83 bool Run() override {
84 handler_(std::move(event_));
85 return true;
86 }
87
88 private:
89 std::unique_ptr<rtclog::Event> event_;
90 std::function<void(std::unique_ptr<rtclog::Event>)> handler_;
91 };
92
93 // This class exists because C++11 doesn't allow transferring a unique_ptr to
94 // a lambda (a copy constructor is required).
95 class RtcEventLogFileTask final : public rtc::QueuedTask {
96 public:
97 RtcEventLogFileTask(std::unique_ptr<FileWrapper> file,
98 std::function<void(std::unique_ptr<FileWrapper>)> handler)
99 : file_(std::move(file)),
100 handler_(handler) {}
101
102 bool Run() override {
103 handler_(std::move(file_));
104 return true;
105 }
106
107 private:
108 std::unique_ptr<FileWrapper> file_;
109 std::function<void(std::unique_ptr<FileWrapper>)> handler_;
110 };
111 } // namespace
112
56 class RtcEventLogImpl final : public RtcEventLog { 113 class RtcEventLogImpl final : public RtcEventLog {
57 friend std::unique_ptr<RtcEventLog> RtcEventLog::Create(); 114 friend std::unique_ptr<RtcEventLog> RtcEventLog::Create();
58 115
59 public: 116 public:
60 ~RtcEventLogImpl() override; 117 ~RtcEventLogImpl() override;
61 118
62 bool StartLogging(const std::string& file_name, 119 bool StartLogging(const std::string& file_name,
63 int64_t max_size_bytes) override; 120 int64_t max_size_bytes) override;
64 bool StartLogging(rtc::PlatformFile platform_file, 121 bool StartLogging(rtc::PlatformFile platform_file,
65 int64_t max_size_bytes) override; 122 int64_t max_size_bytes) override;
(...skipping 22 matching lines...) Expand all
88 const AudioEncoderRuntimeConfig& config) override; 145 const AudioEncoderRuntimeConfig& config) override;
89 void LogProbeClusterCreated(int id, 146 void LogProbeClusterCreated(int id,
90 int bitrate_bps, 147 int bitrate_bps,
91 int min_probes, 148 int min_probes,
92 int min_bytes) override; 149 int min_bytes) override;
93 void LogProbeResultSuccess(int id, int bitrate_bps) override; 150 void LogProbeResultSuccess(int id, int bitrate_bps) override;
94 void LogProbeResultFailure(int id, 151 void LogProbeResultFailure(int id,
95 ProbeFailureReason failure_reason) override; 152 ProbeFailureReason failure_reason) override;
96 153
97 private: 154 private:
98 // Private constructor to ensure that creation is done by RtcEventLog::Create. 155 void StartLoggingInternal(std::unique_ptr<FileWrapper> file,
99 RtcEventLogImpl(); 156 int64_t max_size_bytes);
157
158 RtcEventLogImpl(); // Creation is done by RtcEventLog::Create.
100 159
101 void StoreEvent(std::unique_ptr<rtclog::Event> event); 160 void StoreEvent(std::unique_ptr<rtclog::Event> event);
102 void LogProbeResult(int id, 161 void LogProbeResult(int id,
103 rtclog::BweProbeResult::ResultType result, 162 rtclog::BweProbeResult::ResultType result,
104 int bitrate_bps); 163 int bitrate_bps);
105 164
106 static volatile int log_count_; 165 // Attempts to append an event to the output protobuf string, while observing
166 // the (potential) limitation on file size. (False is returned when observing
167 // the limitation disallows appending the event.)
nisse-webrtc 2017/08/30 12:18:35 Suggested rephrasing of this comment: Appends an
eladalon 2017/08/30 13:01:18 Done.
168 bool AppendEventToString(rtclog::Event* event,
169 ProtoString* output_string) RTC_WARN_UNUSED_RESULT;
nisse-webrtc 2017/08/30 12:18:36 How do you expect callers to handle errors? If t
eladalon 2017/08/30 13:01:18 This is not an error, this is an indication that a
nisse-webrtc 2017/08/30 13:13:23 You could move that responsibility to this class,
eladalon 2017/08/30 14:15:44 It would indeed be better, because now we still ha
nisse-webrtc 2017/08/30 14:27:01 It's just that I don't think RTC_WARN_UNUSED_RESUL
terelius 2017/08/30 15:06:04 Nisse, I think it is fine to land this CL as-is. T
eladalon 2017/08/30 15:19:20 1. StoreEvent -> LogToFile -> AppendEventToString
107 170
108 // Message queue for passing control messages to the logging thread. 171 void LogToMemory(std::unique_ptr<rtclog::Event> event);
109 SwapQueue<RtcEventLogHelperThread::ControlMessage> message_queue_;
110 172
111 // Message queue for passing events to the logging thread. 173 void StartLogFile();
112 SwapQueue<std::unique_ptr<rtclog::Event> > event_queue_; 174 void LogToFile(std::unique_ptr<rtclog::Event> event);
175 void StopLogFile(int64_t stop_time);
113 176
114 RtcEventLogHelperThread helper_thread_; 177 // TODO(eladalon): Known issue - there's a race over |log_count_|.
115 rtc::ThreadChecker thread_checker_; 178 static std::atomic<int> log_count_;
nisse-webrtc 2017/08/30 12:18:35 I'm not sure use of std::atomic is allowed yet.
eladalon 2017/08/30 13:01:18 1. I think it's allowed. I'll send you a relevant
nisse-webrtc 2017/08/30 13:13:23 I've checked the Chromium style guide, it's allowe
179
180 rtc::SequencedTaskChecker worker_sequence_checker_;
181 rtc::SequencedTaskChecker owner_sequence_checker_;
182
183 // History containing all past configuration events.
184 std::vector<std::unique_ptr<rtclog::Event>> config_history_
185 GUARDED_BY(worker_sequence_checker_);
186
187 // History containing the most recent (non-configuration) events (~ 10 s).
188 std::deque<std::unique_ptr<rtclog::Event>> history_
189 GUARDED_BY(worker_sequence_checker_);
190
191 std::unique_ptr<FileWrapper> file_ GUARDED_BY(worker_sequence_checker_);
192
193 size_t max_size_bytes_ GUARDED_BY(worker_sequence_checker_);
194 size_t written_bytes_ GUARDED_BY(worker_sequence_checker_);
195
196 // Keep this last to ensure it destructs first, or else tasks living on the
197 // queue might access other members after they've been torn down.
198 rtc::TaskQueue task_queue_;
116 199
117 RTC_DISALLOW_COPY_AND_ASSIGN(RtcEventLogImpl); 200 RTC_DISALLOW_COPY_AND_ASSIGN(RtcEventLogImpl);
118 }; 201 };
119 202
120 namespace { 203 namespace {
121 // The functions in this namespace convert enums from the runtime format 204 // The functions in this namespace convert enums from the runtime format
122 // that the rest of the WebRtc project can use, to the corresponding 205 // that the rest of the WebRtc project can use, to the corresponding
123 // serialized enum which is defined by the protobuf. 206 // serialized enum which is defined by the protobuf.
124 207
125 rtclog::VideoReceiveConfig_RtcpMode ConvertRtcpMode(RtcpMode rtcp_mode) { 208 rtclog::VideoReceiveConfig_RtcpMode ConvertRtcpMode(RtcpMode rtcp_mode) {
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after
157 return rtclog::BweProbeResult::INVALID_SEND_RECEIVE_INTERVAL; 240 return rtclog::BweProbeResult::INVALID_SEND_RECEIVE_INTERVAL;
158 case kInvalidSendReceiveRatio: 241 case kInvalidSendReceiveRatio:
159 return rtclog::BweProbeResult::INVALID_SEND_RECEIVE_RATIO; 242 return rtclog::BweProbeResult::INVALID_SEND_RECEIVE_RATIO;
160 case kTimeout: 243 case kTimeout:
161 return rtclog::BweProbeResult::TIMEOUT; 244 return rtclog::BweProbeResult::TIMEOUT;
162 } 245 }
163 RTC_NOTREACHED(); 246 RTC_NOTREACHED();
164 return rtclog::BweProbeResult::SUCCESS; 247 return rtclog::BweProbeResult::SUCCESS;
165 } 248 }
166 249
167 // The RTP and RTCP buffers reserve space for twice the expected number of
168 // sent packets because they also contain received packets.
169 static const int kEventsPerSecond = 1000;
170 static const int kControlMessagesPerSecond = 10;
171 } // namespace 250 } // namespace
172 251
173 volatile int RtcEventLogImpl::log_count_ = 0; 252 std::atomic<int> RtcEventLogImpl::log_count_(0);
174 253
175 // RtcEventLogImpl member functions.
176 RtcEventLogImpl::RtcEventLogImpl() 254 RtcEventLogImpl::RtcEventLogImpl()
177 // Allocate buffers for roughly one second of history. 255 : file_(FileWrapper::Create()),
178 : message_queue_(kControlMessagesPerSecond), 256 max_size_bytes_(std::numeric_limits<decltype(max_size_bytes_)>::max()),
179 event_queue_(kEventsPerSecond), 257 written_bytes_(0),
180 helper_thread_(&message_queue_, &event_queue_), 258 task_queue_("rtc_event_log") {
181 thread_checker_() { 259 worker_sequence_checker_.Detach();
182 thread_checker_.DetachFromThread();
183 } 260 }
184 261
185 RtcEventLogImpl::~RtcEventLogImpl() { 262 RtcEventLogImpl::~RtcEventLogImpl() {
186 // The RtcEventLogHelperThread destructor closes the file 263 RTC_DCHECK_CALLED_SEQUENTIALLY(&owner_sequence_checker_);
187 // and waits for the thread to terminate. 264
188 int count = rtc::AtomicOps::Decrement(&RtcEventLogImpl::log_count_); 265 // If we're logging to the file, this will stop that. Blocking function.
266 StopLogging();
267
268 int count = std::atomic_fetch_sub(&RtcEventLogImpl::log_count_, 1) - 1;
189 RTC_DCHECK_GE(count, 0); 269 RTC_DCHECK_GE(count, 0);
190 } 270 }
191 271
192 bool RtcEventLogImpl::StartLogging(const std::string& file_name, 272 bool RtcEventLogImpl::StartLogging(const std::string& file_name,
193 int64_t max_size_bytes) { 273 int64_t max_size_bytes) {
194 RTC_DCHECK(thread_checker_.CalledOnValidThread()); 274 RTC_DCHECK_CALLED_SEQUENTIALLY(&owner_sequence_checker_);
195 RtcEventLogHelperThread::ControlMessage message; 275
196 message.message_type = RtcEventLogHelperThread::ControlMessage::START_FILE; 276 auto file = rtc::WrapUnique<FileWrapper>(FileWrapper::Create());
197 message.max_size_bytes = max_size_bytes <= 0 277 if (!file->OpenFile(file_name.c_str(), false)) {
198 ? std::numeric_limits<int64_t>::max()
199 : max_size_bytes;
200 message.start_time = rtc::TimeMicros();
201 message.stop_time = std::numeric_limits<int64_t>::max();
202 message.file.reset(FileWrapper::Create());
203 if (!message.file->OpenFile(file_name.c_str(), false)) {
204 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started."; 278 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started.";
205 return false; 279 return false;
206 } 280 }
207 if (!message_queue_.Insert(&message)) { 281
208 LOG(LS_ERROR) << "Message queue full. Can't start logging."; 282 StartLoggingInternal(std::move(file), max_size_bytes);
209 return false; 283
210 }
211 helper_thread_.SignalNewEvent();
212 LOG(LS_INFO) << "Starting WebRTC event log.";
213 return true; 284 return true;
214 } 285 }
215 286
216 bool RtcEventLogImpl::StartLogging(rtc::PlatformFile platform_file, 287 bool RtcEventLogImpl::StartLogging(rtc::PlatformFile platform_file,
217 int64_t max_size_bytes) { 288 int64_t max_size_bytes) {
218 RTC_DCHECK(thread_checker_.CalledOnValidThread()); 289 RTC_DCHECK_CALLED_SEQUENTIALLY(&owner_sequence_checker_);
219 RtcEventLogHelperThread::ControlMessage message; 290
220 message.message_type = RtcEventLogHelperThread::ControlMessage::START_FILE; 291 auto file = rtc::WrapUnique<FileWrapper>(FileWrapper::Create());
221 message.max_size_bytes = max_size_bytes <= 0
222 ? std::numeric_limits<int64_t>::max()
223 : max_size_bytes;
224 message.start_time = rtc::TimeMicros();
225 message.stop_time = std::numeric_limits<int64_t>::max();
226 message.file.reset(FileWrapper::Create());
227 FILE* file_handle = rtc::FdopenPlatformFileForWriting(platform_file); 292 FILE* file_handle = rtc::FdopenPlatformFileForWriting(platform_file);
228 if (!file_handle) { 293 if (!file_handle) {
229 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started."; 294 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started.";
230 // Even though we failed to open a FILE*, the platform_file is still open 295 // Even though we failed to open a FILE*, the platform_file is still open
231 // and needs to be closed. 296 // and needs to be closed.
232 if (!rtc::ClosePlatformFile(platform_file)) { 297 if (!rtc::ClosePlatformFile(platform_file)) {
233 LOG(LS_ERROR) << "Can't close file."; 298 LOG(LS_ERROR) << "Can't close file.";
234 } 299 }
235 return false; 300 return false;
236 } 301 }
237 if (!message.file->OpenFromFileHandle(file_handle)) { 302 if (!file->OpenFromFileHandle(file_handle)) {
238 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started."; 303 LOG(LS_ERROR) << "Can't open file. WebRTC event log not started.";
239 return false; 304 return false;
240 } 305 }
241 if (!message_queue_.Insert(&message)) { 306
242 LOG(LS_ERROR) << "Message queue full. Can't start logging."; 307 StartLoggingInternal(std::move(file), max_size_bytes);
243 return false; 308
244 }
245 helper_thread_.SignalNewEvent();
246 LOG(LS_INFO) << "Starting WebRTC event log.";
247 return true; 309 return true;
248 } 310 }
249 311
250 void RtcEventLogImpl::StopLogging() { 312 void RtcEventLogImpl::StopLogging() {
251 RTC_DCHECK(thread_checker_.CalledOnValidThread()); 313 RTC_DCHECK_CALLED_SEQUENTIALLY(&owner_sequence_checker_);
252 RtcEventLogHelperThread::ControlMessage message; 314
253 message.message_type = RtcEventLogHelperThread::ControlMessage::STOP_FILE;
254 message.stop_time = rtc::TimeMicros();
255 while (!message_queue_.Insert(&message)) {
256 // TODO(terelius): We would like to have a blocking Insert function in the
257 // SwapQueue, but for the time being we will just clear any previous
258 // messages.
259 // Since StopLogging waits for the thread, it is essential that we don't
260 // clear any STOP_FILE messages. To ensure that there is only one call at a
261 // time, we require that all calls to StopLogging are made on the same
262 // thread.
263 LOG(LS_ERROR) << "Message queue full. Clearing queue to stop logging.";
264 message_queue_.Clear();
265 }
266 LOG(LS_INFO) << "Stopping WebRTC event log."; 315 LOG(LS_INFO) << "Stopping WebRTC event log.";
267 helper_thread_.WaitForFileFinished(); 316
317 const int64_t stop_time = rtc::TimeMicros();
318
319 task_queue_.PostTask([this, stop_time]() {
320 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
321 if (file_->is_open()) {
322 StopLogFile(stop_time);
323 }
324 });
325
326 // Wait for an indication that the previous task, which stops the log file,
327 // has been executed to completion.
328 rtc::Event file_finished(true, false);
329 task_queue_.PostTask([this, &file_finished]() {
330 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
331 file_finished.Set();
332 });
333 file_finished.Wait(rtc::Event::kForever);
334
335 LOG(LS_INFO) << "WebRTC event log successfully stopped.";
268 } 336 }
269 337
270 void RtcEventLogImpl::LogVideoReceiveStreamConfig( 338 void RtcEventLogImpl::LogVideoReceiveStreamConfig(
271 const rtclog::StreamConfig& config) { 339 const rtclog::StreamConfig& config) {
272 std::unique_ptr<rtclog::Event> event(new rtclog::Event()); 340 std::unique_ptr<rtclog::Event> event(new rtclog::Event());
273 event->set_timestamp_us(rtc::TimeMicros()); 341 event->set_timestamp_us(rtc::TimeMicros());
274 event->set_type(rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT); 342 event->set_type(rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT);
275 343
276 rtclog::VideoReceiveConfig* receiver_config = 344 rtclog::VideoReceiveConfig* receiver_config =
277 event->mutable_video_receiver_config(); 345 event->mutable_video_receiver_config();
(...skipping 280 matching lines...) Expand 10 before | Expand all | Expand 10 after
558 event->set_type(rtclog::Event::BWE_PROBE_RESULT_EVENT); 626 event->set_type(rtclog::Event::BWE_PROBE_RESULT_EVENT);
559 627
560 auto probe_result = event->mutable_probe_result(); 628 auto probe_result = event->mutable_probe_result();
561 probe_result->set_id(id); 629 probe_result->set_id(id);
562 probe_result->set_result(result); 630 probe_result->set_result(result);
563 if (result == rtclog::BweProbeResult::SUCCESS) 631 if (result == rtclog::BweProbeResult::SUCCESS)
564 probe_result->set_bitrate_bps(bitrate_bps); 632 probe_result->set_bitrate_bps(bitrate_bps);
565 StoreEvent(std::move(event)); 633 StoreEvent(std::move(event));
566 } 634 }
567 635
636 void RtcEventLogImpl::StartLoggingInternal(std::unique_ptr<FileWrapper> file,
637 int64_t max_size_bytes) {
638 LOG(LS_INFO) << "Starting WebRTC event log.";
639
640 max_size_bytes = (max_size_bytes <= 0) ? std::numeric_limits<int64_t>::max()
641 : max_size_bytes;
642 auto file_handler = [this, max_size_bytes](
643 std::unique_ptr<FileWrapper> file) {
644 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
645 if (!file_->is_open()) {
646 max_size_bytes_ = max_size_bytes;
647 file_= std::move(file);
648 StartLogFile();
649 } else {
650 // Already started. Ignore message and close file handle.
651 file->CloseFile();
652 }
653 };
654 auto task = rtc::WrapUnique<rtc::QueuedTask>(
655 new RtcEventLogFileTask(std::move(file), file_handler));
656 task_queue_.PostTask(std::move(task));
657 }
658
568 void RtcEventLogImpl::StoreEvent(std::unique_ptr<rtclog::Event> event) { 659 void RtcEventLogImpl::StoreEvent(std::unique_ptr<rtclog::Event> event) {
569 RTC_DCHECK(event.get() != nullptr); 660 RTC_DCHECK(event);
570 if (!event_queue_.Insert(&event)) { 661
571 LOG(LS_ERROR) << "WebRTC event log queue full. Dropping event."; 662 auto event_handler = [this](std::unique_ptr<rtclog::Event> rtclog_event) {
663 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
664 if (file_->is_open()) {
665 LogToFile(std::move(rtclog_event));
666 } else {
667 LogToMemory(std::move(rtclog_event));
668 }
669 };
670
671 auto task = rtc::WrapUnique<rtc::QueuedTask>(
672 new RtcEventLogTask(std::move(event), event_handler));
673
674 task_queue_.PostTask(std::move(task));
nisse-webrtc 2017/08/30 12:18:35 Now you set this up in three steps: 1. named lamb
eladalon 2017/08/30 13:01:19 We have to have RtcEventLogTask (now named Resourc
nisse-webrtc 2017/08/30 13:13:23 ResourceOwningTask is fine, I was merely suggestin
eladalon 2017/08/30 14:15:44 I see two issues: 1. The nit - that the lambda is
nisse-webrtc 2017/08/30 14:27:01 I'm not talking of either of those... I would like
eladalon 2017/08/30 15:19:20 1. We can do that, because unique_ptr<Derived> is
nisse-webrtc 2017/08/31 06:53:15 I'd expect the unique_ptr to support implicit con
eladalon 2017/08/31 09:05:49 You're right, I skimmed through the compiler's err
675 }
676
677 bool RtcEventLogImpl::AppendEventToString(rtclog::Event* event,
678 ProtoString* output_string) {
679 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
680
681 // Even though we're only serializing a single event during this call, what
682 // we intend to get is a list of events, with a tag and length preceding
683 // each actual event. To produce that, we serialize a list of a single event.
684 // If we later serialize additional events, the resulting ProtoString will
685 // be a proper concatenation of all those events.
686
687 rtclog::EventStream event_stream;
688 event_stream.add_stream();
689
690 // As a tweak, we swap the new event into the event-stream, write that to
691 // file, then swap back. This saves on some copying.
692 rtclog::Event* output_event = event_stream.mutable_stream(0);
693 output_event->Swap(event);
694
695 bool appended;
696 size_t potential_new_size =
697 written_bytes_ + output_string->size() + event_stream.ByteSize();
698 if (potential_new_size <= max_size_bytes_) {
699 event_stream.AppendToString(output_string);
700 appended = true;
701 } else {
702 appended = false;
572 } 703 }
573 helper_thread_.SignalNewEvent(); 704
705 // When the function returns, the original Event will be unchanged.
706 output_event->Swap(event);
707
708 return appended;
709 }
710
711 void RtcEventLogImpl::LogToMemory(std::unique_ptr<rtclog::Event> event) {
712 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
713 RTC_DCHECK(!file_->is_open());
714
715 if (IsConfigEvent(*event.get())) {
716 config_history_.push_back(std::move(event));
717 } else {
718 history_.push_back(std::move(event));
719 if (history_.size() > kEventsInHistory) {
720 history_.pop_front();
721 }
722 }
723 }
724
725 void RtcEventLogImpl::StartLogFile() {
726 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
727 RTC_DCHECK(file_->is_open());
728
729 ProtoString output_string;
730
731 // Create and serialize the LOG_START event.
732 // The timestamp used will correspond to when logging has started. The log
733 // may contain events earlier than the LOG_START event. (In general, the
734 // timestamps in the log are not monotonic.)
735 rtclog::Event start_event;
736 start_event.set_timestamp_us(rtc::TimeMicros());
737 start_event.set_type(rtclog::Event::LOG_START);
738 bool appended = AppendEventToString(&start_event, &output_string);
739
740 // Serialize the config information for all old streams, including streams
741 // which were already logged to previous files.
742 for (auto& event : config_history_) {
743 if (!appended) {
744 break;
745 }
746 appended = AppendEventToString(event.get(), &output_string);
747 }
748
749 // Serialize the events in the event queue.
750 while (appended && !history_.empty()) {
751 appended = AppendEventToString(history_.front().get(), &output_string);
752 if (appended) {
753 // Known issue - if writing to the file fails, these events will have
754 // been lost. If we try to open a new file, these events will be missing
755 // from it.
756 history_.pop_front();
757 }
758 }
759
760 // Write to file.
761 if (!file_->Write(output_string.data(), output_string.size())) {
762 LOG(LS_ERROR) << "FileWrapper failed to write WebRtcEventLog file.";
763 // The current FileWrapper implementation closes the file on error.
764 RTC_DCHECK(!file_->is_open());
765 return;
766 }
767 written_bytes_ += output_string.size();
768
769 if (!appended) {
770 RTC_DCHECK(file_->is_open());
771 StopLogFile(rtc::TimeMicros());
772 }
773 }
774
775 void RtcEventLogImpl::LogToFile(std::unique_ptr<rtclog::Event> event) {
776 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
777 RTC_DCHECK(file_->is_open());
778
779 ProtoString output_string;
780
781 bool appended = AppendEventToString(event.get(), &output_string);
782
783 if (IsConfigEvent(*event.get())) {
784 config_history_.push_back(std::move(event));
785 }
786
787 if (!appended) {
terelius 2017/08/30 15:06:04 This is nitpicking, but you are losing the event f
eladalon 2017/08/30 15:19:20 Done.
788 RTC_DCHECK(file_->is_open());
789 StopLogFile(rtc::TimeMicros());
790 return;
791 }
792
793 // Write string to file.
794 if (file_->Write(output_string.data(), output_string.size())) {
795 written_bytes_ += output_string.size();
796 } else {
797 LOG(LS_ERROR) << "FileWrapper failed to write WebRtcEventLog file.";
798 // The current FileWrapper implementation closes the file on error.
799 RTC_DCHECK(!file_->is_open());
800 }
801 }
802
803 void RtcEventLogImpl::StopLogFile(int64_t stop_time) {
804 RTC_DCHECK_CALLED_SEQUENTIALLY(&worker_sequence_checker_);
805 RTC_DCHECK(file_->is_open());
806
807 ProtoString output_string;
808
809 rtclog::Event end_event;
810 end_event.set_timestamp_us(stop_time);
811 end_event.set_type(rtclog::Event::LOG_END);
812 bool appended = AppendEventToString(&end_event, &output_string);
813
814 if (appended) {
815 if (!file_->Write(output_string.data(), output_string.size())) {
816 LOG(LS_ERROR) << "FileWrapper failed to write WebRtcEventLog file.";
817 // The current FileWrapper implementation closes the file on error.
818 RTC_DCHECK(!file_->is_open());
819 }
820 written_bytes_ += output_string.size();
821 }
822
823 max_size_bytes_ = std::numeric_limits<int64_t>::max();
824 written_bytes_ = 0;
825
826 file_->CloseFile();
827 RTC_DCHECK(!file_->is_open());
574 } 828 }
575 829
576 bool RtcEventLog::ParseRtcEventLog(const std::string& file_name, 830 bool RtcEventLog::ParseRtcEventLog(const std::string& file_name,
577 rtclog::EventStream* result) { 831 rtclog::EventStream* result) {
578 char tmp_buffer[1024]; 832 char tmp_buffer[1024];
579 int bytes_read = 0; 833 int bytes_read = 0;
580 std::unique_ptr<FileWrapper> dump_file(FileWrapper::Create()); 834 std::unique_ptr<FileWrapper> dump_file(FileWrapper::Create());
581 if (!dump_file->OpenFile(file_name.c_str(), true)) { 835 if (!dump_file->OpenFile(file_name.c_str(), true)) {
582 return false; 836 return false;
583 } 837 }
584 ProtoString dump_buffer; 838 ProtoString dump_buffer;
585 while ((bytes_read = dump_file->Read(tmp_buffer, sizeof(tmp_buffer))) > 0) { 839 while ((bytes_read = dump_file->Read(tmp_buffer, sizeof(tmp_buffer))) > 0) {
586 dump_buffer.append(tmp_buffer, bytes_read); 840 dump_buffer.append(tmp_buffer, bytes_read);
587 } 841 }
588 dump_file->CloseFile(); 842 dump_file->CloseFile();
589 return result->ParseFromString(dump_buffer); 843 return result->ParseFromString(dump_buffer);
590 } 844 }
591 845
592 #endif // ENABLE_RTC_EVENT_LOG 846 #endif // ENABLE_RTC_EVENT_LOG
593 847
594 // RtcEventLog member functions. 848 // RtcEventLog member functions.
595 std::unique_ptr<RtcEventLog> RtcEventLog::Create() { 849 std::unique_ptr<RtcEventLog> RtcEventLog::Create() {
596 #ifdef ENABLE_RTC_EVENT_LOG 850 #ifdef ENABLE_RTC_EVENT_LOG
851 // TODO(eladalon): Known issue - there's a race over |log_count_| here.
597 constexpr int kMaxLogCount = 5; 852 constexpr int kMaxLogCount = 5;
598 int count = rtc::AtomicOps::Increment(&RtcEventLogImpl::log_count_); 853 int count = 1 + std::atomic_fetch_add(&RtcEventLogImpl::log_count_, 1);
599 if (count > kMaxLogCount) { 854 if (count > kMaxLogCount) {
600 LOG(LS_WARNING) << "Denied creation of additional WebRTC event logs. " 855 LOG(LS_WARNING) << "Denied creation of additional WebRTC event logs. "
601 << count - 1 << " logs open already."; 856 << count - 1 << " logs open already.";
602 rtc::AtomicOps::Decrement(&RtcEventLogImpl::log_count_); 857 std::atomic_fetch_sub(&RtcEventLogImpl::log_count_, 1);
603 return std::unique_ptr<RtcEventLog>(new RtcEventLogNullImpl()); 858 return std::unique_ptr<RtcEventLog>(new RtcEventLogNullImpl());
604 } 859 }
605 return std::unique_ptr<RtcEventLog>(new RtcEventLogImpl()); 860 return std::unique_ptr<RtcEventLog>(new RtcEventLogImpl());
606 #else 861 #else
607 return std::unique_ptr<RtcEventLog>(new RtcEventLogNullImpl()); 862 return CreateNull();
608 #endif // ENABLE_RTC_EVENT_LOG 863 #endif // ENABLE_RTC_EVENT_LOG
609 } 864 }
610 865
611 std::unique_ptr<RtcEventLog> RtcEventLog::CreateNull() { 866 std::unique_ptr<RtcEventLog> RtcEventLog::CreateNull() {
612 return std::unique_ptr<RtcEventLog>(new RtcEventLogNullImpl()); 867 return std::unique_ptr<RtcEventLog>(new RtcEventLogNullImpl());
613 } 868 }
614 869
615 } // namespace webrtc 870 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/logging/rtc_event_log/rtc_event_log.h ('k') | webrtc/logging/rtc_event_log/rtc_event_log_helper_thread.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698