python中的queue.Queue()队列如何持久化存储?
即时程序中途退出重启,queue.Queu还能保存未消费的消息队列。
网友回复
我来介绍几种实现 Queue 持久化的方法:
使用 pickle 序列化存储(简单但不推荐用于生产环境):import queue
import pickle
import atexit
class PersistentQueue:
def __init__(self, filename="queue.pkl"):
self.filename = filename
# 尝试加载已存在的队列数据
try:
with open(self.filename, 'rb') as f:
self.queue = pickle.load(f)
except FileNotFoundError:
self.queue = queue.Queue()
# 注册程序退出时的保存操作
atexit.register(self.save)
def put(self, item):
self.queue.put(item)
self.save() # 每次放入数据后保存
def get(self):
item = self.queue.get()
self.save() # 每次获取数据后保存
return item
def save(self):
with open(self.filename, 'wb') as f:
pickle.dump(self.queue, f) 使用 SQLite 存储(推荐用于小型应用):import sqlite3
import queue
import json
class SQLiteQueue:
def __init__(self, db_path="queue.db"):
self.conn = sqlite3.connect(db_path)
self.cursor = self.conn.cursor()
self.cursor.execute('''
CREATE ...点击查看剩余70%


