FluentData/Command/DbCommand.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

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

using System;

namespace FluentData
{
	internal partial class DbCommand : IDisposable, IDbCommand
	{
		private readonly DbCommandData _data;

		public DbCommand(
			DbContext dbContext,
			System.Data.IDbCommand dbCommand,
			DbContextData dbContextData)
		{
			_data = new DbCommandData();
			_data.Context = dbContext;
			_data.ContextData = dbContextData;
			_data.InnerCommand = dbCommand;
			_data.Command = this;
			_data.ExecuteQueryHandler = new ExecuteQueryHandler(_data, this);
		}

		internal IDbCommand UseMultipleResultset
		{
			get
			{
				if (!_data.ContextData.Provider.SupportsMultipleResultset)
					throw new FluentDataException("The selected database does not support multiple resultset");

				_data.UseMultipleResultsets = true;
				return this;
			}
		}

		public IDbCommand CommandType(DbCommandTypes dbCommandType)
		{
			_data.CommandType = dbCommandType;
			_data.InnerCommand.CommandType = (System.Data.CommandType) dbCommandType;
			return this;
		}

		internal void ClosePrivateConnection()
		{
			if (!_data.ContextData.UseTransaction
				&& !_data.ContextData.UseSharedConnection)
			{
				_data.InnerCommand.Connection.Close();

				if (_data.ContextData.OnConnectionClosed != null)
					_data.ContextData.OnConnectionClosed(new OnConnectionClosedEventArgs(_data.InnerCommand.Connection));
			}
		}

		public void Dispose()
		{
			if (_data.Reader != null)
				_data.Reader.Close();

			ClosePrivateConnection();
		}
	}