diff options
author | Ivan Topolnjak <ivantopo@gmail.com> | 2014-04-06 02:29:08 -0300 |
---|---|---|
committer | Ivan Topolnjak <ivantopo@gmail.com> | 2014-04-06 02:29:08 -0300 |
commit | 45c7988ab2fe5b83d67e51a903830ae68b54284e (patch) | |
tree | a3ddd29c8c6e9977dcccbf1ca277618f386be690 /kamon-core/src/main/scala/akka/instrumentation | |
parent | 58b5438784219878880252e0b13be9a7ed7c74c0 (diff) | |
parent | f5023049f0abb7d7e61241e1dcbbea477d48b27f (diff) | |
download | Kamon-45c7988ab2fe5b83d67e51a903830ae68b54284e.tar.gz Kamon-45c7988ab2fe5b83d67e51a903830ae68b54284e.tar.bz2 Kamon-45c7988ab2fe5b83d67e51a903830ae68b54284e.zip |
Merge branch 'master' of github.com:kamon-io/Kamon
Diffstat (limited to 'kamon-core/src/main/scala/akka/instrumentation')
-rw-r--r-- | kamon-core/src/main/scala/akka/instrumentation/UnboundedMailboxInstrumentation.scala | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/kamon-core/src/main/scala/akka/instrumentation/UnboundedMailboxInstrumentation.scala b/kamon-core/src/main/scala/akka/instrumentation/UnboundedMailboxInstrumentation.scala new file mode 100644 index 00000000..6a335dfa --- /dev/null +++ b/kamon-core/src/main/scala/akka/instrumentation/UnboundedMailboxInstrumentation.scala @@ -0,0 +1,58 @@ +/* + * ========================================================================================= + * Copyright © 2013-2014 the kamon project <http://kamon.io/> + * + * 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 akka.instrumentation + +import org.aspectj.lang.annotation._ +import java.util.concurrent.atomic.AtomicInteger +import org.aspectj.lang.ProceedingJoinPoint + +/** + * This aspect adds a counter to the message queue to get O(1) time in the .numberOfMessages method implementation + */ +@Aspect +class UnboundedMailboxInstrumentation { + + @DeclareMixin("akka.dispatch.UnboundedMailbox$MessageQueue") + def mixinMailboxToMailboxSizeAware: MessageQueueSizeCounting = new MessageQueueSizeCounting {} + + @Pointcut("execution(* akka.dispatch.UnboundedMailbox$MessageQueue.enqueue(..)) && this(queue)") + def enqueuePointcut(queue: MessageQueueSizeCounting): Unit = {} + + @Pointcut("execution(* akka.dispatch.UnboundedMailbox$MessageQueue.dequeue()) && this(queue)") + def dequeuePointcut(queue: MessageQueueSizeCounting): Unit = {} + + @Pointcut("execution(* akka.dispatch.UnboundedMailbox$MessageQueue.numberOfMessages()) && this(queue)") + def numberOfMessagesPointcut(queue: MessageQueueSizeCounting): Unit = {} + + @After("dequeuePointcut(queue)") + def afterDequeuePointcut(queue: MessageQueueSizeCounting): Unit = { + if (queue.queueSize.get() > 0) queue.queueSize.decrementAndGet() + } + + @After("enqueuePointcut(queue)") + def afterEnqueue(queue: MessageQueueSizeCounting): Unit = { + queue.queueSize.incrementAndGet() + } + + @Around("numberOfMessagesPointcut(queue)") + def aroundNumberOfMessages(pjp: ProceedingJoinPoint, queue: MessageQueueSizeCounting): Any = { + queue.queueSize.get() + } +} + +trait MessageQueueSizeCounting { + val queueSize = new AtomicInteger +} |