2019-02-11 10:46:39 +08:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include "common.h"
|
|
|
|
|
|
|
|
|
|
class RateLimiter
|
|
|
|
|
{
|
|
|
|
|
public:
|
|
|
|
|
RateLimiter(int requests, int delay) : requestsLeft(requests), delay(delay) {}
|
|
|
|
|
bool Request() { CreateTimerQueueTimer(&DUMMY, timerQueue, [](void* This, BOOLEAN) { ++((RateLimiter*)This)->requestsLeft; }, this, delay, 0, 0); return --requestsLeft > 0; }
|
|
|
|
|
int delay;
|
|
|
|
|
|
|
|
|
|
private:
|
2019-02-20 15:19:30 +08:00
|
|
|
inline static HANDLE DUMMY;
|
2019-02-11 10:46:39 +08:00
|
|
|
std::atomic<int> requestsLeft;
|
|
|
|
|
AutoHandle<Functor<DeleteTimerQueue>> timerQueue = CreateTimerQueue();
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
inline std::wstring StringToWideString(const std::string& text)
|
|
|
|
|
{
|
|
|
|
|
std::vector<wchar_t> buffer(text.size() + 1);
|
|
|
|
|
MultiByteToWideChar(CP_UTF8, 0, text.c_str(), -1, buffer.data(), buffer.size());
|
|
|
|
|
return buffer.data();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
inline std::string WideStringToString(const std::wstring& text)
|
|
|
|
|
{
|
|
|
|
|
std::vector<char> buffer((text.size() + 1) * 4);
|
|
|
|
|
WideCharToMultiByte(CP_UTF8, 0, text.c_str(), -1, buffer.data(), buffer.size(), nullptr, nullptr);
|
|
|
|
|
return buffer.data();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
inline const std::wstring NEWLINE = L"\n";
|
