feat(SnowflakeUtil): allow snowflakes to be generated dynamically

This commit is contained in:
SpaceEEC
2018-04-27 20:34:48 +02:00
parent 44fefdfa49
commit d9a091f674

View File

@@ -27,11 +27,18 @@ class SnowflakeUtil {
/** /**
* Generates a Discord snowflake. * Generates a Discord snowflake.
* <info>This hardcodes the worker ID as 1 and the process ID as 0.</info> * <info>This hardcodes the worker ID as 1 and the process ID as 0.</info>
* @param {number|Date} [timestamp=Date.now()] Timestamp or date of the snowflake to generate
* @returns {Snowflake} The generated snowflake * @returns {Snowflake} The generated snowflake
*/ */
static generate() { static generate(timestamp = Date.now()) {
if (timestamp instanceof Date) timestamp = timestamp.getTime();
if (typeof timestamp !== 'number' || isNaN(timestamp)) {
throw new TypeError(
`"timestamp" argument must be a number (received ${isNaN(timestamp) ? 'NaN' : typeof timestamp})`
);
}
if (INCREMENT >= 4095) INCREMENT = 0; if (INCREMENT >= 4095) INCREMENT = 0;
const BINARY = `${pad((Date.now() - EPOCH).toString(2), 42)}0000100000${pad((INCREMENT++).toString(2), 12)}`; const BINARY = `${pad((timestamp - EPOCH).toString(2), 42)}0000100000${pad((INCREMENT++).toString(2), 12)}`;
return Long.fromString(BINARY, 2).toString(); return Long.fromString(BINARY, 2).toString();
} }