124 lines
4.1 KiB
Kotlin
124 lines
4.1 KiB
Kotlin
/*
|
|
* Copyright 2017 Hippo Seven
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package com.hippo.database
|
|
|
|
import android.content.Context
|
|
import android.database.sqlite.SQLiteOpenHelper
|
|
import android.util.SparseArray
|
|
|
|
class MSQLiteBuilder {
|
|
companion object {
|
|
const val COLUMN_ID = "_id"
|
|
private val JAVA_TYPE_TO_SQLITE_TYPE: MutableMap<Class<*>?, String> = HashMap()
|
|
private fun javaTypeToSQLiteType(clazz: Class<*>?): String {
|
|
return JAVA_TYPE_TO_SQLITE_TYPE[clazz]
|
|
?: throw IllegalStateException("Unknown type: $clazz")
|
|
}
|
|
|
|
init {
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Boolean::class.javaPrimitiveType] =
|
|
"INTEGER NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Byte::class.javaPrimitiveType] =
|
|
"INTEGER NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Short::class.javaPrimitiveType] =
|
|
"INTEGER NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Int::class.javaPrimitiveType] = "INTEGER NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Long::class.javaPrimitiveType] =
|
|
"INTEGER NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Float::class.javaPrimitiveType] =
|
|
"REAL NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[Double::class.javaPrimitiveType] = "REAL NOT NULL DEFAULT 0"
|
|
JAVA_TYPE_TO_SQLITE_TYPE[String::class.java] = "TEXT"
|
|
}
|
|
}
|
|
|
|
private val statementsMap = SparseArray<List<String>?>()
|
|
private var version = 0
|
|
private var statements: MutableList<String>? = null
|
|
|
|
/**
|
|
* Bump database version.
|
|
*/
|
|
fun version(version: Int): MSQLiteBuilder {
|
|
check(version > this.version) {
|
|
("New version must be bigger than current version. "
|
|
+ "current version: " + this.version + ", new version: " + version + ".")
|
|
}
|
|
this.version = version
|
|
statements = ArrayList()
|
|
statementsMap.put(version, statements)
|
|
return this
|
|
}
|
|
/**
|
|
* Creates a table.
|
|
*/
|
|
/**
|
|
* Creates a table with int [.COLUMN_ID] primary key.
|
|
*/
|
|
@JvmOverloads
|
|
fun createTable(
|
|
table: String,
|
|
column: String = COLUMN_ID,
|
|
clazz: Class<*>? = Int::class.javaPrimitiveType
|
|
): MSQLiteBuilder {
|
|
return statement("CREATE TABLE " + table + " (" + column + " " + javaTypeToSQLiteType(clazz) + " PRIMARY KEY);")
|
|
}
|
|
|
|
/**
|
|
* Drops a table.
|
|
*/
|
|
fun dropTable(table: String): MSQLiteBuilder {
|
|
return statement("DROP TABLE $table;")
|
|
}
|
|
|
|
/**
|
|
* Inserts a column to the table.
|
|
*/
|
|
fun insertColumn(table: String, column: String, clazz: Class<*>?): MSQLiteBuilder {
|
|
return statement(
|
|
"ALTER TABLE $table ADD COLUMN $column " + javaTypeToSQLiteType(
|
|
clazz
|
|
) + ";"
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Add a statement.
|
|
*/
|
|
fun statement(statement: String): MSQLiteBuilder {
|
|
check(!(version == 0 || statements == null)) { "Call version() first!" }
|
|
statements!!.add(statement)
|
|
return this
|
|
}
|
|
|
|
/**
|
|
* Build a SQLiteOpenHelper from it.
|
|
*/
|
|
fun build(context: Context?, name: String?, version: Int): SQLiteOpenHelper {
|
|
return MSQLiteOpenHelper(context, name, version, this)
|
|
}
|
|
|
|
fun getStatements(oldVersion: Int, newVersion: Int): List<String> {
|
|
val result: MutableList<String> = ArrayList()
|
|
for (i in oldVersion + 1..newVersion) {
|
|
val list = statementsMap[i]
|
|
if (list != null) {
|
|
result.addAll(list)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
} |