linux页面算法源码,LRU算法原理解析

LRU是Least Recently Used的缩写，即最近最少使用，常用于页面置换算法，是为虚拟页式存储管理服务的。

现代操作系统提供了一种对主存的抽象概念虚拟内存，来对主存进行更好地管理。他将主存看成是一个存储在磁盘上的地址空间的高速缓存，在主存中只保存活动区域，并根据需要在主存和磁盘之间来回传送数据。虚拟内存被组织为存放在磁盘上的N个连续的字节组成的数组，每个字节都有唯一的虚拟地址，作为到数组的索引。虚拟内存被分割为大小固定的数据块虚拟页(Virtual Page,VP)，这些数据块作为主存和磁盘之间的传输单元。类似地，物理内存被分割为物理页(Physical Page,PP)。

虚拟内存使用页表来记录和判断一个虚拟页是否缓存在物理内存中：

如上图所示，当CPU访问虚拟页VP3时，发现VP3并未缓存在物理内存之中，这称之为缺页，现在需要将VP3从磁盘复制到物理内存中，但在此之前，为了保持原有空间的大小，需要在物理内存中选择一个牺牲页，将其复制到磁盘中，这称之为交换或者页面调度，图中的牺牲页为VP4。把哪个页面调出去可以达到调动尽量少的目的？最好是每次调换出的页面是所有内存页面中最迟将被使用的——这可以最大限度的推迟页面调换，这种算法，被称为理想页面置换算法，但这种算法很难完美达到。

为了尽量减少与理想算法的差距，产生了各种精妙的算法，LRU算法便是其中一个。

LRU原理

LRU 算法的设计原则是：如果一个数据在最近一段时间没有被访问到，那么在将来它被访问的可能性也很小。也就是说，当限定的空间已存满数据时，应当把最久没有被访问到的数据淘汰。

根据LRU原理和Redis实现所示，假定系统为某进程分配了3个物理块，进程运行时的页面走向为 7 0 1 2 0 3 0 4，开始时3个物理块均为空，那么LRU算法是如下工作的：

基于哈希表和双向链表的LRU算法实现

如果要自己实现一个LRU算法，可以用哈希表加双向链表实现：

设计思路是，使用哈希表存储 key，值为链表中的节点，节点中存储值，双向链表来记录节点的顺序，头部为最近访问节点。

LRU算法中有两种基本操作：

get(key)：查询key对应的节点，如果key存在，将节点移动至链表头部。

set(key, value)： 设置key对应的节点的值。如果key不存在，则新建节点，置于链表开头。如果链表长度超标，则将处于尾部的最后一个节点去掉。如果节点存在，更新节点的值，同时将节点置于链表头部。

LRU缓存机制

leetcode上有一道关于LRU缓存机制的题目：

运用你所掌握的数据结构，设计和实现一个 LRU (最近最少使用) 缓存机制。它应该支持以下操作： 获取数据 get 和 写入数据 put 。

获取数据 get(key) - 如果密钥 (key) 存在于缓存中，则获取密钥的值(总是正数)，否则返回 -1。 写入数据 put(key, value) - 如果密钥不存在，则写入其数据值。当缓存容量达到上限时，它应该在写入新数据之前删除最近最少使用的数据值，从而为新的数据值留出空间。

进阶:

你是否可以在 O(1) 时间复杂度内完成这两种操作？

示例:

LRUCache cache = new LRUCache( 2 /* 缓存容量 */ );

cache.put(1, 1);

cache.put(2, 2);

cache.get(1); // 返回 1

cache.put(3, 3); // 该操作会使得密钥 2 作废

cache.get(2); // 返回 -1 (未找到)

cache.put(4, 4); // 该操作会使得密钥 1 作废

cache.get(1); // 返回 -1 (未找到)

cache.get(3); // 返回 3

cache.get(4); // 返回 4

我们可以自己实现双向链表，也可以使用现成的数据结构，Python中的数据结构OrderedDict是一个有序哈希表，可以记住加入哈希表的键的顺序，相当于同时实现了哈希表与双向链表。OrderedDict是将最新数据放置于末尾的:

In [35]: from collections import OrderedDict

In [36]: lru = OrderedDict()

In [37]: lru[1] = 1

In [38]: lru[2] = 2

In [39]: lru

Out[39]: OrderedDict([(1, 1), (2, 2)])

In [40]: lru.popitem()

Out[40]: (2, 2)

OrderedDict有两个重要方法：

popitem(last=True): 返回一个键值对，当last=True时，按照LIFO的顺序，否则按照FIFO的顺序。

move_to_end(key, last=True): 将现有 key 移动到有序字典的任一端。 如果 last 为True(默认)则将元素移至末尾；如果 last 为False则将元素移至开头。

删除数据时，可以使用popitem(last=False)将开头最近未访问的键值对删除。访问或者设置数据时，使用move_to_end(key, last=True)将键值对移动至末尾。

代码实现：

from collections import OrderedDict

class LRUCache:

def __init__(self, capacity: int):

self.lru = OrderedDict()

self.capacity = capacity

def get(self, key: int) -> int:

self._update(key)

return self.lru.get(key, -1)

def put(self, key: int, value: int) -> None:

self._update(key)

self.lru[key] = value

