org.unidal.eunit.benchmark.handler.CpuHandler Maven / Gradle / Ivy

Go to download
package org.unidal.eunit.benchmark.handler;

import java.lang.reflect.AnnotatedElement;

import org.unidal.eunit.benchmark.CpuMeta;
import org.unidal.eunit.benchmark.model.entity.CaseEntity;
import org.unidal.eunit.benchmark.model.entity.CpuEntity;
import org.unidal.eunit.benchmark.model.entity.SuiteEntity;
import org.unidal.eunit.model.entity.EunitClass;
import org.unidal.eunit.model.entity.EunitMethod;
import org.unidal.eunit.testfwk.spi.IAnnotationHandler;
import org.unidal.eunit.testfwk.spi.IClassContext;

public enum CpuHandler implements IAnnotationHandler {
   INSTANCE;

   @Override
   public Class getTargetAnnotation() {
      return CpuMeta.class;
   }

   @Override
   public void handle(IClassContext ctx, CpuMeta meta, AnnotatedElement target) {
      Object source = ctx.forEunit().peek();
      CpuEntity cpu = new CpuEntity();

      cpu.setLoops(meta.loops());
      cpu.setWarmups(meta.warmups());

      if (source instanceof EunitMethod) {
         EunitMethod eunitMethod = (EunitMethod) source;
         CaseEntity c = ctx.forModel().peek();

         eunitMethod.setTest(true);
         c.setCpu(cpu);
      } else if (source instanceof EunitClass) {
         SuiteEntity s = ctx.forModel().peek();

         s.setCpu(cpu);
      }
   }

   @Override
   public boolean isAfter() {
      return false;
   }

   @Override
   public String toString() {
      return String.format("%s.%s", getClass().getSimpleName(), name());
   }
}