BN089-BS开发框架MVC4版/source20131023/Zephyr.Net/Zephyr.Data/Command/Handlers/QueryCustomEntityHandler.cs

微信扫码关注公众号，代码整理好后第一时间通知你。

using System;

namespace Zephyr.Data
{
	internal class QueryCustomEntityHandler<TEntity> : IQueryTypeHandler<TEntity>
	{
		private readonly AutoMapper _autoMapper;
		private readonly DbCommandData _data;

		public QueryCustomEntityHandler(DbCommandData data)
		{
			_data = data;
			_autoMapper = new AutoMapper(_data, typeof(TEntity));
		}

		public bool IterateDataReader { get { return true; } }

		public object HandleType(Action<TEntity, IDataReader> customMapperReader, Action<TEntity, dynamic> customMapperDynamic)
		{
			var item = (TEntity)_data.Context.Data.EntityFactory.Create(typeof(TEntity));

			if (customMapperReader != null)
				customMapperReader(item, _data.Reader);
			else if (customMapperDynamic != null)
				customMapperDynamic(item, new DynamicDataReader(_data.Reader.InnerReader));
			else
				_autoMapper.AutoMap(item);

			return item;
		}
	}
}