| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "google_apis/gcm/engine/heartbeat_manager.h" | 5 #include "google_apis/gcm/engine/heartbeat_manager.h" |
| 6 | 6 |
| 7 #include "base/timer/timer.h" |
| 7 #include "google_apis/gcm/protocol/mcs.pb.h" | 8 #include "google_apis/gcm/protocol/mcs.pb.h" |
| 8 #include "net/base/network_change_notifier.h" | 9 #include "net/base/network_change_notifier.h" |
| 9 | 10 |
| 10 namespace gcm { | 11 namespace gcm { |
| 11 | 12 |
| 12 namespace { | 13 namespace { |
| 13 // The default heartbeat when on a mobile or unknown network . | 14 // The default heartbeat when on a mobile or unknown network . |
| 14 const int64 kCellHeartbeatDefaultMs = 1000 * 60 * 28; // 28 minutes. | 15 const int64 kCellHeartbeatDefaultMs = 1000 * 60 * 28; // 28 minutes. |
| 15 // The default heartbeat when on WiFi (also used for ethernet). | 16 // The default heartbeat when on WiFi (also used for ethernet). |
| 16 const int64 kWifiHeartbeatDefaultMs = 1000 * 60 * 15; // 15 minutes. | 17 const int64 kWifiHeartbeatDefaultMs = 1000 * 60 * 15; // 15 minutes. |
| 17 // The default heartbeat ack interval. | 18 // The default heartbeat ack interval. |
| 18 const int64 kHeartbeatAckDefaultMs = 1000 * 60 * 1; // 1 minute. | 19 const int64 kHeartbeatAckDefaultMs = 1000 * 60 * 1; // 1 minute. |
| 19 } // namespace | 20 } // namespace |
| 20 | 21 |
| 21 HeartbeatManager::HeartbeatManager() | 22 HeartbeatManager::HeartbeatManager(scoped_ptr<base::Timer> heartbeat_timer) |
| 22 : waiting_for_ack_(false), | 23 : waiting_for_ack_(false), |
| 23 heartbeat_interval_ms_(0), | 24 heartbeat_interval_ms_(0), |
| 24 server_interval_ms_(0), | 25 server_interval_ms_(0), |
| 25 heartbeat_timer_(true /* retain user task */, | 26 heartbeat_timer_(heartbeat_timer.Pass()), |
| 26 false /* not repeating */), | |
| 27 weak_ptr_factory_(this) {} | 27 weak_ptr_factory_(this) {} |
| 28 | 28 |
| 29 HeartbeatManager::~HeartbeatManager() {} | 29 HeartbeatManager::~HeartbeatManager() {} |
| 30 | 30 |
| 31 void HeartbeatManager::Start( | 31 void HeartbeatManager::Start( |
| 32 const base::Closure& send_heartbeat_callback, | 32 const base::Closure& send_heartbeat_callback, |
| 33 const base::Closure& trigger_reconnect_callback) { | 33 const base::Closure& trigger_reconnect_callback) { |
| 34 DCHECK(!send_heartbeat_callback.is_null()); | 34 DCHECK(!send_heartbeat_callback.is_null()); |
| 35 DCHECK(!trigger_reconnect_callback.is_null()); | 35 DCHECK(!trigger_reconnect_callback.is_null()); |
| 36 send_heartbeat_callback_ = send_heartbeat_callback; | 36 send_heartbeat_callback_ = send_heartbeat_callback; |
| 37 trigger_reconnect_callback_ = trigger_reconnect_callback; | 37 trigger_reconnect_callback_ = trigger_reconnect_callback; |
| 38 | 38 |
| 39 // Kicks off the timer. | 39 // Kicks off the timer. |
| 40 waiting_for_ack_ = false; | 40 waiting_for_ack_ = false; |
| 41 RestartTimer(); | 41 RestartTimer(); |
| 42 } | 42 } |
| 43 | 43 |
| 44 void HeartbeatManager::Stop() { | 44 void HeartbeatManager::Stop() { |
| 45 heartbeat_timer_.Stop(); | 45 heartbeat_timer_->Stop(); |
| 46 waiting_for_ack_ = false; | 46 waiting_for_ack_ = false; |
| 47 } | 47 } |
| 48 | 48 |
| 49 void HeartbeatManager::OnHeartbeatAcked() { | 49 void HeartbeatManager::OnHeartbeatAcked() { |
| 50 if (!heartbeat_timer_.IsRunning()) | 50 if (!heartbeat_timer_->IsRunning()) |
| 51 return; | 51 return; |
| 52 | 52 |
| 53 DCHECK(!send_heartbeat_callback_.is_null()); | 53 DCHECK(!send_heartbeat_callback_.is_null()); |
| 54 DCHECK(!trigger_reconnect_callback_.is_null()); | 54 DCHECK(!trigger_reconnect_callback_.is_null()); |
| 55 waiting_for_ack_ = false; | 55 waiting_for_ack_ = false; |
| 56 RestartTimer(); | 56 RestartTimer(); |
| 57 } | 57 } |
| 58 | 58 |
| 59 void HeartbeatManager::UpdateHeartbeatConfig( | 59 void HeartbeatManager::UpdateHeartbeatConfig( |
| 60 const mcs_proto::HeartbeatConfig& config) { | 60 const mcs_proto::HeartbeatConfig& config) { |
| 61 if (!config.IsInitialized() || | 61 if (!config.IsInitialized() || |
| 62 !config.has_interval_ms() || | 62 !config.has_interval_ms() || |
| 63 config.interval_ms() <= 0) { | 63 config.interval_ms() <= 0) { |
| 64 return; | 64 return; |
| 65 } | 65 } |
| 66 DVLOG(1) << "Updating heartbeat interval to " << config.interval_ms(); | 66 DVLOG(1) << "Updating heartbeat interval to " << config.interval_ms(); |
| 67 server_interval_ms_ = config.interval_ms(); | 67 server_interval_ms_ = config.interval_ms(); |
| 68 } | 68 } |
| 69 | 69 |
| 70 base::TimeTicks HeartbeatManager::GetNextHeartbeatTime() const { | 70 base::TimeTicks HeartbeatManager::GetNextHeartbeatTime() const { |
| 71 if (heartbeat_timer_.IsRunning()) | 71 if (heartbeat_timer_->IsRunning()) |
| 72 return heartbeat_timer_.desired_run_time(); | 72 return heartbeat_timer_->desired_run_time(); |
| 73 else | 73 else |
| 74 return base::TimeTicks(); | 74 return base::TimeTicks(); |
| 75 } | 75 } |
| 76 | 76 |
| 77 void HeartbeatManager::OnHeartbeatTriggered() { | 77 void HeartbeatManager::OnHeartbeatTriggered() { |
| 78 if (waiting_for_ack_) { | 78 if (waiting_for_ack_) { |
| 79 LOG(WARNING) << "Lost connection to MCS, reconnecting."; | 79 LOG(WARNING) << "Lost connection to MCS, reconnecting."; |
| 80 Stop(); | 80 Stop(); |
| 81 trigger_reconnect_callback_.Run(); | 81 trigger_reconnect_callback_.Run(); |
| 82 return; | 82 return; |
| (...skipping 19 matching lines...) Expand all Loading... |
| 102 // For unknown connections, use the longer cellular heartbeat interval. | 102 // For unknown connections, use the longer cellular heartbeat interval. |
| 103 heartbeat_interval_ms_ = kCellHeartbeatDefaultMs; | 103 heartbeat_interval_ms_ = kCellHeartbeatDefaultMs; |
| 104 } | 104 } |
| 105 DVLOG(1) << "Sending next heartbeat in " | 105 DVLOG(1) << "Sending next heartbeat in " |
| 106 << heartbeat_interval_ms_ << " ms."; | 106 << heartbeat_interval_ms_ << " ms."; |
| 107 } else { | 107 } else { |
| 108 heartbeat_interval_ms_ = kHeartbeatAckDefaultMs; | 108 heartbeat_interval_ms_ = kHeartbeatAckDefaultMs; |
| 109 DVLOG(1) << "Resetting timer for ack with " | 109 DVLOG(1) << "Resetting timer for ack with " |
| 110 << heartbeat_interval_ms_ << " ms interval."; | 110 << heartbeat_interval_ms_ << " ms interval."; |
| 111 } | 111 } |
| 112 heartbeat_timer_.Start(FROM_HERE, | 112 heartbeat_timer_->Start(FROM_HERE, |
| 113 base::TimeDelta::FromMilliseconds( | 113 base::TimeDelta::FromMilliseconds( |
| 114 heartbeat_interval_ms_), | 114 heartbeat_interval_ms_), |
| 115 base::Bind(&HeartbeatManager::OnHeartbeatTriggered, | 115 base::Bind(&HeartbeatManager::OnHeartbeatTriggered, |
| 116 weak_ptr_factory_.GetWeakPtr())); | 116 weak_ptr_factory_.GetWeakPtr())); |
| 117 } | 117 } |
| 118 | 118 |
| 119 } // namespace gcm | 119 } // namespace gcm |
| OLD | NEW |