首页 > 解决方案 > JavaScript:对象键的快速随机索引

问题描述

我正在尝试从用作快速索引的字典的对象中随机选择一个随机属性。

考虑以下代码:

//Create test dictionary
var dict = {};
for(var i = 0; i < 10000; i++) {
  var num = Math.random() * 1000000 << 0;
  dict[num] = { Id: num };
}

//Fuzz
const NUM_RUNS = 1000;
var start = new Date();
for(var i = 0; i < NUM_RUNS; i++)
  getRandom(dict);
var end = new Date();

var runTime = (end.getTime() - start.getTime()) / 1000;
var timePerCall = (runTime / NUM_RUNS) * Math.pow(10,9);
console.log('Total Calls: ' + NUM_RUNS);
console.log('Total Runtime: ' + runTime + ' seconds');
console.log('Time Per Call: ' + timePerCall + ' nanoseconds');



function getRandom(dict) {
  var keys = Object.keys(dict);
  var index = keys[Math.random() * keys.length << 0];
  return dict[index];
}

如您所见,使用Object.keys()非常昂贵,尤其是随着字典的增长

我正在寻找一种最佳方式来实现字典中元素的随机选择。当然,字典中有 10000 多个元素是一个极端的情况,但我仍然希望能够处理它。

标签: javascriptoptimization

解决方案


如果缓存Object.keys数组不可行,您可以使用以下命令维护自己的副本Proxy

function generateKeyTracker (keysPropertyName) {
  const set = new Set();
  function defineProperty (target, property, descriptor) {
    target[property] = descriptor.value;

    if (set.has(property)) return true;

    set.add(property);
    target[keysPropertyName].push(property);
    return true;
  }
  function deleteProperty (target, property) {
    delete target[property];
    
    if (!set.delete(property)) return true;

    target[keysPropertyName] = target[keysPropertyName].filter(key => key !== property);
    return true;
  }
  return {defineProperty, deleteProperty};
}

//Create test dictionary
var dict = new Proxy(
  Object.defineProperty({}, '__keys', {
    configurable: true,
    enumerable: false,
    writable: true,
    value: []
  }),
  generateKeyTracker('__keys')
);

for(var i = 0; i < 1e4; i++) {
  var num = Math.random() * 1e6 << 0;
  dict[num] = { Id: num };
}

//Fuzz
const NUM_RUNS = 1e6;
var start = performance.now();
for(var i = 0; i < NUM_RUNS; i++)
  getRandom(dict);
var end = performance.now();

var runTime = end - start;
var timePerCall = (runTime / NUM_RUNS);
console.log(`Total Calls: ${NUM_RUNS}`);
console.log(`Total Runtime: ${runTime} ms`);
console.log(`Time Per Call: ${timePerCall * 1e6} ns`);

function getRandom(dict) {
  var index = Math.random() * dict.__keys.length << 0;
  return dict[dict.__keys[index]];
}

这使用属性创建删除的陷阱来跟踪dict存储在不可枚举属性中的属性键dict.__keys


推荐阅读