通过定时自动删除元素设置类


问题内容

我正在使用Python编写软件,并且需要一个类来存储一组元素(顺序无关紧要,并且不是重复的元素),就像Python类一样set,但是我需要在一段时间后自动删除这些元素。

为此,我想覆盖该set.add方法,并在此超时后添加具有默认值的参数。

我的问题是实现该目标的最佳方法:线程?类似gobject的超时?

欢迎所有建议!


问题答案:

只是一个想法(当然,这不是最好的):使用字典存储每个添加项的时间戳以及每个项目的具体超时。然后,当您要检查某项是否在集合中时,您必须将当前时间与字典中的值进行比较。这样,您无需在超时结束时启动新线程来删除每个项目(只需将键保留在字典中,并在再次添加该项目时对其进行更新)。

有了这个解决方案,您必须实现__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