PreparedQueryResultAdapter.kt
/*
* Copyright 2018 The Android Open Source Project
*
* 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 androidx.room.solver.prepared.result
import androidx.room.ext.KotlinTypeNames
import androidx.room.ext.L
import androidx.room.ext.N
import androidx.room.ext.T
import androidx.room.ext.typeName
import androidx.room.parser.QueryType
import androidx.room.solver.CodeGenScope
import androidx.room.solver.prepared.binder.PreparedQueryResultBinder
import com.squareup.javapoet.FieldSpec
import isInt
import isKotlinUnit
import isLong
import isVoid
import isVoidObject
import javax.lang.model.type.TypeMirror
/**
* An adapter for [PreparedQueryResultBinder] that executes queries with INSERT, UPDATE or DELETE
* statements.
*/
class PreparedQueryResultAdapter(
private val returnType: TypeMirror,
private val queryType: QueryType
) {
companion object {
fun create(
returnType: TypeMirror,
queryType: QueryType
) = if (isValidReturnType(returnType, queryType)) {
PreparedQueryResultAdapter(returnType, queryType)
} else {
null
}
private fun isValidReturnType(returnType: TypeMirror, queryType: QueryType): Boolean {
if (returnType.isVoid() || returnType.isVoidObject() || returnType.isKotlinUnit()) {
return true
} else {
return when (queryType) {
QueryType.INSERT -> returnType.isLong()
QueryType.UPDATE, QueryType.DELETE -> returnType.isInt()
else -> false
}
}
}
}
fun executeAndReturn(
stmtQueryVal: String,
preparedStmtField: String?,
dbField: FieldSpec,
scope: CodeGenScope
) {
scope.builder().apply {
val stmtMethod = if (queryType == QueryType.INSERT) {
"executeInsert"
} else {
"executeUpdateDelete"
}
addStatement("$N.beginTransaction()", dbField)
beginControlFlow("try").apply {
if (returnType.isVoid() || returnType.isVoidObject() || returnType.isKotlinUnit()) {
addStatement("$L.$L()", stmtQueryVal, stmtMethod)
addStatement("$N.setTransactionSuccessful()", dbField)
if (returnType.isVoidObject()) {
addStatement("return null")
} else if (returnType.isKotlinUnit()) {
addStatement("return $T.INSTANCE", KotlinTypeNames.UNIT)
}
} else {
val resultVar = scope.getTmpVar("_result")
addStatement(
"final $L $L = $L.$L()",
returnType.typeName(), resultVar, stmtQueryVal, stmtMethod
)
addStatement("$N.setTransactionSuccessful()", dbField)
addStatement("return $L", resultVar)
}
}
nextControlFlow("finally").apply {
addStatement("$N.endTransaction()", dbField)
if (preparedStmtField != null) {
addStatement("$N.release($L)", preparedStmtField, stmtQueryVal)
}
}
endControlFlow()
}
}
}