-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
94 lines (81 loc) · 1.77 KB
/
index.js
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
var toDictionary = function(items, keySelector, itemSelector, skipMissingKeys) {
if (!keySelector) throw "Missin keySelector";
var res = {};
if (!items || !items.length) {
return res;
}
for (var i = 0; i < items.length; i++) {
var key = keySelector(items[i]);
if (key === undefined) {
if (skipMissingKeys) {
continue;
}
throw "Missing key at" + items[i];
}
var value = itemSelector ? itemSelector(items[i]) : items[i];
res[key] = value;
}
return res;
};
var toLookup = function(items, keySelector, itemSelector, skipMissingKeys) {
if (!keySelector) throw "Missin keySelector";
var res = {};
if (!items || !items.length) {
return res;
}
for (var i = 0; i < items.length; i++) {
var key = keySelector(items[i]);
if (key === undefined) {
if (skipMissingKeys) {
continue;
}
throw "Missing key at" + items[i];
}
var value = itemSelector ? itemSelector(items[i]) : items[i];
if (!res[key]) {
res[key] = [];
}
res[key].push(value);
}
return res;
};
var distinctBy = function(items, keySelector) {
if (!keySelector) {
throw "Missin keySelector";
}
var added = {};
var result = [];
for (var i = 0; i < items.length; i++) {
var key = keySelector(items[i]);
if (!added[key]) {
added[key] = true;
result.push(items[i]);
}
}
return result;
};
function groupBy(items, selector) {
var res = [];
var keyMap = {};
for (var i = 0; i < items.length; i++) {
var item = items[i];
var key = selector(item);
if (!keyMap[key]) {
var group = {
key: key,
items: [item]
};
keyMap[key] = group;
res.push(group);
} else {
keyMap[key].items.push(item);
}
}
return res;
}
module.exports = exports = {
toDictionary: toDictionary,
toLookup: toLookup,
distinctBy: distinctBy,
groupBy: groupBy
};