mybatis核心组件详解——MapperAnnotationBuilder

MapperAnnotationBuilder(org.apache.ibatis.builder.annotation.MapperAnnotationBuilder),mapper注解构建器。

它的职责很简单,就是解析指定的mapper接口对应的Class对象中,包含的所有mybatis框架中定义的注解,并生成Cache、ResultMap、MappedStatement三种类型对象。


MapperAnnotationBuilder是以Class.toString()方法生成的字符串,作为Class对象的唯一标识的,在解析完Class对象后,会调用Configuration.addLoadedResource()方法把这个字符串加入配置对象的内部集合中,以防止重复解析。


MapperAnnotationBuilder总会优先解析xml配置文件,并且这个xml配置文件必须与Class对象所在的包路径一致,且文件名要与类名一致。在解析完xml配置文件后,才会开始解析Class对象中包含的注解。

源码解读:

MapperAnnotationBuilder数据结构如下:

	// 只有4个元素:@Select、@Insert、@Update、@Delete,sql语句保存在注解中
	private final Set<Class<? extends Annotation>> sqlAnnotationTypes = new HashSet<Class<? extends Annotation>>();
	// 只有4个元素:@SelectProvider、@InsertProvider、@UpdateProvider、@DeleteProvider
	// sql语句保存在注解指定的类的指定方法中
	private final Set<Class<? extends Annotation>> sqlProviderAnnotationTypes = new HashSet<Class<? extends Annotation>>();
	// 核心配置对象
	private Configuration configuration;
	// Mapper构建助手,用于组装解析出来的配置,生成Cache、ResultMap、MappedStatement等对象,
	// 并添加到Configuration配置对象中
	private MapperBuilderAssistant assistant;
	// 要解析的目标mapper接口的Class对象
	private Class<?> type;

MapperAnnotationBuilder解析逻辑如下:

public void parse() {
		// Class对象的唯一标识,如:
		// TopicMapper对应的字符串为"interface com.lixin.mapper.TopicMapper"
		String resource = type.toString();
		// 如果当前Class对象已经解析过,则不在解析
		if (!configuration.isResourceLoaded(resource)) {
			// 加载并解析指定的xml配置文件,Class所在的包对应文件路径,Class类名对应文件名称,如:
			// com.lixin.mapper.TopicMapper类对应的配置文件为com/lixin/mapper/TopicMapper.xml
			loadXmlResource();
			// 把Class对应的标识添加到已加载的资源列表中
			configuration.addLoadedResource(resource);
			// 设置当前namespace为接口Class的全限定名
			assistant.setCurrentNamespace(type.getName());
			// 解析缓存对象
			parseCache();
			// 解析缓存引用,会覆盖之前解析的缓存对象
			parseCacheRef();
			// 获取所有方法,解析方法上的注解,生成MappedStatement和ResultMap
			Method[] methods = type.getMethods();
			// 遍历所有获取到的方法
			for (Method method : methods) {
				try {
					// 解析一个方法生成对应的MapperedStatement对象
					// 并添加到配置对象中
					parseStatement(method);
				} catch (IncompleteElementException e) {
					configuration.addIncompleteMethod(new MethodResolver(this, method));
				}
			}
		}
		// 解析挂起的方法
		parsePendingMethods();
	}

loadXmlResource()方法,就是在解析指定的xml配置文件。

这里存在一个需要注意的命名问题:

XMLMapperBuilder每解析一个xml配置文件,都会以文件所在路径为xml文件的唯一标识,并把标识添加到已加载的资源文件列表中(如:com/lixin/mapper/topicMapper.xml)。但是loadXmlResource()方法中避免重复加载检查的key的却是"namespace:"+类全限定名的格式。

	private void loadXmlResource() {
		// 如果已加载资源列表中指定key已存在,则不再解析xml文件
		// 资源名称为namepace:全限定名
		if (!configuration.isResourceLoaded("namespace:" + type.getName())) {
			// 根据Class对象生成xml配置文件路径
			String xmlResource = type.getName().replace('.', '/') + ".xml";
			InputStream inputStream = null;
			try {
				// 获取文件字节流
				inputStream = Resources.getResourceAsStream(type.getClassLoader(), xmlResource);
			} catch (IOException e) {
				// ignore, resource is not required
			}
			// 如果xml文件存在,则创建XMLMapperBuilder进行解析
			if (inputStream != null) {
				XMLMapperBuilder xmlParser = new XMLMapperBuilder(inputStream, assistant.getConfiguration(),
						xmlResource, configuration.getSqlFragments(), type.getName());
				xmlParser.parse();
			}
		}
	}

