package nl.beesting.beangenerator.locators;

 import nl.beesting.beangenerator.util.DataConfiguration;

 import org.aopalliance.intercept.MethodInvocation;

 /**

  * ClassLocator that makes use of the annotaions.

  * @author Erik Jan de Wit

  */

 public class AnnotationClassLocator implements ClassLocator {

         public Class<?> findClass(MethodInvocation invocation) {

                 DataConfiguration configuration = invocation.getMethod().getAnnotation(DataConfiguration.class);

                 return configuration.returnType();

         }

 }