if len(self.lru) > self.capacity:

self.lru.popitem(False)

def _update(self, key: int):

if key in self.lru:

self.lru.move_to_end(key)

OrderedDict源码分析

OrderedDict其实也是用哈希表与双向链表实现的：

class OrderedDict(dict):

'Dictionary that remembers insertion order'

# An inherited dict maps keys to values.

# The inherited dict provides __getitem__, __len__, __contains__, and get.

# The remaining methods are order-aware.

# Big-O running times for all methods are the same as regular dictionaries.

# The internal self.__map dict maps keys to links in a doubly linked list.

# The circular doubly linked list starts and ends with a sentinel element.

# The sentinel element never gets deleted (this simplifies the algorithm).

# The sentinel is in self.__hardroot with a weakref proxy in self.__root.

# The prev links are weakref proxies (to prevent circular references).

# Individual links are kept alive by the hard reference in self.__map.

# Those hard references disappear when a key is deleted from an OrderedDict.

def __init__(*args, **kwds):

'''Initialize an ordered dictionary.  The signature is the same as

regular dictionaries.  Keyword argument order is preserved.

'''

if not args:

raise TypeError("descriptor '__init__' of 'OrderedDict' object "

"needs an argument")

self, *args = args

if len(args) > 1:

raise TypeError('expected at most 1 arguments, got %d' % len(args))

try:

self.__root

except AttributeError:

self.__hardroot = _Link()

self.__root = root = _proxy(self.__hardroot)

root.prev = root.next = root

self.__map = {}

self.__update(*args, **kwds)

def __setitem__(self, key, value,

dict_setitem=dict.__setitem__, proxy=_proxy, Link=_Link):

'od.__setitem__(i, y) <==> od[i]=y'

# Setting a new item creates a new link at the end of the linked list,

# and the inherited dictionary is updated with the new key/value pair.

if key not in self:

self.__map[key] = link = Link()

root = self.__root

last = root.prev

link.prev, link.next, link.key = last, root, key

last.next = link

root.prev = proxy(link)

dict_setitem(self, key, value)

由源码看出，OrderedDict使用self.__map = {}作为哈希表，其中保存了key与链表中的节点Link()的键值对，self.__map[key] = link = Link():

class _Link(object):

__slots__ = 'prev', 'next', 'key', '__weakref__'

节点Link()中保存了指向前一个节点的指针prev，指向后一个节点的指针next以及key值。

而且，这里的链表是一个环形双向链表,OrderedDict使用一个哨兵元素root作为链表的head与tail：

self.__hardroot = _Link()

self.__root = root = _proxy(self.__hardroot)

root.prev = root.next = root

由__setitem__可知，向OrderedDict中添加新值时，链表变为如下的环形结构：

next            next            next

root new node1 new node2 root

prev            prev            prev

root.next为链表的第一个节点，root.prev为链表的最后一个节点。

由于OrderedDict继承自dict，键值对是保存在OrderedDict自身中的，链表节点中只保存了key，并未保存value。

如果我们要自己实现的话，无需如此复杂，可以将value置于节点之中，链表只需要实现插入最前端与移除最后端节点的功能即可：

from _weakref import proxy as _proxy

class Node:

__slots__ = ('prev', 'next', 'key', 'value', '__weakref__')

class LRUCache:

def __init__(self, capacity: int):

self.__hardroot = Node()

self.__root = root = _proxy(self.__hardroot)

root.prev = root.next = root

self.__map = {}

self.capacity = capacity

def get(self, key: int) -> int:

if key in self.__map:

self.move_to_head(key)

return self.__map[key].value

else:

return -1

def put(self, key: int, value: int) -> None:

if key in self.__map:

node = self.__map[key]

node.value = value

self.move_to_head(key)

else:

node = Node()

node.key = key

node.value = value

self.__map[key] = node

self.add_head(node)

if len(self.__map) > self.capacity:

self.rm_tail()

def move_to_head(self, key: int) -> None:

if key in self.__map:

node = self.__map[key]

node.prev.next = node.next

node.next.prev = node.prev

head = self.__root.next

self.__root.next = node

node.prev = self.__root

node.next = head

head.prev = node

def add_head(self, node: Node) -> None:

head = self.__root.next

self.__root.next = node

node.prev = self.__root

node.next = head

head.prev = node

def rm_tail(self) -> None:

tail = self.__root.prev

del self.__map[tail.key]

tail.prev.next = self.__root

self.__root.prev = tail.prev

node-lru-cache

在实际应用中，要实现LRU缓存算法，还要实现很多额外的功能。

有一个用Javascript实现的很好的node-lru-cache包：

var LRU = require("lru-cache")

, options = { max: 500

, length: function (n, key) { return n * 2 + key.length }

, dispose: function (key, n) { n.close() }

, maxAge: 1000 * 60 * 60 }

, cache = new LRU(options)

, otherCache = new LRU(50) // sets just the max size

cache.set("key", "value")

cache.get("key") // "value"

这个包不是用缓存key的数量来判断是否要启动LRU淘汰算法，而是使用保存的键值对的实际大小来判断。选项options中可以设置缓存所占空间的上限max，判断键值对所占空间的函数length，还可以设置键值对的过期时间maxAge等，有兴趣的可以看下。

参考链接