之所以使用这种格式,是因为当前并不知道xml文件的真实名称是什么,与Class全限定名只是约定(换句话说,xml文件路径完全可以不遵循这种约定)。因此,为了避免重复加载,XMLMapperBuilder在解析完配置文件后,会调用bindMapperForNamespace()方法,尝试加载配置文件中根元素的namespace属性获取Class对象,并且添加"namespace:"+全限定名格式的额外的key到已加载资源列表中,来通知MapperAnnotationBuilder。


解析缓存对象:

	private void parseCache() {
		// 获取接口上的@CacheNamespace注解
		CacheNamespace cacheDomain = type.getAnnotation(CacheNamespace.class);
		// 如果存在该注解,则调用构建助手创建缓存对象
		if (cacheDomain != null) {
			assistant.useNewCache(cacheDomain.implementation(), cacheDomain.eviction(), cacheDomain.flushInterval(),
					cacheDomain.size(), cacheDomain.readWrite(), null);
		}
	}

@CacheNamespace注解对应mapper.xml配置文件中的<cache>元素,但是注解方式不支持properties自定义属性的配置。

解析缓存引用:

	private void parseCacheRef() {
		// 获取接口上的@CacheNamespaceRef注解
		CacheNamespaceRef cacheDomainRef = type.getAnnotation(CacheNamespaceRef.class);
		// 如果存在该注解,则调用构建助手添加引用关系,
		// 以Class对象的全限定名为目标namespace
		if (cacheDomainRef != null) {
			assistant.useCacheRef(cacheDomainRef.value().getName());
		}
	}

@CacheNamespaceRef注解对应mapper.xml配置文件中的<cache-ref namespace=""/>元素。


解析MappedStatement和ResultMap:

