OKHttpInterceptor to trace AWS X-Ray

OkHttpXRayInterceptor.kt

package bot.data

import com.amazonaws.xray.AWSXRay
import com.amazonaws.xray.entities.Namespace
import okhttp3.Interceptor
import okhttp3.Response

class OkHttpXRayInterceptor : Interceptor {
    companion object {
        private val recorder = AWSXRay.getGlobalRecorder()
    }

    override fun intercept(chain: Interceptor.Chain): Response {
        val request = chain.request()
        val segment = recorder.beginSubsegment("OkHttp")?.apply {
            namespace = Namespace.REMOTE.toString()
            putHttp("request", mapOf(
                    "url" to request.url.toString(),
                    "method" to request.method
            ))
        }
        return chain.proceed(request).also { response ->
            segment?.run {
                isThrottle = response.code == 429
                isError = (400 until 500).contains(response.code)
                isFault = (500 until 600).contains(response.code)
                putHttp("response", mapOf("status" to response.code))
                close()
            }
        }
    }
}