Change AbstractBenchmark code to avoid collisions in files map
This commit is contained in:
committed by
Mikhail Glukhikh
parent
c9d5202971
commit
7cbd838720
+3
-8
@@ -19,7 +19,7 @@ import java.util.concurrent.TimeUnit
|
|||||||
@Fork(1)
|
@Fork(1)
|
||||||
@State(Scope.Benchmark)
|
@State(Scope.Benchmark)
|
||||||
abstract class AbstractBenchmark {
|
abstract class AbstractBenchmark {
|
||||||
private val files = mutableMapOf<String, Pair<File, String>>()
|
private val files = mutableMapOf<File, String>()
|
||||||
abstract val generator: TreeGenerator
|
abstract val generator: TreeGenerator
|
||||||
|
|
||||||
protected fun readFiles(ignoreTestData: Boolean, path: String) {
|
protected fun readFiles(ignoreTestData: Boolean, path: String) {
|
||||||
@@ -32,17 +32,12 @@ abstract class AbstractBenchmark {
|
|||||||
if (file.extension != "kt") continue
|
if (file.extension != "kt") continue
|
||||||
|
|
||||||
val text = FileUtil.loadFile(file, CharsetToolkit.UTF8, true).trim()
|
val text = FileUtil.loadFile(file, CharsetToolkit.UTF8, true).trim()
|
||||||
files[file.name] = Pair(file, text)
|
files[file] = text
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//TODO null check
|
|
||||||
protected fun getFileByName(fileName: String): Pair<File, String> {
|
|
||||||
return files[fileName]!!
|
|
||||||
}
|
|
||||||
|
|
||||||
protected fun forEachFile(f: (String, File) -> Unit) {
|
protected fun forEachFile(f: (String, File) -> Unit) {
|
||||||
for ((file, text) in files.values) {
|
for ((file, text) in files) {
|
||||||
f(text, file)
|
f(text, file)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user