Use a thread local script engine

This commit is contained in:
Shadowfacts 2017-08-30 19:35:26 -04:00
parent dc40cc2647
commit 7df6669bd0
Signed by: shadowfacts
GPG Key ID: 94A5AB95422746E5
2 changed files with 15 additions and 9 deletions

View File

@ -2,8 +2,10 @@ package net.shadowfacts.ekt
import java.io.File import java.io.File
import javax.script.ScriptContext import javax.script.ScriptContext
import javax.script.ScriptEngine
import javax.script.ScriptEngineManager import javax.script.ScriptEngineManager
import javax.script.SimpleScriptContext import javax.script.SimpleScriptContext
import kotlin.concurrent.getOrSet
/** /**
* @author shadowfacts * @author shadowfacts
@ -42,9 +44,7 @@ fun include(include: String, init: net.shadowfacts.ekt.EKT.DataProvider.() -> Un
_result.toString() _result.toString()
""" """
private val engine by lazy { private val engine = ThreadLocal<ScriptEngine>()
ScriptEngineManager().getEngineByExtension("kts")
}
fun render(env: TemplateEnvironment, template: String = env.template): String { fun render(env: TemplateEnvironment, template: String = env.template): String {
if (env.cacheDir != null && env.cacheFile.exists()) { if (env.cacheDir != null && env.cacheFile.exists()) {
@ -122,6 +122,7 @@ _result.toString()
} }
internal fun eval(script: String, env: TemplateEnvironment): String { internal fun eval(script: String, env: TemplateEnvironment): String {
val engine = engine.getOrSet { ScriptEngineManager().getEngineByExtension("kts") }
engine.context = SimpleScriptContext() engine.context = SimpleScriptContext()
val bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE) val bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE)
bindings.putAll(env.data) bindings.putAll(env.data)

View File

@ -1,17 +1,22 @@
package net.shadowfacts.ekt package net.shadowfacts.ekt
import java.io.File import java.io.File
import kotlin.concurrent.thread
/** /**
* @author shadowfacts * @author shadowfacts
*/ */
fun main(args: Array<String>) { fun main(args: Array<String>) {
val res = EKT.renderClasspath("template", "/templates", cacheDir = File("cache")) { for (i in 0..2) {
thread {
println("Calling from: $i")
render()
}
}
}
fun render(): String {
return EKT.renderClasspath("template", "/templates") {
"list" to (listOf(1, 2, 3) asType "List<Int>") "list" to (listOf(1, 2, 3) asType "List<Int>")
} }
File("result.txt").apply {
if (!exists()) createNewFile()
writeText(res)
}
} }