Add start_time in past check

Author: ZipFile

src/sonyflake_turbo/pure.py

@@ -1,5 +1,5 @@
 from threading import Lock
-from time import sleep, time_ns
+from time import sleep, time, time_ns
 from typing import overload
 
 try:
@@ -75,6 +75,9 @@ def __init__(self, *machine_ids: int, start_time: int = SONYFLAKE_EPOCH):
         if not isinstance(start_time, int):
             raise TypeError("start_time must be an integer")
 
+        if start_time >= time():
+            raise ValueError("start_time must be in the past")
+
         self._machine_ids = sorted(machine_ids_set)
         self._max_i = len(machine_ids_set) * SONYFLAKE_SEQUENCE_COUNT
         self._start_time = start_time * 100

src/sonyflake_turbo/sonyflake.c

@@ -78,6 +78,7 @@ static int sonyflake_init(PyObject *py_self, PyObject *args, PyObject *kwargs) {
 	PyObject *start_time_obj = NULL;
 	Py_ssize_t machine_ids_len = 0;
 	long long start_time = 0;
+	struct timespec now = {0, 0};
 	long machine_id = 0;
 	Py_ssize_t i = 0;
 
@@ -148,6 +149,13 @@ static int sonyflake_init(PyObject *py_self, PyObject *args, PyObject *kwargs) {
 		goto err;
 	}
 
+	timespec_get(&now, TIME_UTC);
+
+	if (start_time >= now.tv_sec) {
+		PyErr_SetString(PyExc_ValueError, "start_time must be in the past");
+		goto err;
+	}
+
 	self->start_time.tv_sec = start_time;
 	self->start_time.tv_nsec = 0;
 

tests/test_sonyflake_turbo.py

@@ -3,7 +3,7 @@
 import threading
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
-from time import perf_counter, sleep
+from time import perf_counter, sleep, time
 
 from pytest import mark, raises
 
@@ -64,6 +64,10 @@ def test_non_int_start_time(self, cls: type) -> None:
                 start_time=datetime(2025, 1, 1),
             )
 
+    def test_start_time_in_future(self, cls: type) -> None:
+        with raises(ValueError, match=r"start_time must be in the past"):
+            cls(0x0000, start_time=int(time() + 10))
+
     @mark.parametrize(
         ["use_iter", "n"],
         [(use_iter, n) for use_iter in [True, False] for n in [1, 100, 250000]],