|
|
@@ -7,7 +7,15 @@ import java.lang.System.getenv
|
|
|
import co.there4.hexagon.settings.SettingsManager.setting
|
|
|
import co.there4.hexagon.repository.mongoDatabase
|
|
|
import co.there4.hexagon.util.err
|
|
|
+import com.zaxxer.hikari.HikariConfig
|
|
|
+import com.zaxxer.hikari.HikariDataSource
|
|
|
import java.io.Closeable
|
|
|
+import java.sql.Connection
|
|
|
+import java.sql.ResultSet.CONCUR_READ_ONLY
|
|
|
+import java.sql.ResultSet.TYPE_FORWARD_ONLY
|
|
|
+import java.util.concurrent.ExecutorService
|
|
|
+import java.util.concurrent.Executors
|
|
|
+import javax.sql.DataSource
|
|
|
import kotlin.reflect.KProperty1
|
|
|
|
|
|
internal val DB_ROWS = 10000
|
|
|
@@ -17,8 +25,11 @@ private val DB = setting<String>("database") ?: "hello_world"
|
|
|
private val WORLD: String = setting<String>("worldCollection") ?: "world"
|
|
|
private val FORTUNE: String = setting<String>("fortuneCollection") ?: "fortune"
|
|
|
|
|
|
+private val executor: ExecutorService = Executors.newFixedThreadPool(8)
|
|
|
+
|
|
|
internal fun createStore(engine: String): Repository = when (engine) {
|
|
|
"mongodb" -> MongoDbRepository()
|
|
|
+ "mysql" -> MySqlRepository()
|
|
|
else -> error("Unsupported database")
|
|
|
}
|
|
|
|
|
|
@@ -45,8 +56,107 @@ internal class MongoDbRepository : Repository {
|
|
|
|
|
|
override fun replaceWorlds(queries: Int) = (1..queries).map {
|
|
|
val id = rnd()
|
|
|
- val newWorld = World(id, id)
|
|
|
- worldRepository.replaceObject(newWorld)
|
|
|
+ val newWorld = worldRepository.find(id)?.copy(randomNumber = rnd()) ?: err
|
|
|
+ executor.execute { worldRepository.replaceObject(newWorld) }
|
|
|
newWorld
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+internal class MySqlRepository : Repository {
|
|
|
+ private val SELECT_WORLD = "select * from world where id = ?"
|
|
|
+ private val UPDATE_WORLD = "update world set randomNumber = ? where id = ?"
|
|
|
+ private val SELECT_FORTUNES = "select * from fortune"
|
|
|
+
|
|
|
+ private val DATA_SOURCE: DataSource
|
|
|
+
|
|
|
+ init {
|
|
|
+ val config = HikariConfig()
|
|
|
+ config.jdbcUrl = "jdbc:mysql://$DB_HOST/$DB?" +
|
|
|
+ "useSSL=false&" +
|
|
|
+ "rewriteBatchedStatements=true&" +
|
|
|
+ "jdbcCompliantTruncation=false&" +
|
|
|
+ "elideSetAutoCommits=true&" +
|
|
|
+ "useLocalSessionState=true&" +
|
|
|
+ "cachePrepStmts=true&" +
|
|
|
+ "cacheCallableStmts=true&" +
|
|
|
+ "alwaysSendSetIsolation=false&" +
|
|
|
+ "prepStmtCacheSize=4096&" +
|
|
|
+ "cacheServerConfiguration=true&" +
|
|
|
+ "prepStmtCacheSqlLimit=2048&" +
|
|
|
+ "traceProtocol=false&" +
|
|
|
+ "useUnbufferedInput=false&" +
|
|
|
+ "useReadAheadInput=false&" +
|
|
|
+ "maintainTimeStats=false&" +
|
|
|
+ "useServerPrepStmts=true&" +
|
|
|
+ "cacheRSMetadata=true"
|
|
|
+ config.maximumPoolSize = 256
|
|
|
+ config.username = "benchmarkdbuser"
|
|
|
+ config.password = "benchmarkdbpass"
|
|
|
+ DATA_SOURCE = HikariDataSource(config)
|
|
|
+ }
|
|
|
+
|
|
|
+ override fun findFortunes(): List<Fortune> {
|
|
|
+ var fortunes = listOf<Fortune>()
|
|
|
+
|
|
|
+ val connection = KConnection(DATA_SOURCE.connection ?: err)
|
|
|
+ connection.use { con: Connection ->
|
|
|
+ val rs = con.prepareStatement(SELECT_FORTUNES).executeQuery()
|
|
|
+ while (rs.next())
|
|
|
+ fortunes += Fortune(rs.getInt(1), rs.getString(2))
|
|
|
+ }
|
|
|
+
|
|
|
+ return fortunes
|
|
|
+ }
|
|
|
+
|
|
|
+ class KConnection(conn: Connection) : Connection by conn, Closeable
|
|
|
+
|
|
|
+ override fun findWorlds(queries: Int): List<World> {
|
|
|
+ var worlds: List<World> = listOf()
|
|
|
+
|
|
|
+ KConnection(DATA_SOURCE.connection).use { con: Connection ->
|
|
|
+ val stmtSelect = con.prepareStatement(SELECT_WORLD)
|
|
|
+
|
|
|
+ for (ii in 0..queries - 1) {
|
|
|
+ stmtSelect.setInt(1, rnd())
|
|
|
+ val rs = stmtSelect.executeQuery()
|
|
|
+ rs.next()
|
|
|
+ worlds += World(rs.getInt(1), rs.getInt(2))
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return worlds
|
|
|
+ }
|
|
|
+
|
|
|
+ override fun replaceWorlds(queries: Int): List<World> {
|
|
|
+ var worlds: List<World> = listOf()
|
|
|
+
|
|
|
+ KConnection(DATA_SOURCE.connection).use { con: Connection ->
|
|
|
+ val stmtSelect = con.prepareStatement(SELECT_WORLD, TYPE_FORWARD_ONLY, CONCUR_READ_ONLY)
|
|
|
+
|
|
|
+ for (ii in 0..queries - 1) {
|
|
|
+ stmtSelect.setInt(1, rnd())
|
|
|
+ val rs = stmtSelect.executeQuery()
|
|
|
+ rs.next()
|
|
|
+
|
|
|
+ val world = World(rs.getInt(1), rs.getInt(2)).copy(randomNumber = rnd())
|
|
|
+ worlds += world
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ executor.execute {
|
|
|
+ KConnection(DATA_SOURCE.connection).use { con: Connection ->
|
|
|
+ val stmtUpdate = con.prepareStatement(UPDATE_WORLD)
|
|
|
+
|
|
|
+ for ((_, id, randomNumber) in worlds) {
|
|
|
+ stmtUpdate.setInt(1, randomNumber)
|
|
|
+ stmtUpdate.setInt(2, id)
|
|
|
+ stmtUpdate.addBatch()
|
|
|
+ }
|
|
|
+
|
|
|
+ stmtUpdate.executeBatch()
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return worlds
|
|
|
+ }
|
|
|
+}
|
Juanjo Aguililla