Index: extensions/browser/api/cast_channel/logger.cc |
diff --git a/extensions/browser/api/cast_channel/logger.cc b/extensions/browser/api/cast_channel/logger.cc |
index c1e8c9bc3949ea9debfbd991d4482e34dc5a0210..6ad8165c74988d29a29592b0292f0e030f0d2c72 100644 |
--- a/extensions/browser/api/cast_channel/logger.cc |
+++ b/extensions/browser/api/cast_channel/logger.cc |
@@ -64,10 +64,7 @@ Logger::AggregatedSocketEventLog::~AggregatedSocketEventLog() { |
Logger::Logger(scoped_ptr<base::TickClock> clock, |
base::TimeTicks unix_epoch_time_ticks) |
- : clock_(clock.Pass()), |
- unix_epoch_time_ticks_(unix_epoch_time_ticks), |
- num_evicted_aggregated_socket_events_(0), |
- num_evicted_socket_events_(0) { |
+ : clock_(clock.Pass()), unix_epoch_time_ticks_(unix_epoch_time_ticks) { |
DCHECK(clock_); |
// Logger may not be necessarily be created on the IO thread, but logging |
@@ -123,6 +120,22 @@ void Logger::LogSocketEventWithRv(int channel_id, |
event.set_return_value(rv); |
LogSocketEvent(channel_id, event); |
+ |
+ if ((event_type == proto::SOCKET_READ || event_type == proto::SOCKET_WRITE) && |
+ rv > 0) { |
+ AggregatedSocketEvent& aggregated_socket_event = |
+ aggregated_socket_events_.find(channel_id) |
+ ->second->aggregated_socket_event; |
mark a. foltz
2014/08/13 22:06:31
Is this lookup guaranteed to succeed?
imcheng
2014/08/14 00:24:34
Yes, because this entry was just added in LogSocke
|
+ if (event_type == proto::SOCKET_READ) { |
+ aggregated_socket_event.set_bytes_read( |
+ aggregated_socket_event.bytes_read() + rv); |
+ log_.set_total_bytes_read(log_.total_bytes_read() + rv); |
+ } else { |
+ aggregated_socket_event.set_bytes_written( |
+ aggregated_socket_event.bytes_written() + rv); |
+ log_.set_total_bytes_written(log_.total_bytes_written() + rv); |
+ } |
+ } |
} |
void Logger::LogSocketReadyState(int channel_id, proto::ReadyState new_state) { |
@@ -214,8 +227,11 @@ void Logger::LogSocketEvent(int channel_id, const SocketEvent& socket_event) { |
AggregatedSocketEventLogMap::iterator erase_it = |
aggregated_socket_events_.begin(); |
- num_evicted_aggregated_socket_events_++; |
- num_evicted_socket_events_ += erase_it->second->socket_events.size(); |
+ log_.set_num_evicted_aggregated_socket_events( |
+ log_.num_evicted_aggregated_socket_events() + 1); |
+ log_.set_num_evicted_socket_events( |
+ log_.num_evicted_socket_events() + |
+ erase_it->second->socket_events.size()); |
aggregated_socket_events_.erase(erase_it); |
} |
@@ -230,7 +246,7 @@ void Logger::LogSocketEvent(int channel_id, const SocketEvent& socket_event) { |
std::deque<proto::SocketEvent>& socket_events = it->second->socket_events; |
if (socket_events.size() >= kMaxEventsPerSocket) { |
socket_events.pop_front(); |
- num_evicted_socket_events_++; |
+ log_.set_num_evicted_socket_events(log_.num_evicted_socket_events() + 1); |
} |
socket_events.push_back(socket_event); |
@@ -240,9 +256,7 @@ bool Logger::LogToString(std::string* output) const { |
output->clear(); |
Log log; |
- log.set_num_evicted_aggregated_socket_events( |
- num_evicted_aggregated_socket_events_); |
- log.set_num_evicted_socket_events(num_evicted_socket_events_); |
+ log.CopyFrom(log_); |
for (AggregatedSocketEventLogMap::const_iterator it = |
aggregated_socket_events_.begin(); |
@@ -268,8 +282,7 @@ bool Logger::LogToString(std::string* output) const { |
void Logger::Reset() { |
aggregated_socket_events_.clear(); |
- num_evicted_aggregated_socket_events_ = 0; |
- num_evicted_socket_events_ = 0; |
+ log_.Clear(); |
} |
} // namespace cast_channel |