通过定时自动删除元素设置类
问题内容:
我正在使用Python编写软件,并且需要一个类来存储一组元素(顺序无关紧要,并且不是重复的元素),就像Python类一样set
,但是我需要在一段时间后自动删除这些元素。
为此,我想覆盖该set.add
方法,并在此超时后添加具有默认值的参数。
我的问题是实现该目标的最佳方法:线程?类似gobject的超时?
欢迎所有建议!
问题答案:
只是一个想法(当然,这不是最好的):使用字典存储每个添加项的时间戳以及每个项目的具体超时。然后,当您要检查某项是否在集合中时,您必须将当前时间与字典中的值进行比较。这样,您无需在超时结束时启动新线程来删除每个项目(只需将键保留在字典中,并在再次添加该项目时对其进行更新)。
有了这个解决方案,您必须实现__contains__
和__iter__
远离add
,以确保'a' in myset
并iter(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