python - 设置具有定时自动删除元素的类

Question

我正在用 Python 编写软件，我需要一个类来存储一组元素（顺序不相关且不重复元素），例如 Python 类set，但我需要在一段时间后自动删除这些元素.

为此，我想覆盖该set.add方法，在此超时中添加一个具有默认值的参数。

我的问题是实现它的最佳方法：线程？类似 gobject 的超时？

欢迎所有建议！

Answer 1

只是一个想法（当然，它不是最好的）：使用字典来存储每个添加的时间戳以及每个项目的具体超时。然后，当您要检查某个项目是否在集合中时，您必须将当前时间与字典中的值进行比较。这样你就不需要在超时完成时启动一个新线程来删除每个项目（只需将键保留在字典中，如果再次添加项目则更新它）。

使用此解决方案，您必须实施,以确保__contains__并返回一致的结果__iter__。add'a' in mysetiter(myset)

import time

class TimedSet(set):
    def __init__(self):
        self.__table = {}
    def add(self, item, timeout=1):
        self.__table[item] = time.time() + timeout
        set.add(self, item)
    def __contains__(self, item):
        return time.time() < self.__table.get(item)
    def __iter__(self):
        for item in set.__iter__(self):
            if time.time() < self.__table.get(item):
                yield item

以及一个可能的用法示例：

t_set = TimedSet()
t_set.add('a')
time.sleep(0.6)
print 'a' in t_set
time.sleep(0.6)
print 'a' in t_set

t_set.add('x', 0.3)
t_set.add('y', 0.4)
t_set.add('z', 0.5)
time.sleep(0.35)
for item in t_set:
    print item

Answer 2

我看不出有任何理由不使用多线程。因为它很容易实现，代码最少。大致是这样的：

import threading
import time

def ttl_set_remove(my_set, item, ttl):
    time.sleep(ttl)
    my_set.remove(item)

class MySet(set):
    def add(self, item, ttl):
        set.add(self, item)
        t = threading.Thread(target=ttl_set_remove, args=(self, item, ttl))
        t.start()

测试：

s = MySet()
s.add('a', 20)
s.add('b', 10)
s.add('c', 2)

print(s)
time.sleep(5)
print(s)

>>> 
MySet({'c', 'b', 'a'})
MySet({'b', 'a'})