MeasureCallbackStub.kt

/*
 * Copyright (C) 2021 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.health.services.client.impl

import androidx.annotation.GuardedBy
import androidx.annotation.VisibleForTesting
import androidx.health.services.client.MeasureCallback
import androidx.health.services.client.data.DataType
import androidx.health.services.client.impl.ipc.internal.ListenerKey
import androidx.health.services.client.impl.response.AvailabilityResponse
import androidx.health.services.client.impl.response.DataPointsResponse
import com.google.common.util.concurrent.MoreExecutors
import java.util.HashMap
import java.util.concurrent.Executor

/**
 * A stub implementation for IMeasureCallback.
 *
 * @hide
 */
public class MeasureCallbackStub
private constructor(callbackKey: MeasureCallbackKey, private val callback: MeasureCallback) :
    IMeasureCallback.Stub() {

    public val listenerKey: ListenerKey = ListenerKey(callbackKey)

    @get:VisibleForTesting
    public var executor: Executor = MoreExecutors.directExecutor()
        private set

    override fun onAvailabilityChanged(response: AvailabilityResponse) {
        executor.execute {
            callback.onAvailabilityChanged(response.dataType, response.availability)
        }
    }

    override fun onData(response: DataPointsResponse) {
        executor.execute { callback.onData(response.dataPoints) }
    }

    /**
     * Its important to use the same stub for registration and un-registration, to ensure same
     * binder object is passed by framework to service side of the IPC.
     */
    public class MeasureCallbackCache private constructor() {
        @GuardedBy("this")
        private val listeners: MutableMap<MeasureCallbackKey, MeasureCallbackStub> = HashMap()

        @Synchronized
        public fun getOrCreate(
            dataType: DataType,
            measureCallback: MeasureCallback,
            executor: Executor
        ): MeasureCallbackStub {
            val callbackKey = MeasureCallbackKey(dataType, measureCallback)

            // If a measure callback happens for the same datatype with same callback, pass the same
            // stub instance, but update the executor, as executor might have changed. Its ok to
            // register
            // the callback once again, as on our service implementation re-registering for same
            // datatype
            // with  same callback is a no-op.
            var measureCallbackStub = listeners[callbackKey]
            if (measureCallbackStub == null) {
                measureCallbackStub = MeasureCallbackStub(callbackKey, measureCallback)
                listeners[callbackKey] = measureCallbackStub
            }
            measureCallbackStub.executor = executor
            return measureCallbackStub
        }

        @Synchronized
        public fun remove(
            dataType: DataType,
            measureCallback: MeasureCallback
        ): MeasureCallbackStub? {
            val callbackKey = MeasureCallbackKey(dataType, measureCallback)
            return listeners.remove(callbackKey)
        }

        public companion object {
            @JvmField public val INSTANCE: MeasureCallbackCache = MeasureCallbackCache()
        }
    }

    private data class MeasureCallbackKey(
        private val dataType: DataType,
        private val measureCallback: MeasureCallback
    )
}