97 lines
2.6 KiB
C++
97 lines
2.6 KiB
C++
|
/*
|
||
|
|
* Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE file in the root of the source
|
||
|
|
* tree. An additional intellectual property rights grant can be found
|
||
|
|
* in the file PATENTS. All contributing project authors may
|
||
|
|
* be found in the AUTHORS file in the root of the source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "webrtc/p2p/client/socketmonitor.h"
|
||
|
|
|
||
|
|
#include "webrtc/base/common.h"
|
||
|
|
|
||
|
|
namespace cricket {
|
||
|
|
|
||
|
|
enum {
|
||
|
|
MSG_MONITOR_POLL,
|
||
|
|
MSG_MONITOR_START,
|
||
|
|
MSG_MONITOR_STOP,
|
||
|
|
MSG_MONITOR_SIGNAL
|
||
|
|
};
|
||
|
|
|
||
|
|
ConnectionMonitor::ConnectionMonitor(ConnectionStatsGetter* stats_getter,
|
||
|
|
rtc::Thread* worker_thread,
|
||
|
|
rtc::Thread* monitoring_thread) {
|
||
|
|
stats_getter_ = stats_getter;
|
||
|
|
worker_thread_ = worker_thread;
|
||
|
|
monitoring_thread_ = monitoring_thread;
|
||
|
|
monitoring_ = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
ConnectionMonitor::~ConnectionMonitor() {
|
||
|
|
worker_thread_->Clear(this);
|
||
|
|
monitoring_thread_->Clear(this);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ConnectionMonitor::Start(int milliseconds) {
|
||
|
|
rate_ = milliseconds;
|
||
|
|
if (rate_ < 250)
|
||
|
|
rate_ = 250;
|
||
|
|
worker_thread_->Post(RTC_FROM_HERE, this, MSG_MONITOR_START);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ConnectionMonitor::Stop() {
|
||
|
|
worker_thread_->Post(RTC_FROM_HERE, this, MSG_MONITOR_STOP);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ConnectionMonitor::OnMessage(rtc::Message *message) {
|
||
|
|
rtc::CritScope cs(&crit_);
|
||
|
|
switch (message->message_id) {
|
||
|
|
case MSG_MONITOR_START:
|
||
|
|
ASSERT(rtc::Thread::Current() == worker_thread_);
|
||
|
|
if (!monitoring_) {
|
||
|
|
monitoring_ = true;
|
||
|
|
PollConnectionStats_w();
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case MSG_MONITOR_STOP:
|
||
|
|
ASSERT(rtc::Thread::Current() == worker_thread_);
|
||
|
|
if (monitoring_) {
|
||
|
|
monitoring_ = false;
|
||
|
|
worker_thread_->Clear(this);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case MSG_MONITOR_POLL:
|
||
|
|
ASSERT(rtc::Thread::Current() == worker_thread_);
|
||
|
|
PollConnectionStats_w();
|
||
|
|
break;
|
||
|
|
|
||
|
|
case MSG_MONITOR_SIGNAL: {
|
||
|
|
ASSERT(rtc::Thread::Current() == monitoring_thread_);
|
||
|
|
std::vector<ConnectionInfo> infos = connection_infos_;
|
||
|
|
crit_.Leave();
|
||
|
|
SignalUpdate(this, infos);
|
||
|
|
crit_.Enter();
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ConnectionMonitor::PollConnectionStats_w() {
|
||
|
|
ASSERT(rtc::Thread::Current() == worker_thread_);
|
||
|
|
rtc::CritScope cs(&crit_);
|
||
|
|
|
||
|
|
// Gather connection infos
|
||
|
|
stats_getter_->GetConnectionStats(&connection_infos_);
|
||
|
|
|
||
|
|
// Signal the monitoring thread, start another poll timer
|
||
|
|
monitoring_thread_->Post(RTC_FROM_HERE, this, MSG_MONITOR_SIGNAL);
|
||
|
|
worker_thread_->PostDelayed(RTC_FROM_HERE, rate_, this, MSG_MONITOR_POLL);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace cricket
|