Develop (#88)
Merge develop into master * Update Translation to 1.3.0 (#86) * Update Translation to 1.3.0 * Tweak SQL_ERROR word order Co-authored-by: NPBeta <shanhang007@gmail.com> * Update version to 1.3.1 * Remove empty body of PausedTaskEntry Co-authored-by: NPBeta <shanhang007@gmail.com>pull/105/head v1.3.1
GitHub
parent
4d4107aaf3
commit
82662bc4d3
@ -1,170 +1,189 @@
|
||||
package net.trivernis.chunkmaster.lib.database
|
||||
|
||||
import net.trivernis.chunkmaster.Chunkmaster
|
||||
import org.apache.commons.lang.exception.ExceptionUtils
|
||||
import java.lang.Exception
|
||||
import java.sql.Connection
|
||||
import java.sql.DriverManager
|
||||
import java.sql.ResultSet
|
||||
|
||||
class SqliteManager(private val chunkmaster: Chunkmaster) {
|
||||
private val tables = listOf(
|
||||
Pair(
|
||||
"generation_tasks",
|
||||
listOf(
|
||||
Pair("id", "integer PRIMARY KEY AUTOINCREMENT"),
|
||||
Pair("center_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("center_z", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("last_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("last_z", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("world", "text UNIQUE NOT NULL DEFAULT 'world'"),
|
||||
Pair("radius", "integer DEFAULT -1"),
|
||||
Pair("shape", "text NOT NULL DEFAULT 'square'"),
|
||||
Pair("state", "text NOT NULL DEFAULT 'GENERATING'")
|
||||
)
|
||||
),
|
||||
Pair(
|
||||
"world_properties",
|
||||
listOf(
|
||||
Pair("name", "text PRIMARY KEY"),
|
||||
Pair("center_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("center_z", "integer NOT NULL DEFAULT 0")
|
||||
)
|
||||
),
|
||||
Pair(
|
||||
"pending_chunks",
|
||||
listOf(
|
||||
Pair("id", "integer PRIMARY KEY AUTOINCREMENT"),
|
||||
Pair("task_id", "integer NOT NULL"),
|
||||
Pair("chunk_x", "integer NOT NULL"),
|
||||
Pair("chunk_z", "integer NOT NULL")
|
||||
)
|
||||
)
|
||||
)
|
||||
private val needUpdate = HashSet<Pair<String, Pair<String, String>>>()
|
||||
private val needCreation = HashSet<String>()
|
||||
private var connection: Connection? = null
|
||||
private var activeTasks = 0
|
||||
|
||||
val worldProperties = WorldProperties(this)
|
||||
val pendingChunks = PendingChunks(this)
|
||||
val generationTasks = GenerationTasks(this)
|
||||
|
||||
/**
|
||||
* Returns the connection to the database
|
||||
*/
|
||||
fun getConnection(): Connection? {
|
||||
if (this.connection != null) {
|
||||
return this.connection
|
||||
}
|
||||
try {
|
||||
Class.forName("org.sqlite.JDBC")
|
||||
this.connection = DriverManager.getConnection("jdbc:sqlite:${chunkmaster.dataFolder.absolutePath}/" +
|
||||
"${chunkmaster.config.getString("database.filename")}")
|
||||
return this.connection
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("DATABASE_CONNECTION_ERROR"))
|
||||
chunkmaster.logger.severe(e.message)
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks for and performs an update
|
||||
*/
|
||||
fun init() {
|
||||
this.checkUpdate()
|
||||
this.performUpdate()
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks which tables need an update or creation.
|
||||
*/
|
||||
private fun checkUpdate() {
|
||||
val meta = getConnection()!!.metaData
|
||||
|
||||
for (table in tables) {
|
||||
val resTables = meta.getTables(null, null, table.first, null)
|
||||
|
||||
if (resTables.next()) { // table exists
|
||||
for (column in table.second) {
|
||||
val resColumn = meta.getColumns(null, null, table.first, column.first)
|
||||
if (!resColumn.next()) {
|
||||
needUpdate.add(Pair(table.first, column))
|
||||
}
|
||||
resColumn.close()
|
||||
}
|
||||
} else {
|
||||
needCreation.add(table.first)
|
||||
}
|
||||
resTables.close()
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Executes a sql statement on the database.
|
||||
*/
|
||||
fun executeStatement(sql: String, values: HashMap<Int, Any>, callback: ((ResultSet?) -> Unit)?) {
|
||||
val connection = getConnection()
|
||||
activeTasks++
|
||||
if (connection != null) {
|
||||
try {
|
||||
//println("'$sql' with values $values")
|
||||
val statement = connection.prepareStatement(sql)
|
||||
for (parameterValue in values) {
|
||||
statement.setObject(parameterValue.key, parameterValue.value)
|
||||
}
|
||||
statement.execute()
|
||||
val res: ResultSet? = statement.resultSet
|
||||
if (callback != null) {
|
||||
callback(res)
|
||||
}
|
||||
statement.close()
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("SQL_ERROR", e.toString()))
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
} finally {
|
||||
activeTasks--
|
||||
if (activeTasks == 0) {
|
||||
connection.close()
|
||||
this.connection = null
|
||||
}
|
||||
}
|
||||
} else {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("NO_DATABASE_CONNECTION"))
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates or updates tables that needed an update.
|
||||
*/
|
||||
private fun performUpdate() {
|
||||
for (table in needCreation) {
|
||||
try {
|
||||
var tableDef = "CREATE TABLE IF NOT EXISTS $table ("
|
||||
|
||||
for (column in tables.find{it.first == table}!!.second) {
|
||||
tableDef += "${column.first} ${column.second},"
|
||||
}
|
||||
tableDef = tableDef.substringBeforeLast(",") + ");"
|
||||
chunkmaster.logger.finest(chunkmaster.langManager.getLocalized("CREATE_TABLE_DEFINITION", table, tableDef))
|
||||
executeStatement(tableDef, HashMap(), null)
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("TABLE_CREATE_ERROR", table))
|
||||
chunkmaster.logger.severe(e.message)
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
}
|
||||
}
|
||||
for (table in needUpdate) {
|
||||
val updateSql = "ALTER TABLE ${table.first} ADD COLUMN ${table.second.first} ${table.second.second}"
|
||||
try {
|
||||
executeStatement(updateSql, HashMap(), null)
|
||||
chunkmaster.logger.finest(chunkmaster.langManager.getLocalized("UPDATE_TABLE_DEFINITION", table.first, updateSql))
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("UPDATE_TABLE_FAILED", table.first, updateSql))
|
||||
chunkmaster.logger.severe(e.message)
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
}
|
||||
}
|
||||
}
|
||||
package net.trivernis.chunkmaster.lib.database
|
||||
|
||||
import net.trivernis.chunkmaster.Chunkmaster
|
||||
import org.apache.commons.lang.exception.ExceptionUtils
|
||||
import java.sql.Connection
|
||||
import java.sql.DriverManager
|
||||
import java.sql.ResultSet
|
||||
|
||||
class SqliteManager(private val chunkmaster: Chunkmaster) {
|
||||
private val tables = listOf(
|
||||
Pair(
|
||||
"generation_tasks",
|
||||
listOf(
|
||||
Pair("id", "integer PRIMARY KEY AUTOINCREMENT"),
|
||||
Pair("center_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("center_z", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("last_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("last_z", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("world", "text UNIQUE NOT NULL DEFAULT 'world'"),
|
||||
Pair("radius", "integer DEFAULT -1"),
|
||||
Pair("shape", "text NOT NULL DEFAULT 'square'"),
|
||||
Pair("state", "text NOT NULL DEFAULT 'GENERATING'")
|
||||
)
|
||||
),
|
||||
Pair(
|
||||
"world_properties",
|
||||
listOf(
|
||||
Pair("name", "text PRIMARY KEY"),
|
||||
Pair("center_x", "integer NOT NULL DEFAULT 0"),
|
||||
Pair("center_z", "integer NOT NULL DEFAULT 0")
|
||||
)
|
||||
),
|
||||
Pair(
|
||||
"pending_chunks",
|
||||
listOf(
|
||||
Pair("id", "integer PRIMARY KEY AUTOINCREMENT"),
|
||||
Pair("task_id", "integer NOT NULL"),
|
||||
Pair("chunk_x", "integer NOT NULL"),
|
||||
Pair("chunk_z", "integer NOT NULL")
|
||||
)
|
||||
)
|
||||
)
|
||||
private val needUpdate = HashSet<Pair<String, Pair<String, String>>>()
|
||||
private val needCreation = HashSet<String>()
|
||||
private var connection: Connection? = null
|
||||
private var activeTasks = 0
|
||||
|
||||
val worldProperties = WorldProperties(this)
|
||||
val pendingChunks = PendingChunks(this)
|
||||
val generationTasks = GenerationTasks(this)
|
||||
|
||||
/**
|
||||
* Returns the connection to the database
|
||||
*/
|
||||
fun getConnection(): Connection? {
|
||||
if (this.connection != null) {
|
||||
return this.connection
|
||||
}
|
||||
try {
|
||||
Class.forName("org.sqlite.JDBC")
|
||||
this.connection = DriverManager.getConnection(
|
||||
"jdbc:sqlite:${chunkmaster.dataFolder.absolutePath}/" +
|
||||
"${chunkmaster.config.getString("database.filename")}"
|
||||
)
|
||||
return this.connection
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("DATABASE_CONNECTION_ERROR"))
|
||||
chunkmaster.logger.severe(e.message)
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks for and performs an update
|
||||
*/
|
||||
fun init() {
|
||||
this.checkUpdate()
|
||||
this.performUpdate()
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks which tables need an update or creation.
|
||||
*/
|
||||
private fun checkUpdate() {
|
||||
val meta = getConnection()!!.metaData
|
||||
|
||||
for (table in tables) {
|
||||
val resTables = meta.getTables(null, null, table.first, null)
|
||||
|
||||
if (resTables.next()) { // table exists
|
||||
for (column in table.second) {
|
||||
val resColumn = meta.getColumns(null, null, table.first, column.first)
|
||||
if (!resColumn.next()) {
|
||||
needUpdate.add(Pair(table.first, column))
|
||||
}
|
||||
resColumn.close()
|
||||
}
|
||||
} else {
|
||||
needCreation.add(table.first)
|
||||
}
|
||||
resTables.close()
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Executes a sql statement on the database.
|
||||
*/
|
||||
fun executeStatement(sql: String, values: HashMap<Int, Any>, callback: ((ResultSet?) -> Unit)?) {
|
||||
val connection = getConnection()
|
||||
activeTasks++
|
||||
if (connection != null) {
|
||||
try {
|
||||
//println("'$sql' with values $values")
|
||||
val statement = connection.prepareStatement(sql)
|
||||
for (parameterValue in values) {
|
||||
statement.setObject(parameterValue.key, parameterValue.value)
|
||||
}
|
||||
statement.execute()
|
||||
val res: ResultSet? = statement.resultSet
|
||||
if (callback != null) {
|
||||
callback(res)
|
||||
}
|
||||
statement.close()
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("SQL_ERROR", e.toString()))
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
} finally {
|
||||
activeTasks--
|
||||
if (activeTasks == 0) {
|
||||
connection.close()
|
||||
this.connection = null
|
||||
}
|
||||
}
|
||||
} else {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("NO_DATABASE_CONNECTION"))
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates or updates tables that needed an update.
|
||||
*/
|
||||
private fun performUpdate() {
|
||||
for (table in needCreation) {
|
||||
try {
|
||||
var tableDef = "CREATE TABLE IF NOT EXISTS $table ("
|
||||
|
||||
for (column in tables.find { it.first == table }!!.second) {
|
||||
tableDef += "${column.first} ${column.second},"
|
||||
}
|
||||
tableDef = tableDef.substringBeforeLast(",") + ");"
|
||||
chunkmaster.logger.finest(
|
||||
chunkmaster.langManager.getLocalized(
|
||||
"CREATE_TABLE_DEFINITION",
|
||||
table,
|
||||
tableDef
|
||||
)
|
||||
)
|
||||
executeStatement(tableDef, HashMap(), null)
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(chunkmaster.langManager.getLocalized("TABLE_CREATE_ERROR", table))
|
||||
chunkmaster.logger.severe(e.message)
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
}
|
||||
}
|
||||
for (table in needUpdate) {
|
||||
val updateSql = "ALTER TABLE ${table.first} ADD COLUMN ${table.second.first} ${table.second.second}"
|
||||
try {
|
||||
executeStatement(updateSql, HashMap(), null)
|
||||
chunkmaster.logger.finest(
|
||||
chunkmaster.langManager.getLocalized(
|
||||
"UPDATE_TABLE_DEFINITION",
|
||||
table.first,
|
||||
updateSql
|
||||
)
|
||||
)
|
||||
} catch (e: Exception) {
|
||||
chunkmaster.logger.severe(
|
||||
chunkmaster.langManager.getLocalized(
|
||||
"UPDATE_TABLE_FAILED",
|
||||
table.first,
|
||||
updateSql
|
||||
)
|
||||
)
|
||||
chunkmaster.logger.severe(e.message)
|
||||
chunkmaster.logger.info(ExceptionUtils.getStackTrace(e))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -1,70 +1,69 @@
|
||||
package net.trivernis.chunkmaster.lib.generation.taskentry
|
||||
|
||||
import io.papermc.lib.PaperLib
|
||||
import net.trivernis.chunkmaster.lib.generation.GenerationTask
|
||||
|
||||
class RunningTaskEntry(
|
||||
override val id: Int,
|
||||
override val generationTask: GenerationTask
|
||||
) : TaskEntry {
|
||||
|
||||
private var lastProgress: Pair<Long, Double>? = null
|
||||
private var lastChunkCount: Pair<Long, Int>? = null
|
||||
private var thread = Thread(generationTask)
|
||||
|
||||
/**
|
||||
* Returns the generation Speed
|
||||
*/
|
||||
val generationSpeed: Pair<Double?, Double?>
|
||||
get() {
|
||||
var generationSpeed: Double? = null
|
||||
var chunkGenerationSpeed: Double? = null
|
||||
if (lastProgress != null) {
|
||||
val progressDiff = generationTask.shape.progress() - lastProgress!!.second
|
||||
val timeDiff = (System.currentTimeMillis() - lastProgress!!.first).toDouble() / 1000
|
||||
generationSpeed = progressDiff / timeDiff
|
||||
}
|
||||
if (lastChunkCount != null) {
|
||||
val chunkDiff = generationTask.count - lastChunkCount!!.second
|
||||
val timeDiff = (System.currentTimeMillis() - lastChunkCount!!.first).toDouble() / 1000
|
||||
chunkGenerationSpeed = chunkDiff / timeDiff
|
||||
}
|
||||
lastProgress = Pair(System.currentTimeMillis(), generationTask.shape.progress())
|
||||
lastChunkCount = Pair(System.currentTimeMillis(), generationTask.count)
|
||||
return Pair(generationSpeed, chunkGenerationSpeed)
|
||||
}
|
||||
|
||||
init {
|
||||
lastProgress = Pair(System.currentTimeMillis(), generationTask.shape.progress())
|
||||
lastChunkCount = Pair(System.currentTimeMillis(), generationTask.count)
|
||||
}
|
||||
|
||||
fun start() {
|
||||
thread.start()
|
||||
}
|
||||
|
||||
fun cancel(timeout: Long): Boolean {
|
||||
if (generationTask.isRunning) {
|
||||
generationTask.cancel()
|
||||
thread.interrupt()
|
||||
}
|
||||
return try {
|
||||
joinThread(timeout)
|
||||
} catch (e: InterruptedException) {
|
||||
true
|
||||
}
|
||||
}
|
||||
|
||||
private fun joinThread(timeout: Long): Boolean {
|
||||
var threadStopped = false
|
||||
|
||||
for (i in 0..100) {
|
||||
if (!thread.isAlive || !generationTask.isRunning) {
|
||||
threadStopped = true
|
||||
break
|
||||
}
|
||||
Thread.sleep(timeout / 100)
|
||||
}
|
||||
return threadStopped
|
||||
}
|
||||
package net.trivernis.chunkmaster.lib.generation.taskentry
|
||||
|
||||
import net.trivernis.chunkmaster.lib.generation.GenerationTask
|
||||
|
||||
class RunningTaskEntry(
|
||||
override val id: Int,
|
||||
override val generationTask: GenerationTask
|
||||
) : TaskEntry {
|
||||
|
||||
private var lastProgress: Pair<Long, Double>? = null
|
||||
private var lastChunkCount: Pair<Long, Int>? = null
|
||||
private var thread = Thread(generationTask)
|
||||
|
||||
/**
|
||||
* Returns the generation Speed
|
||||
*/
|
||||
val generationSpeed: Pair<Double?, Double?>
|
||||
get() {
|
||||
var generationSpeed: Double? = null
|
||||
var chunkGenerationSpeed: Double? = null
|
||||
if (lastProgress != null) {
|
||||
val progressDiff = generationTask.shape.progress() - lastProgress!!.second
|
||||
val timeDiff = (System.currentTimeMillis() - lastProgress!!.first).toDouble() / 1000
|
||||
generationSpeed = progressDiff / timeDiff
|
||||
}
|
||||
if (lastChunkCount != null) {
|
||||
val chunkDiff = generationTask.count - lastChunkCount!!.second
|
||||
val timeDiff = (System.currentTimeMillis() - lastChunkCount!!.first).toDouble() / 1000
|
||||
chunkGenerationSpeed = chunkDiff / timeDiff
|
||||
}
|
||||
lastProgress = Pair(System.currentTimeMillis(), generationTask.shape.progress())
|
||||
lastChunkCount = Pair(System.currentTimeMillis(), generationTask.count)
|
||||
return Pair(generationSpeed, chunkGenerationSpeed)
|
||||
}
|
||||
|
||||
init {
|
||||
lastProgress = Pair(System.currentTimeMillis(), generationTask.shape.progress())
|
||||
lastChunkCount = Pair(System.currentTimeMillis(), generationTask.count)
|
||||
}
|
||||
|
||||
fun start() {
|
||||
thread.start()
|
||||
}
|
||||
|
||||
fun cancel(timeout: Long): Boolean {
|
||||
if (generationTask.isRunning) {
|
||||
generationTask.cancel()
|
||||
thread.interrupt()
|
||||
}
|
||||
return try {
|
||||
joinThread(timeout)
|
||||
} catch (e: InterruptedException) {
|
||||
true
|
||||
}
|
||||
}
|
||||
|
||||
private fun joinThread(timeout: Long): Boolean {
|
||||
var threadStopped = false
|
||||
|
||||
for (i in 0..100) {
|
||||
if (!thread.isAlive || !generationTask.isRunning) {
|
||||
threadStopped = true
|
||||
break
|
||||
}
|
||||
Thread.sleep(timeout / 100)
|
||||
}
|
||||
return threadStopped
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue