Home | History | Annotate | Download | only in binderprovider 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package androidx.room.solver.binderprovider
     18 
     19 import androidx.room.ext.RoomRxJava2TypeNames
     20 import androidx.room.ext.RxJava2TypeNames
     21 import androidx.room.processor.Context
     22 import androidx.room.processor.ProcessorErrors
     23 import androidx.room.solver.ObservableQueryResultBinderProvider
     24 import androidx.room.solver.query.result.FlowableQueryResultBinder
     25 import androidx.room.solver.query.result.QueryResultAdapter
     26 import androidx.room.solver.query.result.QueryResultBinder
     27 import javax.lang.model.type.DeclaredType
     28 import javax.lang.model.type.TypeMirror
     29 
     30 class FlowableQueryResultBinderProvider(context: Context) :
     31         ObservableQueryResultBinderProvider(context) {
     32     private val flowableTypeMirror: TypeMirror? by lazy {
     33         context.processingEnv.elementUtils
     34                 .getTypeElement(RxJava2TypeNames.FLOWABLE.toString())?.asType()
     35     }
     36 
     37     private val hasRxJava2Artifact by lazy {
     38         context.processingEnv.elementUtils
     39                 .getTypeElement(RoomRxJava2TypeNames.RX_ROOM.toString()) != null
     40     }
     41 
     42     override fun extractTypeArg(declared: DeclaredType): TypeMirror = declared.typeArguments.first()
     43 
     44     override fun create(typeArg: TypeMirror, resultAdapter: QueryResultAdapter?,
     45                         tableNames: Set<String>): QueryResultBinder {
     46         return FlowableQueryResultBinder(
     47                 typeArg = typeArg,
     48                 queryTableNames = tableNames,
     49                 adapter = resultAdapter)
     50     }
     51 
     52     override fun matches(declared: DeclaredType): Boolean =
     53             declared.typeArguments.size == 1 && isRxJava2Publisher(declared)
     54 
     55     private fun isRxJava2Publisher(declared: DeclaredType): Boolean {
     56         if (flowableTypeMirror == null) {
     57             return false
     58         }
     59         val erasure = context.processingEnv.typeUtils.erasure(declared)
     60         val match = context.processingEnv.typeUtils.isAssignable(flowableTypeMirror, erasure)
     61         if (match && !hasRxJava2Artifact) {
     62             context.logger.e(ProcessorErrors.MISSING_ROOM_RXJAVA2_ARTIFACT)
     63         }
     64         return match
     65     }
     66 }