TracingAspect.kt

import brave.Tracer
import org.aspectj.lang.ProceedingJoinPoint
import org.aspectj.lang.annotation.Around
import org.aspectj.lang.annotation.Aspect
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Component

@Aspect
@Component
class TracingAspect {

    @Autowired
    lateinit var tracer: Tracer

    @Around("@annotation(com.netflix.hystrix.contrib.javanica.annotation.HystrixCommand)")
    @Throws(Throwable::class)
    fun aroundHystrixCommand(joinPoint: ProceedingJoinPoint): Any? {
        return newSpan(joinPoint)
    }

    @Around("@within(org.springframework.stereotype.Service) && execution(public * *(..))")
    @Throws(Throwable::class)
    fun aroundServiceMethods(joinPoint: ProceedingJoinPoint): Any? {
        return newSpan(joinPoint)
    }

    @Around("@within(org.springframework.web.bind.annotation.RestController) && execution(public * *(..))")
    @Throws(Throwable::class)
    fun aroundRestControllerMethods(joinPoint: ProceedingJoinPoint): Any? {
        return newSpan(joinPoint)
    }

    fun newSpan(joinPoint: ProceedingJoinPoint): Any? {
        val newSpan = tracer.nextSpan().name(joinPoint.signature.declaringType.simpleName +"-"+ joinPoint.signature.name).start()
        try {
            return joinPoint.proceed();
        } finally {
            newSpan.finish()
        }
    }
}