blob: 89b26dc81dd71b8ec9936caefd60e000fe5c0508 (
plain)
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
|
/*
This file is part of GNU Taler
(C) 2022 Taler Systems S.A.
GNU Taler is free software; you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software
Foundation; either version 3, or (at your option) any later version.
GNU Taler is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with
GNU Taler; see the file COPYING. If not, see <http://www.gnu.org/licenses/>
@author: Boss Marco
*/
const getRandomInt = function (max: number) {
return Math.floor(Math.random() * max);
};
abstract class BenchMerchantIDGenerator {
abstract getRandomMerchantID(): number;
}
class ZipfGenerator extends BenchMerchantIDGenerator {
weights: number[];
total_weight: number;
constructor(numMerchants: number) {
super();
this.weights = new Array<number>(numMerchants);
for (var i = 0; i < this.weights.length; i++) {
/* we use integers (floor), make sure we have big enough values
* by multiplying with
* numMerchants again */
this.weights[i] = Math.floor((numMerchants / (i + 1)) * numMerchants);
}
this.total_weight = this.weights.reduce((p, n) => p + n);
}
getRandomMerchantID(): number {
let random = getRandomInt(this.total_weight);
let current = 0;
for (var i = 0; i < this.weights.length; i++) {
current += this.weights[i];
if (random <= current) {
return i + 1;
}
}
/* should never come here */
return getRandomInt(this.weights.length);
}
}
class RandomGenerator extends BenchMerchantIDGenerator {
max: number;
constructor(numMerchants: number) {
super();
this.max = numMerchants;
}
getRandomMerchantID() {
return getRandomInt(this.max);
}
}
export default function (
type: string,
maxID: number,
): BenchMerchantIDGenerator {
switch (type) {
case "zipf":
return new ZipfGenerator(maxID);
case "rand":
return new RandomGenerator(maxID);
default:
throw new Error("Valid types are 'zipf' and 'rand'");
}
}
|
