123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226 |
- #ifndef BOOST_BEAST_CORE_RATE_POLICY_HPP
- #define BOOST_BEAST_CORE_RATE_POLICY_HPP
- #include <boost/beast/core/detail/config.hpp>
- #include <cstdint>
- #include <limits>
- namespace boost {
- namespace beast {
- class rate_policy_access
- {
- private:
- #ifndef BOOST_BEAST_DOXYGEN
- template<class, class, class>
- friend class basic_stream;
- #endif
- template<class Policy>
- static
- std::size_t
- available_read_bytes(Policy& policy)
- {
- return policy.available_read_bytes();
- }
- template<class Policy>
- static
- std::size_t
- available_write_bytes(Policy& policy)
- {
- return policy.available_write_bytes();
- }
- template<class Policy>
- static
- void
- transfer_read_bytes(
- Policy& policy, std::size_t n)
- {
- return policy.transfer_read_bytes(n);
- }
- template<class Policy>
- static
- void
- transfer_write_bytes(
- Policy& policy, std::size_t n)
- {
- return policy.transfer_write_bytes(n);
- }
- template<class Policy>
- static
- void
- on_timer(Policy& policy)
- {
- return policy.on_timer();
- }
- };
- class unlimited_rate_policy
- {
- #ifndef BOOST_BEAST_DOXYGEN
- friend class rate_policy_access;
- #endif
- static std::size_t constexpr all =
- (std::numeric_limits<std::size_t>::max)();
- std::size_t
- available_read_bytes() const noexcept
- {
- return all;
- }
- std::size_t
- available_write_bytes() const noexcept
- {
- return all;
- }
- void
- transfer_read_bytes(std::size_t) const noexcept
- {
- }
- void
- transfer_write_bytes(std::size_t) const noexcept
- {
- }
- void
- on_timer() const noexcept
- {
- }
- };
- class simple_rate_policy
- {
- #ifndef BOOST_BEAST_DOXYGEN
- friend class rate_policy_access;
- #endif
- static std::size_t constexpr all =
- (std::numeric_limits<std::size_t>::max)();
- std::size_t rd_remain_ = all;
- std::size_t wr_remain_ = all;
- std::size_t rd_limit_ = all;
- std::size_t wr_limit_ = all;
- std::size_t
- available_read_bytes() const noexcept
- {
- return rd_remain_;
- }
- std::size_t
- available_write_bytes() const noexcept
- {
- return wr_remain_;
- }
- void
- transfer_read_bytes(std::size_t n) noexcept
- {
- if( rd_remain_ != all)
- rd_remain_ =
- (n < rd_remain_) ? rd_remain_ - n : 0;
- }
- void
- transfer_write_bytes(std::size_t n) noexcept
- {
- if( wr_remain_ != all)
- wr_remain_ =
- (n < wr_remain_) ? wr_remain_ - n : 0;
- }
- void
- on_timer() noexcept
- {
- rd_remain_ = rd_limit_;
- wr_remain_ = wr_limit_;
- }
- public:
-
- void
- read_limit(std::size_t bytes_per_second) noexcept
- {
- rd_limit_ = bytes_per_second;
- if( rd_remain_ > bytes_per_second)
- rd_remain_ = bytes_per_second;
- }
-
- void
- write_limit(std::size_t bytes_per_second) noexcept
- {
- wr_limit_ = bytes_per_second;
- if( wr_remain_ > bytes_per_second)
- wr_remain_ = bytes_per_second;
- }
- };
- }
- }
- #endif
|