Java RawSqlSource类代码示例

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String insert(Class<?> entityType) {
	// 命令名称
	String statementId = buildStatementId(entityType, "insert");
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	// 创建命令
	String sql = new SqlBuilder(entityType).insert();
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement.Builder statementBuilder = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.INSERT);
	// 判断是否需要自动生成主键
	EntityMetadata metadata = EntityMetadata.get(entityType);
	for (IdProperty idProperty : metadata.getIdProperties()) {
		KeyGenerator keyGenerator = idProperty.getKeyGenerator();
		if (keyGenerator == null) {
			continue;
		}
		String propertyName = idProperty.getName();
		String columnName = idProperty.getColumnName();
		statementBuilder.keyProperty(propertyName).keyColumn(columnName).keyGenerator(keyGenerator);
	}
	MappedStatement statement = statementBuilder.build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
@Override
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
  // issue #3
  if (script.startsWith("<script>")) {
    XPathParser parser = new XPathParser(script, false, configuration.getVariables(), new XMLMapperEntityResolver());
    return createSqlSource(configuration, parser.evalNode("/script"), parameterType);
  } else {
    // issue #127
    script = PropertyParser.parse(script, configuration.getVariables());
    TextSqlNode textSqlNode = new TextSqlNode(script);
    if (textSqlNode.isDynamic()) {
      return new DynamicSqlSource(configuration, textSqlNode);
    } else {
      return new RawSqlSource(configuration, script, parameterType);
    }
  }
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
@Override
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
  // issue #3
  if (script.startsWith("<script>")) {
    XPathParser parser = new XPathParser(script, false, configuration.getVariables(), new XMLMapperEntityResolver());
    return createSqlSource(configuration, parser.evalNode("/script"), parameterType);
  } else {
    // issue #127
    script = PropertyParser.parse(script, configuration.getVariables());
    TextSqlNode textSqlNode = new TextSqlNode(script);
    //一种是动态，一种是原始
    if (textSqlNode.isDynamic()) {
      return new DynamicSqlSource(configuration, textSqlNode);
    } else {
      return new RawSqlSource(configuration, script, parameterType);
    }
  }
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的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.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String paging(Class<?> entityType, String condition, String orderBy, LockMode lockMode) {
	String statementId = this.buildStatementId(entityType, "[email protected]" + lockMode + "|" + condition + "|" + orderBy);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	ResultMap resultMap = resultMapBuilder.getDefault(entityType);
	String sql = new SqlBuilder(entityType).paging(condition, orderBy, lockMode);
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement statement = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.SELECT).resultMaps(Arrays.asList(resultMap)).build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String select(Class<?> entityType, String condition, String orderBy, LockMode lockMode) {
	String statementId = this.buildStatementId(entityType, "[email protected]" + lockMode + "|" + condition + "|" + orderBy);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	ResultMap resultMap = resultMapBuilder.getDefault(entityType);
	String sql = new SqlBuilder(entityType).select(condition, orderBy, lockMode);
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement statement = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.SELECT).resultMaps(Arrays.asList(resultMap)).build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String count(Class<?> entityType, String condition) {
	String statementId = this.buildStatementId(entityType, "[email protected]" + condition);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	ResultMap resultMap = resultMapBuilder.count(entityType);
	String sql = new SqlBuilder(entityType).count(condition);
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement statement = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.SELECT).resultMaps(Arrays.asList(resultMap)).build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String selectId(Class<?> entityType, String condition, String orderBy , Class<?> resultType) {
	String statementId = this.buildStatementId(entityType, "[email protected]" + resultType.getName() + "|" + condition + "|" + orderBy);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	ResultMap resultMap = resultMapBuilder.getId(entityType , resultType);
	String sql = new SqlBuilder(entityType).selectId(condition, orderBy);
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement statement = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.SELECT).resultMaps(Arrays.asList(resultMap)).build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String update(Class<?> entityType, String[] fields, String condition) {
	// 命令名称
	String statementId = buildStatementId(entityType, "[email protected]" + ArrayUtils.join(fields, ",") + ">" + condition);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	String sql = new SqlBuilder(entityType).update(fields, condition);
	// 创建命令
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement.Builder statementBuilder = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.UPDATE);
	MappedStatement statement = statementBuilder.build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public String delete(Class<?> entityType, String condition) {
	// 命令名称
	String statementId = buildStatementId(entityType, "[email protected]" + condition);
	if (configuration.hasStatement(statementId)) {
		return statementId;
	}
	String sql = new SqlBuilder(entityType).delete(condition);
	// 创建命令
	SqlSource sqlSource = new RawSqlSource(configuration, sql, null);
	MappedStatement.Builder statementBuilder = new MappedStatement.Builder(configuration, statementId, sqlSource,
			SqlCommandType.DELETE);
	MappedStatement statement = statementBuilder.build();
	StatementResultMapHelper.addMappedStatement(configuration, statement);
	return statementId;
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
@Test
public void shouldUseRawSqlSourceForAnStaticStatement() {
  test("getUser1", RawSqlSource.class);
}
 

import org.apache.ibatis.scripting.defaults.RawSqlSource; //导入依赖的package包/类
public PageRawSqlSource(RawSqlSource sqlSource) {
    MetaObject metaObject = SystemMetaObject.forObject(sqlSource);
    this.sqlSource = new PageStaticSqlSource((StaticSqlSource) metaObject.getValue("sqlSource"));
}