Java ProviderSqlSource类代码示例

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
private SqlSource getSqlSourceFromAnnotations(Method method, Class<?> parameterType, LanguageDriver languageDriver) {
    try {
        Class<? extends Annotation> sqlAnnotationType = getSqlAnnotationType(method);
        Class<? extends Annotation> sqlProviderAnnotationType = getSqlProviderAnnotationType(method);
        if (sqlAnnotationType != null) {
            if (sqlProviderAnnotationType != null) {
                throw new BindingException("You cannot supply both a static SQL and SqlProvider to method named " + method.getName());
            }
            Annotation sqlAnnotation = method.getAnnotation(sqlAnnotationType);
            final String[] strings = (String[]) sqlAnnotation.getClass().getMethod("value").invoke(sqlAnnotation);
            return buildSqlSourceFromStrings(strings, parameterType, languageDriver);
        } else if (sqlProviderAnnotationType != null) {
            Annotation sqlProviderAnnotation = method.getAnnotation(sqlProviderAnnotationType);
            return new ProviderSqlSource(assistant.getConfiguration(), sqlProviderAnnotation);
        }
        return null;
    } catch (Exception e) {
        throw new BuilderException("Could not find value method on SQL annotation.  Cause: " + e, e);
    }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
/**
 * 配置指定的接口
 *
 * @param configuration
 * @param mapperInterface
 */
public void processConfiguration(Configuration configuration, Class<?> mapperInterface) {
    String prefix;
    if (mapperInterface != null) {
        prefix = mapperInterface.getCanonicalName();
    } else {
        prefix = "";
    }
    for (Object object : new ArrayList<Object>(configuration.getMappedStatements())) {
        if (object instanceof MappedStatement) {
            MappedStatement ms = (MappedStatement) object;
            if (ms.getId().startsWith(prefix) && isMapperMethod(ms.getId())) {
                if (ms.getSqlSource() instanceof ProviderSqlSource) {
                    setSqlSource(ms);
                }
            }
        }
    }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
private SqlSource getSqlSourceFromAnnotations(Method method, Class<?> parameterType, LanguageDriver languageDriver) {
    try {
        Class<? extends Annotation> sqlAnnotationType = getSqlAnnotationType(method);
        Class<? extends Annotation> sqlProviderAnnotationType = getSqlProviderAnnotationType(method);
        if (sqlAnnotationType != null) {
            if (sqlProviderAnnotationType != null) {
                throw new BindingException("You cannot supply both a static SQL and SqlProvider to method named " + method.getName());
            }
            Annotation sqlAnnotation = method.getAnnotation(sqlAnnotationType);
            final String[] strings = (String[]) sqlAnnotation.getClass().getMethod("value").invoke(sqlAnnotation);
            return buildSqlSourceFromStrings(strings, parameterType, languageDriver);
        } else if (sqlProviderAnnotationType != null) {
            Annotation sqlProviderAnnotation = method.getAnnotation(sqlProviderAnnotationType);
            return new ProviderSqlSource(assistant.getConfiguration(), sqlProviderAnnotation, type, method);
        }
        return null;
    } catch (Exception e) {
        throw new BuilderException("Could not find value method on SQL annotation.  Cause: " + e, e);
    }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
/**
 * 修改SqlSource
 *
 * @param ms
 * @throws Throwable
 */
public void processMappedStatement(MappedStatement ms) throws Throwable {
    SqlSource sqlSource = ms.getSqlSource();
    MetaObject msObject = SystemMetaObject.forObject(ms);
    SqlSource pageSqlSource;
    if (sqlSource instanceof StaticSqlSource) {
        pageSqlSource = new PageStaticSqlSource((StaticSqlSource) sqlSource);
    } else if (sqlSource instanceof RawSqlSource) {
        pageSqlSource = new PageRawSqlSource((RawSqlSource) sqlSource);
    } else if (sqlSource instanceof ProviderSqlSource) {
        pageSqlSource = new PageProviderSqlSource((ProviderSqlSource) sqlSource);
    } else if (sqlSource instanceof DynamicSqlSource) {
        pageSqlSource = new PageDynamicSqlSource((DynamicSqlSource) sqlSource);
    } else {
        throw new RuntimeException("无法处理该类型[" + sqlSource.getClass() + "]的SqlSource");
    }
    msObject.setValue("sqlSource", pageSqlSource);
    //由于count查询需要修改返回值，因此这里要创建一个Count查询的MS
    msCountMap.put(ms.getId(), MSUtils.newCountMappedStatement(ms));
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void methodNotFound() throws NoSuchMethodException {
  expectedException.expect(BuilderException.class);
  expectedException.expectMessage(is("Error creating SqlSource for SqlProvider. Method 'methodNotFound' not found in SqlProvider 'org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder'."));
  new ProviderSqlSource(new Configuration(),
          ErrorMapper.class.getMethod("methodNotFound").getAnnotation(SelectProvider.class));
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void methodOverload() throws NoSuchMethodException {
  expectedException.expect(BuilderException.class);
  expectedException.expectMessage(is("Error creating SqlSource for SqlProvider. Method 'overload' is found multiple in SqlProvider 'org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder'. Sql provider method can not overload."));
  new ProviderSqlSource(new Configuration(),
          ErrorMapper.class.getMethod("methodOverload", String.class).getAnnotation(SelectProvider.class));
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void notSupportParameterObjectOnMultipleArguments() throws NoSuchMethodException {
  expectedException.expect(BuilderException.class);
  expectedException.expectMessage(is("Error invoking SqlProvider method (org.apache.ibatis.submitted.sqlprovider.OurSqlBuilder.buildGetUsersByNameQuery). Cannot invoke a method that holds multiple arguments using a specifying parameterObject. In this case, please specify a 'java.util.Map' object."));
  new ProviderSqlSource(new Configuration(),
          Mapper.class.getMethod("getUsersByName", String.class, String.class).getAnnotation(SelectProvider.class))
          .getBoundSql(new Object());
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void notSupportParameterObjectOnNamedArgument() throws NoSuchMethodException {
  expectedException.expect(BuilderException.class);
  expectedException.expectMessage(is("Error invoking SqlProvider method (org.apache.ibatis.submitted.sqlprovider.OurSqlBuilder.buildGetUsersByNameWithParamNameQuery). Cannot invoke a method that holds named argument(@Param) using a specifying parameterObject. In this case, please specify a 'java.util.Map' object."));
  new ProviderSqlSource(new Configuration(),
          Mapper.class.getMethod("getUsersByNameWithParamName", String.class).getAnnotation(SelectProvider.class))
          .getBoundSql(new Object());
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void invokeError() throws NoSuchMethodException {
  expectedException.expect(BuilderException.class);
  expectedException.expectMessage(is("Error invoking SqlProvider method (org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder.invokeError).  Cause: java.lang.reflect.InvocationTargetException"));
  new ProviderSqlSource(new Configuration(),
          ErrorMapper.class.getMethod("invokeError").getAnnotation(SelectProvider.class))
          .getBoundSql(new Object());
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
public PageProviderSqlSource(ProviderSqlSource provider) {
    MetaObject metaObject = SystemMetaObject.forObject(provider);
    this.sqlSourceParser = (SqlSourceBuilder) metaObject.getValue("sqlSourceParser");
    this.providerType = (Class<?>) metaObject.getValue("providerType");
    this.providerMethod = (Method) metaObject.getValue("providerMethod");
    this.configuration = (Configuration) metaObject.getValue("sqlSourceParser.configuration");
    try {
        //先针对3.3.1和之前版本做判断
        this.providerTakesParameterObject = (Boolean) metaObject.getValue("providerTakesParameterObject");
    } catch (ReflectionException e) {
        //3.4.0+版本，解决#102 by Ian Lim
        providerMethodArgumentNames = (String[]) metaObject.getValue("providerMethodArgumentNames");
    }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
/**
 * 处理 MappedStatement
 *
 * @param ms
 */
public void processMappedStatement(MappedStatement ms){
    MapperTemplate mapperTemplate = isMapperMethod(ms.getId());
    if(mapperTemplate != null && ms.getSqlSource() instanceof ProviderSqlSource) {
        setSqlSource(ms, mapperTemplate);
    }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void methodNotFound() throws NoSuchMethodException {
  try {
    Class<?> mapperType = ErrorMapper.class;
    Method mapperMethod = mapperType.getMethod("methodNotFound");
    new ProviderSqlSource(new Configuration(),
          mapperMethod.getAnnotation(SelectProvider.class), mapperType, mapperMethod);
    fail();
  } catch (BuilderException e) {
    assertTrue(e.getMessage().contains("Error creating SqlSource for SqlProvider. Method 'methodNotFound' not found in SqlProvider 'org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder'."));
  }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void methodOverload() throws NoSuchMethodException {
  try {
    Class<?> mapperType = ErrorMapper.class;
    Method mapperMethod = mapperType.getMethod("methodOverload", String.class);
    new ProviderSqlSource(new Configuration(),
            mapperMethod.getAnnotation(SelectProvider.class), mapperType, mapperMethod);
    fail();
  } catch (BuilderException e) {
    assertTrue(e.getMessage().contains("Error creating SqlSource for SqlProvider. Method 'overload' is found multiple in SqlProvider 'org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder'. Sql provider method can not overload."));
  }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void notSqlProvider() throws NoSuchMethodException {
  try {
    new ProviderSqlSource(new Configuration(), new Object(), null, null);
    fail();
  } catch (BuilderException e) {
    assertTrue(e.getMessage().contains("Error creating SqlSource for SqlProvider.  Cause: java.lang.NoSuchMethodException: java.lang.Object.type()"));
  }
}
 

import org.apache.ibatis.builder.annotation.ProviderSqlSource; //导入依赖的package包/类
@Test
public void multipleProviderContext() throws NoSuchMethodException {
  try {
    Class<?> mapperType = ErrorMapper.class;
    Method mapperMethod = mapperType.getMethod("multipleProviderContext");
    new ProviderSqlSource(new Configuration(),
          mapperMethod.getAnnotation(SelectProvider.class), mapperType, mapperMethod);
    fail();
  } catch (BuilderException e) {
    assertTrue(e.getMessage().contains("Error creating SqlSource for SqlProvider. ProviderContext found multiple in SqlProvider method (org.apache.ibatis.submitted.sqlprovider.SqlProviderTest$ErrorSqlBuilder.multipleProviderContext). ProviderContext can not define multiple in SqlProvider method argument."));
  }
}