jvmCommonMain.kotbase.Query.jvmCommon.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of couchbase-lite Show documentation
Couchbase Lite Community Edition for Kotlin Multiplatform
There is a newer version: 3.1.3-1.1.0
/*
 * Copyright 2022-2023 Jeff Lockhart
 *
 * 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 kotbase

import kotbase.internal.DelegatedClass
import kotlinx.coroutines.CoroutineDispatcher
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Job
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.asExecutor
import kotlinx.coroutines.cancel
import kotlin.coroutines.CoroutineContext
import com.couchbase.lite.From as CBLFrom
import com.couchbase.lite.GroupBy as CBLGroupBy
import com.couchbase.lite.Having as CBLHaving
import com.couchbase.lite.Joins as CBLJoins
import com.couchbase.lite.Limit as CBLLimit
import com.couchbase.lite.OrderBy as CBLOrderBy
import com.couchbase.lite.Query as CBLQuery
import com.couchbase.lite.Select as CBLSelect
import com.couchbase.lite.Where as CBLWhere

internal class DelegatedQuery(actual: CBLQuery) : DelegatedClass(actual), Query {

    override var parameters: Parameters?
        get() = actual.parameters?.asParameters()
        set(value) {
            actual.parameters = value?.actual
        }

    @Throws(CouchbaseLiteException::class)
    override fun execute(): ResultSet =
        ResultSet(actual.execute())

    @Throws(CouchbaseLiteException::class)
    override fun explain(): String =
        actual.explain()

    override fun addChangeListener(listener: QueryChangeListener): ListenerToken =
        actual.addChangeListener(listener.convert())

    @OptIn(ExperimentalStdlibApi::class)
    override fun addChangeListener(context: CoroutineContext, listener: QueryChangeSuspendListener): ListenerToken {
        val scope = CoroutineScope(SupervisorJob() + context)
        val token = actual.addChangeListener(context[CoroutineDispatcher]?.asExecutor(), listener.convert(scope))
        return SuspendListenerToken(scope, token)
    }

    @OptIn(ExperimentalStdlibApi::class)
    override fun addChangeListener(scope: CoroutineScope, listener: QueryChangeSuspendListener) {
        val token = actual.addChangeListener(
            scope.coroutineContext[CoroutineDispatcher]?.asExecutor(),
            listener.convert(scope)
        )
        scope.coroutineContext[Job]?.invokeOnCompletion {
            actual.removeChangeListener(token)
        }
    }

    override fun removeChangeListener(token: ListenerToken) {
        if (token is SuspendListenerToken) {
            actual.removeChangeListener(token.actual)
            token.scope.cancel()
        } else {
            actual.removeChangeListener(token)
        }
    }
}

internal fun CBLQuery.asQuery(): Query = when (this) {
    is CBLSelect -> Select(this)
    is CBLFrom -> From(this)
    is CBLJoins -> Joins(this)
    is CBLWhere -> Where(this)
    is CBLGroupBy -> GroupBy(this)
    is CBLHaving -> Having(this)
    is CBLOrderBy -> OrderBy(this)
    is CBLLimit -> Limit(this)
    else -> error("Unknown Query type ${this::class}")
}