MapperAnnotationBuilder会遍历Class对象中的所有方法,一个Method对象对应一个MappedStatement对象,ResultMap的定义与xml配置文件方式不同,配置文件由单独的<resultMap>元素定义,而注解方式则定义在方法上。每个方法可以创建新的ResultMap对象,也可以引用已经存在的ResultMap对象的id。

	void parseStatement(Method method) {
		// 获取输入参数的类型,这个参数没用,
		// 因为现在都是以输入参数对象为根,通过ognl表达式寻值的
		Class<?> parameterTypeClass = getParameterType(method);
		// 通过方法上的@Lang注解获取语言驱动
		LanguageDriver languageDriver = getLanguageDriver(method);
		// 通过方法上的@Select等注解获取SqlSource
		SqlSource sqlSource = getSqlSourceFromAnnotations(method, parameterTypeClass, languageDriver);
		// 如果成功创建了SqlSource,则继续
		if (sqlSource != null) {
			// 获取方法上的@Options注解
			Options options = method.getAnnotation(Options.class);
			// 映射语句id为类的全限定名.方法名
			final String mappedStatementId = type.getName() + "." + method.getName();
			Integer fetchSize = null;
			Integer timeout = null;
			StatementType statementType = StatementType.PREPARED;
			ResultSetType resultSetType = ResultSetType.FORWARD_ONLY;
			// 通过注解获取Sql命令类型
			SqlCommandType sqlCommandType = getSqlCommandType(method);
			boolean isSelect = sqlCommandType == SqlCommandType.SELECT;
			boolean flushCache = !isSelect;
			boolean useCache = isSelect;

			KeyGenerator keyGenerator;
			String keyProperty = "id";
			String keyColumn = null;
			// 如果是insert或update命令
			if (SqlCommandType.INSERT.equals(sqlCommandType) || SqlCommandType.UPDATE.equals(sqlCommandType)) {
				// 首先检查@SelectKey注解 ,它会覆盖任何其他的配置
				// 获取方法上的SelectKey注解
				SelectKey selectKey = method.getAnnotation(SelectKey.class);
				// 如果存在@SelectKey注解
				if (selectKey != null) {
					keyGenerator = handleSelectKeyAnnotation(selectKey, mappedStatementId, getParameterType(method),
							languageDriver);
					keyProperty = selectKey.keyProperty();
				} else {
					if (options == null) {
						keyGenerator = configuration.isUseGeneratedKeys() ? new Jdbc3KeyGenerator()
								: new NoKeyGenerator();
					} else {
						keyGenerator = options.useGeneratedKeys() ? new Jdbc3KeyGenerator() : new NoKeyGenerator();
						keyProperty = options.keyProperty();
						keyColumn = options.keyColumn();
					}
				}
			} else {
				// 其他sql命令均没有键生成器
				keyGenerator = new NoKeyGenerator();
			}

			if (options != null) {
				flushCache = options.flushCache();
				useCache = options.useCache();
				fetchSize = options.fetchSize() > -1 || options.fetchSize() == Integer.MIN_VALUE ? options.fetchSize()
						: null; // issue #348
				timeout = options.timeout() > -1 ? options.timeout() : null;
				statementType = options.statementType();
				resultSetType = options.resultSetType();
			}
			////////////////////////////////////////////////////////////////////////////
			// 处理方法上的@ResultMap注解
			String resultMapId = null;
			// 获取注解,@ResultMap注解代表引用已经存在的resultMap对象的id
			ResultMap resultMapAnnotation = method.getAnnotation(ResultMap.class);
			// 如果方法上存在@ResultMap注解,则生成引用id即可
			if (resultMapAnnotation != null) {
				// 获取指定的多个resultMapId
				String[] resultMaps = resultMapAnnotation.value();
				StringBuilder sb = new StringBuilder();
				// 遍历String数组,拼接为一个String,逗号分隔
				for (String resultMap : resultMaps) {
					if (sb.length() > 0)
						sb.append(",");
					sb.append(resultMap);
				}
				resultMapId = sb.toString();
				// 不存在@ResultMap注解,且语句为select类型,
				// 则通过解析@Args、@Results等注解生成新的ResultMap对象
			} else if (isSelect) {
				// 生成新的ResultMap对象,加入Configuration配置对象
				// 同时返回resultMapId
				resultMapId = parseResultMap(method);
			}
			// 构建MappedStatement并添加到配置对象中
			assistant.addMappedStatement(mappedStatementId, sqlSource, statementType, sqlCommandType, fetchSize,
					timeout, null, // ParameterMapID
					parameterTypeClass, resultMapId, // ResultMapID
					getReturnType(method), resultSetType, flushCache, useCache, false, // TODO
																						// issue
																						// #577
					keyGenerator, keyProperty, keyColumn, null, languageDriver, null);
		}
	}

从代码逻辑可以看出,如果不存在@Select、@Insert、@Update、@Delete或者对应的@xxxProvider中的任何一个,则后续注解全是无效,只有@Lang会起作用。当使用@ResultMap注解引用已存在的结果映射时,后续关于创建新的结果映射的注解将失效。

解析注解生成新的ResultMap对象逻辑如下:

	private String parseResultMap(Method method) {
		// 获取目标bean的类型
		Class<?> returnType = getReturnType(method);
		// 获取方法上的@ConstructorArgs注解
		ConstructorArgs args = method.getAnnotation(ConstructorArgs.class);
		// 获取方法上的@Results
		Results results = method.getAnnotation(Results.class);
		// 获取方法上的@TypeDiscriminator注解
		TypeDiscriminator typeDiscriminator = method.getAnnotation(TypeDiscriminator.class);
		// 根据方法生成resultMap的唯一标识,格式为:类全限定名.方法名-参数类型简单名称
		String resultMapId = generateResultMapName(method);
		// resultMapId和返回值类型已经解析完毕,
		// 再解析剩下的构造方法映射、属性映射和鉴别器,之后添加结果映射到配置对象中
		applyResultMap(resultMapId, returnType, argsIf(args), resultsIf(results), typeDiscriminator);
		return resultMapId;
	}

注解方式只支持属性映射时使用另外的select语句,不支持嵌套的属性映射的配置。

源码的重点逻辑已经讲解完毕,具体的注解与配置文件元素的一一对应关系,并没有逐一解释,如果读者有疑问可以留言。另外关于注解的配置,我将会单独讲解。

最后给一点建议,注解配置是xml方式配置的子集,且配置后不易修改和查看,因此还是建议用户使用xml的方式来配置mapper。

你可能感兴趣的:(mybatis)