You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							437 lines
						
					
					
						
							10 KiB
						
					
					
				
			
		
		
	
	
							437 lines
						
					
					
						
							10 KiB
						
					
					
				/**
 | 
						|
 * Mnemonist LRUCache
 | 
						|
 * ===================
 | 
						|
 *
 | 
						|
 * JavaScript implementation of the LRU Cache data structure. To save up
 | 
						|
 * memory and allocations this implementation represents its underlying
 | 
						|
 * doubly-linked list as static arrays and pointers. Thus, memory is allocated
 | 
						|
 * only once at instantiation and JS objects are never created to serve as
 | 
						|
 * pointers. This also means this implementation does not trigger too many
 | 
						|
 * garbage collections.
 | 
						|
 *
 | 
						|
 * Note that to save up memory, a LRU Cache can be implemented using a singly
 | 
						|
 * linked list by storing predecessors' pointers as hashmap values.
 | 
						|
 * However, this means more hashmap lookups and would probably slow the whole
 | 
						|
 * thing down. What's more, pointers are not the things taking most space in
 | 
						|
 * memory.
 | 
						|
 */
 | 
						|
var Iterator = require('obliterator/iterator'),
 | 
						|
    forEach = require('obliterator/foreach'),
 | 
						|
    typed = require('./utils/typed-arrays.js'),
 | 
						|
    iterables = require('./utils/iterables.js');
 | 
						|
 | 
						|
/**
 | 
						|
 * LRUCache.
 | 
						|
 *
 | 
						|
 * @constructor
 | 
						|
 * @param {function} Keys     - Array class for storing keys.
 | 
						|
 * @param {function} Values   - Array class for storing values.
 | 
						|
 * @param {number}   capacity - Desired capacity.
 | 
						|
 */
 | 
						|
