-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathlru.py
More file actions
34 lines (25 loc) · 856 Bytes
/
lru.py
File metadata and controls
34 lines (25 loc) · 856 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#! /usr/vin/env python
# -*-coding:utf8-*-
from collections import OrderedDict
class LRUCache(OrderedDict):
'''不能存储可变类型对象,不能并发访问set()'''
def __init__(self,capacity):
self.capacity = capacity
self.cache = OrderedDict()
def get(self,key):
if self.cache.has_key(key):
value = self.cache.pop(key)
self.cache[key] = value
else:
value = None
return value
def set(self,key,value):
if self.cache.has_key(key):
value = self.cache.pop(key)
self.cache[key] = value
else:
if len(self.cache) == self.capacity:
self.cache.popitem(last = False) #pop出第一个item
self.cache[key] = value
else:
self.cache[key] = value