ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-02-12 05:35:50 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

client: Fix wrong pickN generator

Browse Source
This commit is contained in:
Andras Slemmer 2017-02-24 17:36:14 +00:00
parent f0d82e4918
commit 1532318ae7
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
client/src/main/kotlin/net/corda/client/mock/Generator.kt
View File

@ -176,14 +176,15 @@ fun <A : Any> Generator.Companion.replicatePoisson(meanSize: Double, generator:
fun <A : Any> Generator.Companion.pickOne(list: List<A>) = Generator.intRange(0, list.size - 1).map { list[it] } fun <A : Any> Generator.Companion.pickOne(list: List<A>) = Generator.intRange(0, list.size - 1).map { list[it] }
fun <A : Any> Generator.Companion.pickN(number: Int, list: List<A>) = Generator<List<A>> { fun <A : Any> Generator.Companion.pickN(number: Int, list: List<A>) = Generator<List<A>> {
val mask = BitSet(list.size) val mask = BitSet(list.size)
for (i in 0..Math.min(list.size, number) - 1) { val size = Math.min(list.size, number)
mask[i] = 1 for (i in 0..size - 1) {
mask[i] = true
} }
for (i in 0..mask.size() - 1) { for (i in 0..size - 1) {
val byte = mask[i] val bit = mask[i]
val swapIndex = i + it.nextInt(mask.size() - i) val swapIndex = i + it.nextInt(size - i)
mask[i] = mask[swapIndex] mask[i] = mask[swapIndex]
mask[swapIndex] = byte mask[swapIndex] = bit
} }
val resultList = ArrayList<A>() val resultList = ArrayList<A>()
list.forEachIndexed { index, a -> list.forEachIndexed { index, a ->