dsmq 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

dsmq/dsmq.py

@@ -10,6 +10,8 @@ DEFAULT_PORT = 30008
 
 N_RETRIES = 5
 FIRST_RETRY = 0.01  # seconds
+TIME_TO_LIVE = 60  # seconds
+TIME_TO_LIVE = 5  # seconds
 
 
 def start_server(host=DEFAULT_HOST, port=DEFAULT_PORT):
@@ -79,8 +81,9 @@ def _handle_client_connection(socket_conn):
     sqlite_conn = sqlite3.connect("file:mem1?mode=memory&cache=shared")
     cursor = sqlite_conn.cursor()
 
-    creation_time = time.time()
-    last_read = {}
+    client_creation_time = time.time()
+    last_read_times = {}
+    time_of_last_purge = time.time()
 
     while True:
         data = socket_conn.recv(1024)
@@ -123,10 +126,10 @@ VALUES (:timestamp, :topic, :message)
 
         elif msg["action"] == "get":
             try:
-                last_read_time = last_read[topic]
+                last_read_time = last_read_times[topic]
             except KeyError:
-                last_read[topic] = creation_time
-                last_read_time = last_read[topic]
+                last_read_times[topic] = client_creation_time
+                last_read_time = last_read_times[topic]
             msg["last_read_time"] = last_read_time
 
             # This block allows for multiple retries if the database
@@ -159,7 +162,7 @@ AND timestamp = a.min_time
                 result = cursor.fetchall()[0]
                 message = result[0]
                 timestamp = result[1]
-                last_read[topic] = timestamp
+                last_read_times[topic] = timestamp
             except IndexError:
                 # Handle the case where no results are returned
                 message = ""
@@ -169,6 +172,20 @@ AND timestamp = a.min_time
         else:
             print("Action must either be 'put' or 'get'")
 
+        # Periodically clean out messages from the queue that are
+        # past their sell buy date.
+        # This operation is pretty fast. I clock it at 12 us on my machine.
+        if time.time() - time_of_last_purge > TIME_TO_LIVE:
+            cursor.execute(
+                """
+DELETE FROM messages
+WHERE timestamp < :time_threshold
+                """,
+                {"time_threshold": time_of_last_purge}
+            )
+            sqlite_conn.commit()
+            time_of_last_purge = time.time()
+
     sqlite_conn.close()
 
 

{dsmq-0.4.0.dist-info → dsmq-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dsmq
-Version: 0.4.0
+Version: 0.5.0
 Summary: A dead simple message queue
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{dsmq-0.4.0.dist-info → dsmq-0.5.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
 dsmq/__init__.py,sha256=YCgbnQAk8YbtHRyMcU0v2O7RdRhPhlT-vS_q40a7Q6g,50
 dsmq/demo_linux.py,sha256=7yLglGmirDLuuyMxppYSK-dfx2Fg2Q0dIWB4cl2yV1c,622
-dsmq/dsmq.py,sha256=5fo3wQfw2lIXiDGB05sRwVLDN0ZoUSw7e7WO8xOLXJY,5558
+dsmq/dsmq.py,sha256=dKYtGkg8Kwll2sCytCqSI5klFs2cwm2pyXAkyQk4RdU,6206
 dsmq/example_get_client.py,sha256=chFfB2949PBENmgdUc3ASrATq1m4wvHGBzEnOC-o_Xs,296
 dsmq/example_put_client.py,sha256=mUKCRhmUieMZEpHLFWFvzeKB6IR7A8l4tWN8TvPzdKU,348
 dsmq/example_server.py,sha256=kkXOPaaTzVxf9_iIM76zU9pZhkPna_1vcGWkPrhCjus,61
-dsmq-0.4.0.dist-info/METADATA,sha256=T3godcTW__I3b2hj2dgJCjMJEGEo6S1HP_Iq5EQD6Ok,3878
-dsmq-0.4.0.dist-info/WHEEL,sha256=3U_NnUcV_1B1kPkYaPzN-irRckL5VW_lytn0ytO_kRY,87
-dsmq-0.4.0.dist-info/licenses/LICENSE,sha256=3Yu1mAp5VsKmnDtzkiOY7BdmrLeNwwZ3t6iWaLnlL0Y,1071
-dsmq-0.4.0.dist-info/RECORD,,
+dsmq-0.5.0.dist-info/METADATA,sha256=EQISO0WcVsb7smXIie-ZP7pIO1itY8FZfjGw3DPuZDM,3878
+dsmq-0.5.0.dist-info/WHEEL,sha256=3U_NnUcV_1B1kPkYaPzN-irRckL5VW_lytn0ytO_kRY,87
+dsmq-0.5.0.dist-info/licenses/LICENSE,sha256=3Yu1mAp5VsKmnDtzkiOY7BdmrLeNwwZ3t6iWaLnlL0Y,1071
+dsmq-0.5.0.dist-info/RECORD,,