components/proximity_auth/throttled_bluetooth_connection_finder.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/proximity_auth/throttled_bluetooth_connection_finder.h"

#include <utility>

#include "base/bind.h"
#include "base/location.h"
#include "base/task_runner.h"
#include "base/time/time.h"
#include "components/proximity_auth/bluetooth_connection_finder.h"
#include "components/proximity_auth/bluetooth_throttler.h"

namespace proximity_auth {

ThrottledBluetoothConnectionFinder::ThrottledBluetoothConnectionFinder(
    scoped_ptr<BluetoothConnectionFinder> connection_finder,
    scoped_refptr<base::TaskRunner> task_runner,
    BluetoothThrottler* throttler)
    : connection_finder_(std::move(connection_finder)),
      task_runner_(task_runner),
      throttler_(throttler),
      weak_ptr_factory_(this) {}

ThrottledBluetoothConnectionFinder::~ThrottledBluetoothConnectionFinder() {
}

void ThrottledBluetoothConnectionFinder::Find(
    const ConnectionCallback& connection_callback) {
  const base::TimeDelta delay = throttler_->GetDelay();

  // Wait, if needed.
  if (delay != base::TimeDelta()) {
    task_runner_->PostDelayedTask(
        FROM_HERE,
        base::Bind(&ThrottledBluetoothConnectionFinder::Find,
                   weak_ptr_factory_.GetWeakPtr(), connection_callback),
        delay);
    return;
  }

  connection_finder_->Find(
      base::Bind(&ThrottledBluetoothConnectionFinder::OnConnection,
                 weak_ptr_factory_.GetWeakPtr(), connection_callback));
}

void ThrottledBluetoothConnectionFinder::OnConnection(
    const ConnectionCallback& connection_callback,
    scoped_ptr<Connection> connection) {
  throttler_->OnConnection(connection.get());
  connection_callback.Run(std::move(connection));
}

}  // namespace proximity_auth