function LRUCache(Keys, Values, capacity) {
 | 
						|
  if (arguments.length < 2) {
 | 
						|
    capacity = Keys;
 | 
						|
    Keys = null;
 | 
						|
    Values = null;
 | 
						|
  }
 | 
						|
 | 
						|
  this.capacity = capacity;
 | 
						|
 | 
						|
  if (typeof this.capacity !== 'number' || this.capacity <= 0)
 | 
						|
    throw new Error('mnemonist/lru-cache: capacity should be positive number.');
 | 
						|
  else if (!isFinite(this.capacity) || Math.floor(this.capacity) !== this.capacity)
 | 
						|
    throw new Error('mnemonist/lru-cache: capacity should be a finite positive integer.');
 | 
						|
 | 
						|
  var PointerArray = typed.getPointerArray(capacity);
 | 
						|
 | 
						|
  this.forward = new PointerArray(capacity);
 | 
						|
  this.backward = new PointerArray(capacity);
 | 
						|
  this.K = typeof Keys === 'function' ? new Keys(capacity) : new Array(capacity);
 | 
						|
  this.V = typeof Values === 'function' ? new Values(capacity) : new Array(capacity);
 | 
						|
 | 
						|
  // Properties
 | 
						|
  this.size = 0;
 | 
						|
  this.head = 0;
 | 
						|
  this.tail = 0;
 | 
						|
  this.items = {};
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to clear the structure.
 | 
						|
 *
 | 
						|
 * @return {undefined}
 | 
						|
 */
 | 
						|
LRUCache.prototype.clear = function() {
 | 
						|
  this.size = 0;
 | 
						|
  this.head = 0;
 | 
						|
  this.tail = 0;
 | 
						|
  this.items = {};
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to splay a value on top.
 | 
						|
 *
 | 
						|
 * @param  {number}   pointer - Pointer of the value to splay on top.
 | 
						|
 * @return {LRUCache}
 | 
						|
 */
 | 
						|
LRUCache.prototype.splayOnTop = function(pointer) {
 | 
						|
  var oldHead = this.head;
 | 
						|
 | 
						|
  if (this.head === pointer)
 | 
						|
    return this;
 | 
						|
 | 
						|
  var previous = this.backward[pointer],
 | 
						|
      next = this.forward[pointer];
 | 
						|
 | 
						|
  if (this.tail === pointer) {
 | 
						|
    this.tail = previous;
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    this.backward[next] = previous;
 | 
						|
  }
 | 
						|
 | 
						|
  this.forward[previous] = next;
 | 
						|
 | 
						|
  this.backward[oldHead] = pointer;
 | 
						|
  this.head = pointer;
 | 
						|
  this.forward[pointer] = oldHead;
 | 
						|
 | 
						|
  return this;
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to set the value for the given key in the cache.
 | 
						|
 *
 | 
						|
 * @param  {any} key   - Key.
 | 
						|
 * @param  {any} value - Value.
 | 
						|
 * @return {undefined}
 | 
						|
 */
 | 
						|
LRUCache.prototype.set = function(key, value) {
 | 
						|
 | 
						|
  var pointer = this.items[key];
 | 
						|
 | 
						|
  // The key already exists, we just need to update the value and splay on top
 | 
						|
  if (typeof pointer !== 'undefined') {
 | 
						|
    this.splayOnTop(pointer);
 | 
						|
    this.V[pointer] = value;
 | 
						|
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  // The cache is not yet full
 | 
						|
  if (this.size < this.capacity) {
 | 
						|
    pointer = this.size++;
 | 
						|
  }
 | 
						|
 | 
						|
  // Cache is full, we need to drop the last value
 | 
						|
  else {
 | 
						|
    pointer = this.tail;
 | 
						|
    this.tail = this.backward[pointer];
 | 
						|
    delete this.items[this.K[pointer]];
 | 
						|
  }
 | 
						|
 | 
						|
  // Storing key & value
 | 
						|
  this.items[key] = pointer;
 | 
						|
  this.K[pointer] = key;
 | 
						|
  this.V[pointer] = value;
 | 
						|
 | 
						|
  // Moving the item at the front of the list
 | 
						|
  this.forward[pointer] = this.head;
 | 
						|
  this.backward[this.head] = pointer;
 | 
						|
  this.head = pointer;
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to set the value for the given key in the cache
 | 
						|
 *
 | 
						|
 * @param  {any} key   - Key.
 | 
						|
 * @param  {any} value - Value.
 | 
						|
 * @return {{evicted: boolean, key: any, value: any}} An object containing the
 | 
						|
 * key and value of an item that was overwritten or evicted in the set
 | 
						|
 * operation, as well as a boolean indicating whether it was evicted due to
 | 
						|
 * limited capacity. Return value is null if nothing was evicted or overwritten
 | 
						|
 * during the set operation.
 | 
						|
 */
 | 
						|
LRUCache.prototype.setpop = function(key, value) {
 | 
						|
  var oldValue = null;
 | 
						|
  var oldKey = null;
 | 
						|
 | 
						|
  var pointer = this.items[key];
 | 
						|
 | 
						|
  // The key already exists, we just need to update the value and splay on top
 | 
						|
  if (typeof pointer !== 'undefined') {
 | 
						|
    this.splayOnTop(pointer);
 | 
						|
    oldValue = this.V[pointer];
 | 
						|
    this.V[pointer] = value;
 | 
						|
    return {evicted: false, key: key, value: oldValue};
 | 
						|
  }
 | 
						|
 | 
						|
  // The cache is not yet full
 | 
						|
  if (this.size < this.capacity) {
 | 
						|
    pointer = this.size++;
 | 
						|
  }
 | 
						|
 | 
						|
  // Cache is full, we need to drop the last value
 | 
						|
  else {
 | 
						|
    pointer = this.tail;
 | 
						|
    this.tail = this.backward[pointer];
 | 
						|
    oldValue = this.V[pointer];
 | 
						|
    oldKey = this.K[pointer];
 | 
						|
    delete this.items[this.K[pointer]];
 | 
						|
  }
 | 
						|
 | 
						|
  // Storing key & value
 | 
						|
  this.items[key] = pointer;
 | 
						|
  this.K[pointer] = key;
 | 
						|
  this.V[pointer] = value;
 | 
						|
 | 
						|
  // Moving the item at the front of the list
 | 
						|
  this.forward[pointer] = this.head;
 | 
						|
  this.backward[this.head] = pointer;
 | 
						|
  this.head = pointer;
 | 
						|
 | 
						|
  // Return object if eviction took place, otherwise return null
 | 
						|
  if (oldKey) {
 | 
						|
    return {evicted: true, key: oldKey, value: oldValue};
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    return null;
 | 
						|
  }
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to check whether the key exists in the cache.
 | 
						|
 *
 | 
						|
 * @param  {any} key   - Key.
 | 
						|
 * @return {boolean}
 | 
						|
 */
 | 
						|
LRUCache.prototype.has = function(key) {
 | 
						|
  return key in this.items;
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to get the value attached to the given key. Will move the
 | 
						|
 * related key to the front of the underlying linked list.
 | 
						|
 *
 | 
						|
 * @param  {any} key   - Key.
 | 
						|
 * @return {any}
 | 
						|
 */
 | 
						|
LRUCache.prototype.get = function(key) {
 | 
						|
  var pointer = this.items[key];
 | 
						|
 | 
						|
  if (typeof pointer === 'undefined')
 | 
						|
    return;
 | 
						|
 | 
						|
  this.splayOnTop(pointer);
 | 
						|
 | 
						|
  return this.V[pointer];
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to get the value attached to the given key. Does not modify
 | 
						|
 * the ordering of the underlying linked list.
 | 
						|
 *
 | 
						|
 * @param  {any} key   - Key.
 | 
						|
 * @return {any}
 | 
						|
 */
 | 
						|
LRUCache.prototype.peek = function(key) {
 | 
						|
  var pointer = this.items[key];
 | 
						|
 | 
						|
  if (typeof pointer === 'undefined')
 | 
						|
    return;
 | 
						|
 | 
						|
  return this.V[pointer];
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to iterate over the cache's entries using a callback.
 | 
						|
 *
 | 
						|
 * @param  {function}  callback - Function to call for each item.
 | 
						|
 * @param  {object}    scope    - Optional scope.
 | 
						|
 * @return {undefined}
 | 
						|
 */
 | 
						|
LRUCache.prototype.forEach = function(callback, scope) {
 | 
						|
  scope = arguments.length > 1 ? scope : this;
 | 
						|
 | 
						|
  var i = 0,
 | 
						|
      l = this.size;
 | 
						|
 | 
						|
  var pointer = this.head,
 | 
						|
      keys = this.K,
 | 
						|
      values = this.V,
 | 
						|
      forward = this.forward;
 | 
						|
 | 
						|
  while (i < l) {
 | 
						|
 | 
						|
    callback.call(scope, values[pointer], keys[pointer], this);
 | 
						|
    pointer = forward[pointer];
 | 
						|
 | 
						|
    i++;
 | 
						|
  }
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to create an iterator over the cache's keys from most
 | 
						|
 * recently used to least recently used.
 | 
						|
 *
 | 
						|
 * @return {Iterator}
 | 
						|
 */
 | 
						|
LRUCache.prototype.keys = function() {
 | 
						|
  var i = 0,
 | 
						|
      l = this.size;
 | 
						|
 | 
						|
  var pointer = this.head,
 | 
						|
      keys = this.K,
 | 
						|
      forward = this.forward;
 | 
						|
 | 
						|
  return new Iterator(function() {
 | 
						|
    if (i >= l)
 | 
						|
      return {done: true};
 | 
						|
 | 
						|
    var key = keys[pointer];
 | 
						|
 | 
						|
    i++;
 | 
						|
 | 
						|
    if (i < l)
 | 
						|
      pointer = forward[pointer];
 | 
						|
 | 
						|
    return {
 | 
						|
      done: false,
 | 
						|
      value: key
 | 
						|
    };
 | 
						|
  });
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to create an iterator over the cache's values from most
 | 
						|
 * recently used to least recently used.
 | 
						|
 *
 | 
						|
 * @return {Iterator}
 | 
						|
 */
 | 
						|
LRUCache.prototype.values = function() {
 | 
						|
  var i = 0,
 | 
						|
      l = this.size;
 | 
						|
 | 
						|
  var pointer = this.head,
 | 
						|
      values = this.V,
 | 
						|
      forward = this.forward;
 | 
						|
 | 
						|
  return new Iterator(function() {
 | 
						|
    if (i >= l)
 | 
						|
      return {done: true};
 | 
						|
 | 
						|
    var value = values[pointer];
 | 
						|
 | 
						|
    i++;
 | 
						|
 | 
						|
    if (i < l)
 | 
						|
      pointer = forward[pointer];
 | 
						|
 | 
						|
    return {
 | 
						|
      done: false,
 | 
						|
      value: value
 | 
						|
    };
 | 
						|
  });
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Method used to create an iterator over the cache's entries from most
 | 
						|
 * recently used to least recently used.
 | 
						|
 *
 | 
						|
 * @return {Iterator}
 | 
						|
 */
 | 
						|
LRUCache.prototype.entries = function() {
 | 
						|
  var i = 0,
 | 
						|
      l = this.size;
 | 
						|
 | 
						|
  var pointer = this.head,
 | 
						|
      keys = this.K,
 | 
						|
      values = this.V,
 | 
						|
      forward = this.forward;
 | 
						|
 | 
						|
  return new Iterator(function() {
 | 
						|
    if (i >= l)
 | 
						|
      return {done: true};
 | 
						|
 | 
						|
    var key = keys[pointer],
 | 
						|
        value = values[pointer];
 | 
						|
 | 
						|
    i++;
 | 
						|
 | 
						|
    if (i < l)
 | 
						|
      pointer = forward[pointer];
 | 
						|
 | 
						|
    return {
 | 
						|
      done: false,
 | 
						|
      value: [key, value]
 | 
						|
    };
 | 
						|
  });
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Attaching the #.entries method to Symbol.iterator if possible.
 | 
						|
 */
 | 
						|
if (typeof Symbol !== 'undefined')
 | 
						|
  LRUCache.prototype[Symbol.iterator] = LRUCache.prototype.entries;
 | 
						|
 | 
						|
/**
 | 
						|
 * Convenience known methods.
 | 
						|
 */
 | 
						|
LRUCache.prototype.inspect = function() {
 | 
						|
  var proxy = new Map();
 | 
						|
 | 
						|
  var iterator = this.entries(),
 | 
						|
      step;
 | 
						|
 | 
						|
  while ((step = iterator.next(), !step.done))
 | 
						|
    proxy.set(step.value[0], step.value[1]);
 | 
						|
 | 
						|
  // Trick so that node displays the name of the constructor
 | 
						|
  Object.defineProperty(proxy, 'constructor', {
 | 
						|
    value: LRUCache,
 | 
						|
    enumerable: false
 | 
						|
  });
 | 
						|
 | 
						|
  return proxy;
 | 
						|
};
 | 
						|
 | 
						|
if (typeof Symbol !== 'undefined')
 | 
						|
  LRUCache.prototype[Symbol.for('nodejs.util.inspect.custom')] = LRUCache.prototype.inspect;
 | 
						|
 | 
						|
/**
 | 
						|
 * Static @.from function taking an arbitrary iterable & converting it into
 | 
						|
 * a structure.
 | 
						|
 *
 | 
						|
 * @param  {Iterable} iterable - Target iterable.
 | 
						|
 * @param  {function} Keys     - Array class for storing keys.
 | 
						|
 * @param  {function} Values   - Array class for storing values.
 | 
						|
 * @param  {number}   capacity - Cache's capacity.
 | 
						|
 * @return {LRUCache}
 | 
						|
 */
 | 
						|
LRUCache.from = function(iterable, Keys, Values, capacity) {
 | 
						|
  if (arguments.length < 2) {
 | 
						|
    capacity = iterables.guessLength(iterable);
 | 
						|
 | 
						|
    if (typeof capacity !== 'number')
 | 
						|
      throw new Error('mnemonist/lru-cache.from: could not guess iterable length. Please provide desired capacity as last argument.');
 | 
						|
  }
 | 
						|
  else if (arguments.length === 2) {
 | 
						|
    capacity = Keys;
 | 
						|
    Keys = null;
 | 
						|
    Values = null;
 | 
						|
  }
 | 
						|
 | 
						|
  var cache = new LRUCache(Keys, Values, capacity);
 | 
						|
 | 
						|
  forEach(iterable, function(value, key) {
 | 
						|
    cache.set(key, value);
 | 
						|
  });
 | 
						|
 | 
						|
  return cache;
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * Exporting.
 | 
						|
 */
 | 
						|
module.exports = LRUCache;
 |