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

#ifndef SERVICES_NETWORK_P2P_SOCKET_THROTTLER_H_
#define SERVICES_NETWORK_P2P_SOCKET_THROTTLER_H_

#include <stddef.h>

#include <memory>

#include "base/component_export.h"
#include "third_party/webrtc/rtc_base/data_rate_limiter.h"

namespace network {

// A very simple message throtller. User of this class must drop the packet if
// DropNextPacket returns false for that packet. This method verifies the
// current sendrate against the required sendrate.

class COMPONENT_EXPORT(NETWORK_SERVICE) P2PMessageThrottler {
 public:
  P2PMessageThrottler();

  P2PMessageThrottler(const P2PMessageThrottler&) = delete;
  P2PMessageThrottler& operator=(const P2PMessageThrottler&) = delete;

  virtual ~P2PMessageThrottler();

  bool DropNextPacket(size_t packet_len);
  void SetSendIceBandwidth(int bandwith_kbps);

 private:
  std::unique_ptr<webrtc::DataRateLimiter> rate_limiter_;
};

}  // namespace network

#endif  // SERVICES_NETWORK_P2P_SOCKET_THROTTLER_H_