net.niebes.retrofit.metrics.RetrofitMetricsFactory.kt Maven / Gradle / Ivy

Go to download
package net.niebes.retrofit.metrics

import retrofit2.Call
import retrofit2.CallAdapter
import retrofit2.Retrofit
import retrofit2.http.DELETE
import retrofit2.http.GET
import retrofit2.http.HEAD
import retrofit2.http.HTTP
import retrofit2.http.OPTIONS
import retrofit2.http.PATCH
import retrofit2.http.POST
import retrofit2.http.PUT
import java.lang.reflect.Type

open class RetrofitMetricsFactory(
    private var metricsRecorder: MetricsRecorder,
) : CallAdapter.Factory() {

    override operator fun get(
        returnType: Type,
        annotations: Array,
        retrofit: Retrofit,
    ): CallAdapter<*, *>? {
        if (getRawType(returnType) != Call::class.java) {
            return null
        }
        val nextCallAdapter = retrofit.nextCallAdapter(this, returnType, annotations)
        return MeasuredCallAdapter(
            nextCallAdapter,
            RetrofitCallMetricsCollector(
                retrofit.baseUrl().toString(),
                getUri(annotations)!!,
                metricsRecorder
            )
        )
    }

    companion object {

        /**
         * fetches the url pattern to measure routes, not paths
         */
        private fun getUri(annotations: Array): String? =
            annotations.asSequence().mapNotNull { annotation ->
                when (annotation) {
                    is GET -> annotation.value
                    is POST -> annotation.value
                    is PUT -> annotation.value
                    is DELETE -> annotation.value
                    is PATCH -> annotation.value
                    is OPTIONS -> annotation.value
                    is HEAD -> annotation.value
                    is HTTP -> annotation.path
                    else -> null
                }
            }.firstOrNull()
    }
}