Index: webrtc/call/rtc_event_log.cc |
diff --git a/webrtc/call/rtc_event_log.cc b/webrtc/call/rtc_event_log.cc |
index 361db81c068ccc99b480918ad55c05cbd1be1656..8c218d5c77a285efd283ffce4c07d502037611fa 100644 |
--- a/webrtc/call/rtc_event_log.cc |
+++ b/webrtc/call/rtc_event_log.cc |
@@ -10,18 +10,24 @@ |
#include "webrtc/call/rtc_event_log.h" |
-#include <deque> |
#include <vector> |
+#include <limits> |
#include "webrtc/base/checks.h" |
+#include "webrtc/base/constructormagic.h" |
#include "webrtc/base/criticalsection.h" |
+#include "webrtc/base/event.h" |
+#include "webrtc/base/platform_thread.h" |
#include "webrtc/base/thread_annotations.h" |
#include "webrtc/call.h" |
+#include "webrtc/call/ringbuffer.h" |
+#include "webrtc/common_audio/swap_queue.h" |
#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
#include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
#include "webrtc/modules/rtp_rtcp/source/rtcp_utility.h" |
#include "webrtc/system_wrappers/include/clock.h" |
#include "webrtc/system_wrappers/include/file_wrapper.h" |
+#include "webrtc/system_wrappers/include/logging.h" |
#ifdef ENABLE_RTC_EVENT_LOG |
// Files generated at build-time by the protobuf compiler. |
@@ -37,12 +43,17 @@ namespace webrtc { |
#ifndef ENABLE_RTC_EVENT_LOG |
// No-op implementation if flag is not set. |
-class RtcEventLogImpl final : public RtcEventLog { |
+class RtcEventLogNullImpl final : public RtcEventLog { |
public: |
- void SetBufferDuration(int64_t buffer_duration_us) override {} |
- void StartLogging(const std::string& file_name, int duration_ms) override {} |
- bool StartLogging(rtc::PlatformFile log_file) override { return false; } |
- void StopLogging(void) override {} |
+ bool StartLogging(const std::string& file_name, |
+ int64_t max_size_bytes) override { |
+ return false; |
+ } |
+ bool StartLogging(rtc::PlatformFile platform_file, |
+ int64_t max_size_bytes) override { |
+ return false; |
+ } |
+ void StopLogging() override {} |
void LogVideoReceiveStreamConfig( |
const VideoReceiveStream::Config& config) override {} |
void LogVideoSendStreamConfig( |
@@ -65,11 +76,13 @@ class RtcEventLogImpl final : public RtcEventLog { |
class RtcEventLogImpl final : public RtcEventLog { |
public: |
- RtcEventLogImpl(); |
+ explicit RtcEventLogImpl(const Clock* clock); |
+ ~RtcEventLogImpl() override; |
- void SetBufferDuration(int64_t buffer_duration_us) override; |
- void StartLogging(const std::string& file_name, int duration_ms) override; |
- bool StartLogging(rtc::PlatformFile log_file) override; |
+ bool StartLogging(const std::string& file_name, |
+ int64_t max_size_bytes) override; |
+ bool StartLogging(rtc::PlatformFile platform_file, |
+ int64_t max_size_bytes) override; |
void StopLogging() override; |
void LogVideoReceiveStreamConfig( |
const VideoReceiveStream::Config& config) override; |
@@ -88,37 +101,79 @@ class RtcEventLogImpl final : public RtcEventLog { |
int32_t total_packets) override; |
private: |
- // Starts logging. This function assumes the file_ has been opened succesfully |
- // and that the start_time_us_ and _duration_us_ have been set. |
- void StartLoggingLocked() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
- // Stops logging and clears the stored data and buffers. |
- void StopLoggingLocked() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
- // Adds a new event to the logfile if logging is active, or adds it to the |
- // list of recent log events otherwise. |
- void HandleEvent(rtclog::Event* event) EXCLUSIVE_LOCKS_REQUIRED(crit_); |
- // Writes the event to the file. Note that this will destroy the state of the |
- // input argument. |
- void StoreToFile(rtclog::Event* event) EXCLUSIVE_LOCKS_REQUIRED(crit_); |
- // Adds the event to the list of recent events, and removes any events that |
- // are too old and no longer fall in the time window. |
- void AddRecentEvent(const rtclog::Event& event) |
+ // The normal state progression is |
+ // TO_MEMORY -> STARTING_FILE -> TO_FILE -> STOPPING_FILE -> TO_MEMORY |
+ // but it can also go directly from STARTING_FILE to STOPPING_FILE. |
+ enum LogState { |
+ TO_MEMORY, |
+ STARTING_FILE, |
+ TO_FILE, |
+ STOPPING_FILE, |
+ TERMINATE_THREAD, |
+ }; |
+ |
+ static bool ThreadOutputFunction(void* obj); |
+ void TerminateThread(); |
+ |
+ bool AppendEventToString(rtclog::Event* event) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ void AppendEventToHistory(const rtclog::Event& event) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ bool ProcessInOrder(bool memory, int64_t current_time) |
EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ void LogToMemory() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
the sun
2016/02/10 10:47:23
To make things even clearer, the parts of this cla
terelius
2016/02/11 21:27:28
One side effect of moving the state to a separate
the sun
2016/02/12 09:37:32
For StartLogging() it would likely make sense to o
terelius
2016/02/12 14:13:38
But even if StartLogging opens the file we wouldn'
the sun
2016/02/12 15:37:27
You probably want to process the control messages
terelius
2016/02/12 16:22:24
Though if we process the control messages before t
the sun
2016/02/16 13:29:15
I thought a START_FILE message would contain the f
terelius
2016/02/16 14:36:58
Oh, so you are saying that you would always proces
the sun
2016/02/17 09:44:20
No, I would handle all the control messages first,
|
+ void StartLogFile() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ void LogToFile() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ void StopLogFile() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ void WriteLog(); |
+ |
+ // Message queues for passing events to the logging thread. |
+ SwapQueue<rtclog::Event> config_buffer_; |
the sun
2016/02/10 10:47:23
I think it is better to call these config_queue_ e
terelius
2016/02/11 21:27:28
Done.
|
+ SwapQueue<rtclog::Event> rtp_buffer_; |
+ SwapQueue<rtclog::Event> rtcp_buffer_; |
+ SwapQueue<rtclog::Event> acm_playout_buffer_; |
+ SwapQueue<rtclog::Event> bwe_loss_buffer_; |
rtc::CriticalSection crit_; |
- rtc::scoped_ptr<FileWrapper> file_ GUARDED_BY(crit_) = |
- rtc::scoped_ptr<FileWrapper>(FileWrapper::Create()); |
- rtc::PlatformFile platform_file_ GUARDED_BY(crit_) = |
- rtc::kInvalidPlatformFileValue; |
- rtclog::EventStream stream_ GUARDED_BY(crit_); |
- std::deque<rtclog::Event> recent_log_events_ GUARDED_BY(crit_); |
- std::vector<rtclog::Event> config_events_ GUARDED_BY(crit_); |
- |
- // Microseconds to record log events, before starting the actual log. |
- int64_t buffer_duration_us_ GUARDED_BY(crit_); |
- bool currently_logging_ GUARDED_BY(crit_); |
- int64_t start_time_us_ GUARDED_BY(crit_); |
- int64_t duration_us_ GUARDED_BY(crit_); |
+ |
+ // History containing the most recent events (~ 10 s). |
+ RingBuffer<rtclog::Event> history_ GUARDED_BY(crit_); |
+ |
+ // History containing all past configuration events. |
+ std::vector<rtclog::Event> config_history_ GUARDED_BY(crit_); |
+ |
+ LogState state_ GUARDED_BY(crit_); |
+ rtc::scoped_ptr<FileWrapper> file_ GUARDED_BY(crit_); |
+ rtc::PlatformThread thread_ GUARDED_BY(crit_); |
+ |
+ int64_t max_size_bytes_ GUARDED_BY(crit_); |
+ int64_t written_bytes_ GUARDED_BY(crit_); |
+ int64_t start_time_ GUARDED_BY(crit_); |
+ int64_t stop_time_ GUARDED_BY(crit_); |
+ |
+ // We want to extract the elements from the buffers in timestamp order |
+ // and, because we can't peek into the SwapQueue, we need to keep the first |
+ // elements outside. |
+ rtclog::Event config_event_; |
+ rtclog::Event rtp_event_; |
+ rtclog::Event rtcp_event_; |
+ rtclog::Event playout_event_; |
+ rtclog::Event loss_event_; |
+ bool valid_config_event_; |
the sun
2016/02/10 10:47:23
You may be able to avoid these by using rtc::Optio
terelius
2016/02/11 21:27:28
There are several reasons why Optional won't work.
|
+ bool valid_rtp_event_; |
+ bool valid_rtcp_event_; |
+ bool valid_playout_event_; |
+ bool valid_loss_event_; |
+ |
+ // Temporary space for serializing profobuf data. |
+ std::string output_string_ GUARDED_BY(crit_); |
+ |
+ rtc::Event wake_up_; |
+ rtc::Event file_finished_; |
+ |
const Clock* const clock_; |
+ |
+ RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(RtcEventLogImpl); |
}; |
namespace { |
@@ -126,10 +181,6 @@ namespace { |
// that the rest of the WebRtc project can use, to the corresponding |
// serialized enum which is defined by the protobuf. |
-// Do not add default return values to the conversion functions in this |
-// unnamed namespace. The intention is to make the compiler warn if anyone |
-// adds unhandled new events/modes/etc. |
- |
rtclog::VideoReceiveConfig_RtcpMode ConvertRtcpMode(RtcpMode rtcp_mode) { |
switch (rtcp_mode) { |
case RtcpMode::kCompound: |
@@ -159,108 +210,115 @@ rtclog::MediaType ConvertMediaType(MediaType media_type) { |
return rtclog::ANY; |
} |
-} // namespace |
+// The RTP and RTCP buffers reserve space for twice the expected number of |
+// sent packets because they also contain received packets. |
+const int kStreamConfigsPerSecond = 64; // 16 clients w. 4 streams each. |
+const int kRtpPacketsPerSecond = 500; // 125 sent video packets/s @ 1 Mbps. |
+const int kRtcpPacketsPerSecond = 40; // Assume RTCP sent 20 times/s. |
+const int kPlayoutsPerSecond = 100; // Playout called every 10 ms. |
+const int kBweUpdatesPerSecond = 20; // One BWE update per RTCP packet. |
+const int kEventsInHistory = 10000; |
-namespace { |
-bool IsConfigEvent(const rtclog::Event& event) { |
- rtclog::Event_EventType event_type = event.type(); |
- return event_type == rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT || |
- event_type == rtclog::Event::VIDEO_SENDER_CONFIG_EVENT || |
- event_type == rtclog::Event::AUDIO_RECEIVER_CONFIG_EVENT || |
- event_type == rtclog::Event::AUDIO_SENDER_CONFIG_EVENT; |
-} |
} // namespace |
+ |
// RtcEventLogImpl member functions. |
-RtcEventLogImpl::RtcEventLogImpl() |
- : file_(FileWrapper::Create()), |
- stream_(), |
- buffer_duration_us_(10000000), |
- currently_logging_(false), |
- start_time_us_(0), |
- duration_us_(0), |
- clock_(Clock::GetRealTimeClock()) { |
+RtcEventLogImpl::RtcEventLogImpl(const Clock* clock) |
+ // Allocate buffers for roughly one second of history. |
+ : config_buffer_(kStreamConfigsPerSecond), |
+ rtp_buffer_(kRtpPacketsPerSecond), |
+ rtcp_buffer_(kRtcpPacketsPerSecond), |
+ acm_playout_buffer_(kPlayoutsPerSecond), |
+ bwe_loss_buffer_(kBweUpdatesPerSecond), |
+ crit_(), |
+ history_(kEventsInHistory), |
+ config_history_(), |
+ state_(TO_MEMORY), |
+ file_(FileWrapper::Create()), |
+ thread_(&ThreadOutputFunction, this, "RtcEventLog thread"), |
+ max_size_bytes_(std::numeric_limits<int64_t>::max()), |
+ written_bytes_(0), |
+ start_time_(0), |
+ stop_time_(std::numeric_limits<int64_t>::max()), |
+ config_event_(), |
+ rtp_event_(), |
+ rtcp_event_(), |
+ playout_event_(), |
+ loss_event_(), |
+ valid_config_event_(false), |
+ valid_rtp_event_(false), |
+ valid_rtcp_event_(false), |
+ valid_playout_event_(false), |
+ valid_loss_event_(false), |
+ output_string_(), |
+ wake_up_(false, false), |
+ file_finished_(false, false), |
+ clock_(clock) { |
+ thread_.Start(); |
} |
-void RtcEventLogImpl::SetBufferDuration(int64_t buffer_duration_us) { |
- rtc::CritScope lock(&crit_); |
- buffer_duration_us_ = buffer_duration_us; |
+RtcEventLogImpl::~RtcEventLogImpl() { |
+ TerminateThread(); |
+ thread_.Stop(); |
} |
-void RtcEventLogImpl::StartLogging(const std::string& file_name, |
- int duration_ms) { |
+bool RtcEventLogImpl::StartLogging(const std::string& file_name, |
+ int64_t max_size_bytes) { |
rtc::CritScope lock(&crit_); |
- if (currently_logging_) { |
- StopLoggingLocked(); |
+ if (state_ != TO_MEMORY) { |
+ return false; |
} |
+ file_.reset(FileWrapper::Create()); |
if (file_->OpenFile(file_name.c_str(), false) != 0) { |
- return; |
+ return false; |
} |
- start_time_us_ = clock_->TimeInMicroseconds(); |
- duration_us_ = static_cast<int64_t>(duration_ms) * 1000; |
- StartLoggingLocked(); |
+ |
+ max_size_bytes_ = max_size_bytes; |
+ start_time_ = clock_->TimeInMicroseconds(); |
+ stop_time_ = std::numeric_limits<int64_t>::max(); |
+ state_ = STARTING_FILE; |
+ return true; |
} |
-bool RtcEventLogImpl::StartLogging(rtc::PlatformFile log_file) { |
+bool RtcEventLogImpl::StartLogging(rtc::PlatformFile platform_file, |
+ int64_t max_size_bytes) { |
rtc::CritScope lock(&crit_); |
- |
- if (currently_logging_) { |
- StopLoggingLocked(); |
+ if (state_ != TO_MEMORY) { |
+ return false; |
} |
- RTC_DCHECK(platform_file_ == rtc::kInvalidPlatformFileValue); |
- FILE* file_stream = rtc::FdopenPlatformFileForWriting(log_file); |
- if (!file_stream) { |
- rtc::ClosePlatformFile(log_file); |
+ FILE* file_handle = rtc::FdopenPlatformFileForWriting(platform_file); |
+ if (!file_handle) { |
return false; |
} |
- |
- if (file_->OpenFromFileHandle(file_stream, true, false) != 0) { |
- rtc::ClosePlatformFile(log_file); |
+ file_.reset(FileWrapper::Create()); |
+ if (file_->OpenFromFileHandle(file_handle, true, false) != 0) { |
return false; |
} |
- platform_file_ = log_file; |
- // Set the start time and duration to keep logging for 10 minutes. |
- start_time_us_ = clock_->TimeInMicroseconds(); |
- duration_us_ = 10 * 60 * 1000000; |
- StartLoggingLocked(); |
+ |
+ max_size_bytes_ = max_size_bytes; |
+ start_time_ = clock_->TimeInMicroseconds(); |
+ stop_time_ = std::numeric_limits<int64_t>::max(); |
+ state_ = STARTING_FILE; |
return true; |
} |
-void RtcEventLogImpl::StartLoggingLocked() { |
- currently_logging_ = true; |
- |
- // Write all old configuration events to the log file. |
- for (auto& event : config_events_) { |
- StoreToFile(&event); |
- } |
- // Write all recent configuration events to the log file, and |
- // write all other recent events to the log file, ignoring any old events. |
- for (auto& event : recent_log_events_) { |
- if (IsConfigEvent(event)) { |
- StoreToFile(&event); |
- config_events_.push_back(event); |
- } else if (event.timestamp_us() >= start_time_us_ - buffer_duration_us_) { |
- StoreToFile(&event); |
+void RtcEventLogImpl::StopLogging() { |
+ { |
+ rtc::CritScope lock(&crit_); |
+ if (state_ == TO_FILE || state_ == STARTING_FILE) { |
+ stop_time_ = clock_->TimeInMicroseconds(); |
+ wake_up_.Set(); // Request the output thread to wake up. |
+ } else { |
+ return; |
} |
} |
- recent_log_events_.clear(); |
- // Write a LOG_START event to the file. |
- rtclog::Event start_event; |
- start_event.set_timestamp_us(start_time_us_); |
- start_event.set_type(rtclog::Event::LOG_START); |
- StoreToFile(&start_event); |
-} |
- |
-void RtcEventLogImpl::StopLogging() { |
- rtc::CritScope lock(&crit_); |
- StopLoggingLocked(); |
+ file_finished_.Wait( |
+ rtc::Event::kForever); // Wait for the log file to be completed. |
} |
void RtcEventLogImpl::LogVideoReceiveStreamConfig( |
const VideoReceiveStream::Config& config) { |
- rtc::CritScope lock(&crit_); |
- |
rtclog::Event event; |
event.set_timestamp_us(clock_->TimeInMicroseconds()); |
event.set_type(rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT); |
@@ -292,13 +350,13 @@ void RtcEventLogImpl::LogVideoReceiveStreamConfig( |
decoder->set_name(d.payload_name); |
decoder->set_payload_type(d.payload_type); |
} |
- HandleEvent(&event); |
+ if (!config_buffer_.Insert(&event)) { |
+ LOG(LS_WARNING) << "Config buffer full. Not logging config event."; |
+ } |
} |
void RtcEventLogImpl::LogVideoSendStreamConfig( |
const VideoSendStream::Config& config) { |
- rtc::CritScope lock(&crit_); |
- |
rtclog::Event event; |
event.set_timestamp_us(clock_->TimeInMicroseconds()); |
event.set_type(rtclog::Event::VIDEO_SENDER_CONFIG_EVENT); |
@@ -324,7 +382,9 @@ void RtcEventLogImpl::LogVideoSendStreamConfig( |
rtclog::EncoderConfig* encoder = sender_config->mutable_encoder(); |
encoder->set_name(config.encoder_settings.payload_name); |
encoder->set_payload_type(config.encoder_settings.payload_type); |
- HandleEvent(&event); |
+ if (!config_buffer_.Insert(&event)) { |
+ LOG(LS_WARNING) << "Config buffer full. Not logging config event."; |
+ } |
} |
void RtcEventLogImpl::LogRtpHeader(PacketDirection direction, |
@@ -347,7 +407,6 @@ void RtcEventLogImpl::LogRtpHeader(PacketDirection direction, |
header_length += (x_len + 1) * 4; |
} |
- rtc::CritScope lock(&crit_); |
rtclog::Event rtp_event; |
rtp_event.set_timestamp_us(clock_->TimeInMicroseconds()); |
rtp_event.set_type(rtclog::Event::RTP_EVENT); |
@@ -355,14 +414,15 @@ void RtcEventLogImpl::LogRtpHeader(PacketDirection direction, |
rtp_event.mutable_rtp_packet()->set_type(ConvertMediaType(media_type)); |
rtp_event.mutable_rtp_packet()->set_packet_length(packet_length); |
rtp_event.mutable_rtp_packet()->set_header(header, header_length); |
- HandleEvent(&rtp_event); |
+ if (!rtp_buffer_.Insert(&rtp_event)) { |
+ LOG(LS_WARNING) << "RTP buffer full. Not logging RTP packet."; |
+ } |
} |
void RtcEventLogImpl::LogRtcpPacket(PacketDirection direction, |
MediaType media_type, |
const uint8_t* packet, |
size_t length) { |
- rtc::CritScope lock(&crit_); |
rtclog::Event rtcp_event; |
rtcp_event.set_timestamp_us(clock_->TimeInMicroseconds()); |
rtcp_event.set_type(rtclog::Event::RTCP_EVENT); |
@@ -414,23 +474,25 @@ void RtcEventLogImpl::LogRtcpPacket(PacketDirection direction, |
block_begin += block_size; |
} |
rtcp_event.mutable_rtcp_packet()->set_packet_data(buffer, buffer_length); |
- HandleEvent(&rtcp_event); |
+ if (!rtcp_buffer_.Insert(&rtcp_event)) { |
+ LOG(LS_WARNING) << "RTCP buffer full. Not logging RTCP packet."; |
+ } |
} |
void RtcEventLogImpl::LogAudioPlayout(uint32_t ssrc) { |
- rtc::CritScope lock(&crit_); |
rtclog::Event event; |
event.set_timestamp_us(clock_->TimeInMicroseconds()); |
event.set_type(rtclog::Event::AUDIO_PLAYOUT_EVENT); |
auto playout_event = event.mutable_audio_playout_event(); |
playout_event->set_local_ssrc(ssrc); |
- HandleEvent(&event); |
+ if (!acm_playout_buffer_.Insert(&event)) { |
+ LOG(LS_WARNING) << "Playout buffer full. Not logging ACM playout."; |
+ } |
} |
void RtcEventLogImpl::LogBwePacketLossEvent(int32_t bitrate, |
uint8_t fraction_loss, |
int32_t total_packets) { |
- rtc::CritScope lock(&crit_); |
rtclog::Event event; |
event.set_timestamp_us(clock_->TimeInMicroseconds()); |
event.set_type(rtclog::Event::BWE_PACKET_LOSS_EVENT); |
@@ -438,62 +500,285 @@ void RtcEventLogImpl::LogBwePacketLossEvent(int32_t bitrate, |
bwe_event->set_bitrate(bitrate); |
bwe_event->set_fraction_loss(fraction_loss); |
bwe_event->set_total_packets(total_packets); |
- HandleEvent(&event); |
+ if (!bwe_loss_buffer_.Insert(&event)) { |
+ LOG(LS_WARNING) << "BWE loss buffer full. Not logging BWE update."; |
+ } |
+} |
+ |
+bool RtcEventLogImpl::AppendEventToString(rtclog::Event* event) { |
+ rtclog::EventStream event_stream; |
+ event_stream.add_stream(); |
+ event_stream.mutable_stream(0)->Swap(event); |
+ // TODO(terelius): We create a new event strem per event, but it will look |
+ // like a single stream when we read it back from file. |
+ // Is this guaranteed to work e.g. in future versions of protobuf? |
+ bool stop = true; |
+ if (written_bytes_ + static_cast<int64_t>(output_string_.size()) + |
+ event_stream.ByteSize() <= |
+ max_size_bytes_) { |
+ event_stream.AppendToString(&output_string_); |
+ stop = false; |
+ } |
+ // Swap the event back so that we don't mix event types in the queues. |
+ event_stream.mutable_stream(0)->Swap(event); |
+ return stop; |
+} |
+ |
+void RtcEventLogImpl::AppendEventToHistory(const rtclog::Event& event) { |
+ history_.push_back(event); |
} |
-void RtcEventLogImpl::StopLoggingLocked() { |
- if (currently_logging_) { |
- currently_logging_ = false; |
- // Create a LogEnd event |
- rtclog::Event event; |
- event.set_timestamp_us(clock_->TimeInMicroseconds()); |
- event.set_type(rtclog::Event::LOG_END); |
- // Store the event and close the file |
- RTC_DCHECK(file_->Open()); |
- StoreToFile(&event); |
- file_->CloseFile(); |
- if (platform_file_ != rtc::kInvalidPlatformFileValue) { |
- rtc::ClosePlatformFile(platform_file_); |
- platform_file_ = rtc::kInvalidPlatformFileValue; |
+// Traverses the SwapQueues in timestamp order and copies all events earlier |
+// than |current_time| either to the history or to a string that will be |
+// written to disc. |
+bool RtcEventLogImpl::ProcessInOrder(bool memory, int64_t current_time) { |
+ bool stop = false; |
+ enum EventType { |
+ CONFIG_EVENT, |
+ RTP_EVENT, |
+ RTCP_EVENT, |
+ PLAYOUT_EVENT, |
+ LOSS_EVENT |
+ }; |
+ |
+ // Extract the head of each queue. |
+ if (!valid_config_event_) { |
+ valid_config_event_ = config_buffer_.Remove(&config_event_); |
+ } |
+ |
+ if (!valid_rtp_event_) { |
+ valid_rtp_event_ = rtp_buffer_.Remove(&rtp_event_); |
+ } |
+ |
+ if (!valid_rtcp_event_) { |
+ valid_rtcp_event_ = rtcp_buffer_.Remove(&rtcp_event_); |
+ } |
+ |
+ if (!valid_playout_event_) { |
+ valid_playout_event_ = acm_playout_buffer_.Remove(&playout_event_); |
+ } |
+ |
+ if (!valid_loss_event_) { |
+ valid_loss_event_ = bwe_loss_buffer_.Remove(&loss_event_); |
+ } |
+ |
+ while ((valid_config_event_ || valid_rtp_event_ || valid_rtcp_event_ || |
+ valid_playout_event_ || valid_loss_event_) && |
+ !stop) { |
+ // Find the earliest event (in timestamp order). |
+ EventType type = CONFIG_EVENT; |
+ int64_t first_timestamp = |
+ (valid_config_event_ ? config_event_.timestamp_us() |
+ : std::numeric_limits<int64_t>::max()); |
+ if (valid_rtp_event_ && (rtp_event_.timestamp_us() < first_timestamp)) { |
+ first_timestamp = rtp_event_.timestamp_us(); |
+ type = RTP_EVENT; |
+ } |
+ if (valid_rtcp_event_ && (rtcp_event_.timestamp_us() < first_timestamp)) { |
+ first_timestamp = rtcp_event_.timestamp_us(); |
+ type = RTCP_EVENT; |
+ } |
+ if (valid_playout_event_ && |
+ playout_event_.timestamp_us() < first_timestamp) { |
+ first_timestamp = playout_event_.timestamp_us(); |
+ type = PLAYOUT_EVENT; |
+ } |
+ if (valid_loss_event_ && loss_event_.timestamp_us() < first_timestamp) { |
+ first_timestamp = loss_event_.timestamp_us(); |
+ type = LOSS_EVENT; |
+ } |
+ |
+ if (first_timestamp > current_time) { |
+ // We have handled all events earlier than current_time. |
+ break; |
+ } |
+ // Serialize the event and fetch the next event of that type. |
+ switch (type) { |
+ case CONFIG_EVENT: |
+ if (!memory) { |
+ stop = AppendEventToString(&config_event_); |
+ } |
+ config_history_.push_back(config_event_); |
+ if (!stop) { |
+ valid_config_event_ = config_buffer_.Remove(&config_event_); |
+ } |
+ break; |
+ case RTP_EVENT: |
+ if (memory) { |
+ AppendEventToHistory(rtp_event_); |
+ } else { |
+ stop = AppendEventToString(&rtp_event_); |
+ } |
+ if (!stop) { |
+ valid_rtp_event_ = rtp_buffer_.Remove(&rtp_event_); |
+ } |
+ break; |
+ case RTCP_EVENT: |
+ if (memory) { |
+ AppendEventToHistory(rtcp_event_); |
+ } else { |
+ stop = AppendEventToString(&rtcp_event_); |
+ } |
+ if (!stop) { |
+ valid_rtcp_event_ = rtcp_buffer_.Remove(&rtcp_event_); |
+ } |
+ break; |
+ case PLAYOUT_EVENT: |
+ if (memory) { |
+ AppendEventToHistory(playout_event_); |
+ } else { |
+ stop = AppendEventToString(&playout_event_); |
+ } |
+ if (!stop) { |
+ valid_playout_event_ = acm_playout_buffer_.Remove(&playout_event_); |
+ } |
+ break; |
+ case LOSS_EVENT: |
+ if (memory) { |
+ AppendEventToHistory(loss_event_); |
+ } else { |
+ stop = AppendEventToString(&loss_event_); |
+ } |
+ if (!stop) { |
+ valid_loss_event_ = bwe_loss_buffer_.Remove(&loss_event_); |
+ } |
+ break; |
} |
} |
- RTC_DCHECK(!file_->Open()); |
- stream_.Clear(); |
+ // We want to stop logging either if we have struck the file size limit |
+ // or if we have logged all events older than |stop_time_| |
+ return stop || (current_time > stop_time_); |
} |
-void RtcEventLogImpl::HandleEvent(rtclog::Event* event) { |
- if (currently_logging_) { |
- if (clock_->TimeInMicroseconds() < start_time_us_ + duration_us_) { |
- StoreToFile(event); |
- return; |
+void RtcEventLogImpl::LogToMemory() { |
+ // Process each event in order and append it to the appropriate history_. |
+ int64_t current_time = clock_->TimeInMicroseconds(); |
+ ProcessInOrder(true, current_time); |
+} |
+ |
+void RtcEventLogImpl::StartLogFile() { |
+ bool stop = false; |
+ output_string_.clear(); |
+ |
+ // Create and serialize the LOG_START event. |
+ rtclog::Event start_event; |
+ start_event.set_timestamp_us(start_time_); |
+ start_event.set_type(rtclog::Event::LOG_START); |
+ AppendEventToString(&start_event); |
+ |
+ // Serialize the config information for all old streams. |
+ for (rtclog::Event& event : config_history_) { |
+ AppendEventToString(&event); |
+ } |
+ |
+ // Serialize the events in the event queue. |
+ for (int i = 0; !history_.empty() && !stop; i++) { |
+ stop = AppendEventToString(&history_.front()); |
+ if (!stop) { |
+ history_.pop_front(); |
} |
- StopLoggingLocked(); |
} |
- AddRecentEvent(*event); |
+ |
+ // Write to file. |
+ file_->Write(output_string_.data(), output_string_.size()); |
+ written_bytes_ += output_string_.size(); |
+ state_ = (stop ? STOPPING_FILE : TO_FILE); |
} |
-void RtcEventLogImpl::StoreToFile(rtclog::Event* event) { |
- // Reuse the same object at every log event. |
- if (stream_.stream_size() < 1) { |
- stream_.add_stream(); |
+void RtcEventLogImpl::LogToFile() { |
+ output_string_.clear(); |
+ |
+ // Process each event in order and append it to the output_string_. |
+ int64_t current_time = clock_->TimeInMicroseconds(); |
+ bool stop = ProcessInOrder(false, current_time); |
+ if (stop || stop_time_ <= current_time) { |
+ state_ = STOPPING_FILE; |
} |
- RTC_DCHECK_EQ(stream_.stream_size(), 1); |
- stream_.mutable_stream(0)->Swap(event); |
- // TODO(terelius): Doesn't this create a new EventStream per event? |
- // Is this guaranteed to work e.g. in future versions of protobuf? |
- std::string dump_buffer; |
- stream_.SerializeToString(&dump_buffer); |
- file_->Write(dump_buffer.data(), dump_buffer.size()); |
+ |
+ // Write string to file. |
+ file_->Write(output_string_.data(), output_string_.size()); |
+ written_bytes_ += output_string_.size(); |
} |
-void RtcEventLogImpl::AddRecentEvent(const rtclog::Event& event) { |
- recent_log_events_.push_back(event); |
- while (recent_log_events_.front().timestamp_us() < |
- event.timestamp_us() - buffer_duration_us_) { |
- if (IsConfigEvent(recent_log_events_.front())) { |
- config_events_.push_back(recent_log_events_.front()); |
+void RtcEventLogImpl::StopLogFile() { |
+ output_string_.clear(); |
+ |
+ rtclog::Event end_event; |
+ end_event.set_timestamp_us(stop_time_); |
+ end_event.set_type(rtclog::Event::LOG_END); |
+ AppendEventToString(&end_event); |
+ |
+ if (written_bytes_ + static_cast<int64_t>(output_string_.size()) <= |
+ max_size_bytes_) { |
+ file_->Write(output_string_.data(), output_string_.size()); |
+ written_bytes_ += output_string_.size(); |
+ } |
+ |
+ state_ = TO_MEMORY; |
+ max_size_bytes_ = std::numeric_limits<int64_t>::max(); |
+ written_bytes_ = 0; |
+ start_time_ = 0; |
+ stop_time_ = std::numeric_limits<int64_t>::max(); |
+ output_string_.clear(); |
+ file_->CloseFile(); |
+ file_finished_.Set(); |
+} |
+ |
+void RtcEventLogImpl::WriteLog() { |
+ bool sleep; |
+ |
+ while (true) { |
+ { |
+ rtc::CritScope lock(&crit_); |
+ switch (state_) { |
the sun
2016/02/10 10:47:23
Pull this switch out into a separate function of t
terelius
2016/02/11 21:27:28
I don't understand what logic you want to test. Wr
the sun
2016/02/12 09:37:32
I was thinking mostly about the stuff in the funct
terelius
2016/02/12 14:13:38
I still don't understand what you want to pull out
the sun
2016/02/12 15:37:27
Yes, I think you could pretty pretty much write th
|
+ case TO_MEMORY: |
+ LogToMemory(); |
+ break; |
+ case STARTING_FILE: |
+ StartLogFile(); |
+ break; |
+ case TO_FILE: |
+ LogToFile(); |
+ break; |
+ case STOPPING_FILE: |
+ StopLogFile(); |
+ break; |
+ case TERMINATE_THREAD: |
+ if (file_->Open()) { |
+ RTC_DCHECK(stop_time_ < std::numeric_limits<int64_t>::max()); |
+ StopLogFile(); |
+ RTC_DCHECK(stop_time_ == std::numeric_limits<int64_t>::max()); |
+ RTC_DCHECK(!file_->Open()); |
+ state_ = TERMINATE_THREAD; |
+ } |
+ return; |
+ } |
+ int64_t current_time = clock_->TimeInMicroseconds(); |
+ sleep = (state_ != STOPPING_FILE && state_ != TERMINATE_THREAD && |
+ current_time <= stop_time_); |
+ } |
+ |
+ // Accumulate a new batch of events instead of processing them one at a |
+ // time. By waiting, we also release the lock so that the main thread can |
+ // call StopLogging and StartLogging. |
+ if (sleep) { |
+ wake_up_.Wait(50); |
} |
- recent_log_events_.pop_front(); |
+ } |
+} |
+ |
+bool RtcEventLogImpl::ThreadOutputFunction(void* obj) { |
+ RtcEventLogImpl* event_log = static_cast<RtcEventLogImpl*>(obj); |
+ event_log->WriteLog(); |
+ return false; |
+} |
+ |
+void RtcEventLogImpl::TerminateThread() { |
+ { |
+ rtc::CritScope lock(&crit_); |
+ stop_time_ = clock_->TimeInMicroseconds(); |
+ state_ = TERMINATE_THREAD; |
+ wake_up_.Set(); // Wake up the output thread. |
} |
} |
@@ -516,8 +801,12 @@ bool RtcEventLog::ParseRtcEventLog(const std::string& file_name, |
#endif // ENABLE_RTC_EVENT_LOG |
// RtcEventLog member functions. |
-rtc::scoped_ptr<RtcEventLog> RtcEventLog::Create() { |
- return rtc::scoped_ptr<RtcEventLog>(new RtcEventLogImpl()); |
+rtc::scoped_ptr<RtcEventLog> RtcEventLog::Create(const Clock* clock) { |
+#ifdef ENABLE_RTC_EVENT_LOG |
+ return rtc::scoped_ptr<RtcEventLog>(new RtcEventLogImpl(clock)); |
+#else |
+ return rtc::scoped_ptr<RtcEventLog>(new RtcEventLogNullImpl()); |
+#endif // ENABLE_RTC_EVENT_LOG |
} |
} // namespace webrtc |