com.comcast.money.spring.TracedMethodInterceptor.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012 Comcast Cable Communications Management, LLC
 *
 * 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 com.comcast.money.spring

import java.lang.reflect.Method

import com.comcast.money.annotations.Traced
import com.comcast.money.core.logging.MethodTracer
import org.aopalliance.aop.Advice
import org.aopalliance.intercept.{ MethodInterceptor, MethodInvocation }
import org.aspectj.lang.annotation.Aspect
import org.springframework.aop.Pointcut
import org.springframework.aop.support.{ AbstractPointcutAdvisor, StaticMethodMatcherPointcut }
import org.springframework.beans.factory.annotation.{ Autowired, Qualifier }
import org.springframework.stereotype.Component

/**
 * Intercepts methods to start and stop a trace span around the method invocation
 */
@Component
class TracedMethodInterceptor @Autowired() (@Qualifier("springTracer") override val tracer: SpringTracer)
  extends MethodInterceptor
  with MethodTracer {

  override def invoke(invocation: MethodInvocation): AnyRef = {

    Option(invocation.getStaticPart.getAnnotation(classOf[Traced])) map { annotation =>
      traceMethod(invocation.getMethod, annotation, invocation.getArguments, invocation.proceed)
    } getOrElse {
      invocation.proceed()
    }
  }
}

/**
 * Used by spring so that only those methods that have the @Traced annotation
 * are actually advised
 */
@Aspect
@Component
class TracedMethodAdvisor @Autowired() (val interceptor: TracedMethodInterceptor) extends AbstractPointcutAdvisor {

  private val pointcut = new StaticMethodMatcherPointcut {
    override def matches(method: Method, targetClass: Class[_]): Boolean = method.isAnnotationPresent(classOf[Traced])
  }

  override def getPointcut: Pointcut = pointcut

  override def getAdvice: Advice = interceptor
}