mirror of
https://github.com/uowuo/abaddon.git
synced 2026-01-14 05:00:05 +00:00
66 lines
1.7 KiB
C++
66 lines
1.7 KiB
C++
#include "snowflake.hpp"
|
|
#include <ctime>
|
|
#include <iomanip>
|
|
#include <chrono>
|
|
|
|
constexpr static uint64_t DiscordEpochSeconds = 1420070400;
|
|
|
|
Snowflake::Snowflake()
|
|
: m_num(Invalid) {}
|
|
|
|
Snowflake::Snowflake(uint64_t n)
|
|
: m_num(n) {}
|
|
|
|
Snowflake::Snowflake(const std::string &str) {
|
|
if (str.size())
|
|
m_num = std::stoull(str);
|
|
else
|
|
m_num = Invalid;
|
|
}
|
|
Snowflake::Snowflake(const Glib::ustring &str) {
|
|
if (str.size())
|
|
m_num = std::strtoull(str.c_str(), nullptr, 10);
|
|
else
|
|
m_num = Invalid;
|
|
};
|
|
|
|
Snowflake Snowflake::FromNow() {
|
|
using namespace std::chrono;
|
|
// not guaranteed to work but it probably will anyway
|
|
static uint64_t counter = 0;
|
|
const auto millis_since_epoch = static_cast<uint64_t>(duration_cast<milliseconds>(system_clock::now().time_since_epoch()).count());
|
|
const auto epoch = millis_since_epoch - DiscordEpochSeconds * 1000;
|
|
uint64_t snowflake = epoch << 22;
|
|
// worker id and process id would be OR'd in here but there's no point
|
|
snowflake |= counter++ % 4096;
|
|
return snowflake;
|
|
}
|
|
|
|
bool Snowflake::IsValid() const {
|
|
return m_num != Invalid;
|
|
}
|
|
|
|
std::string Snowflake::GetLocalTimestamp() const {
|
|
const time_t secs_since_epoch = (m_num / 4194304000) + 1420070400;
|
|
const std::tm tm = *localtime(&secs_since_epoch);
|
|
std::stringstream ss;
|
|
const static std::locale locale("");
|
|
ss.imbue(locale);
|
|
ss << std::put_time(&tm, "%X %x");
|
|
return ss.str();
|
|
}
|
|
|
|
void from_json(const nlohmann::json &j, Snowflake &s) {
|
|
if (j.is_string()) {
|
|
std::string tmp;
|
|
j.get_to(tmp);
|
|
s.m_num = std::stoull(tmp);
|
|
} else {
|
|
j.get_to(s.m_num);
|
|
}
|
|
}
|
|
|
|
void to_json(nlohmann::json &j, const Snowflake &s) {
|
|
j = std::to_string(s);
|
|
}
|