Sysnet.Common/ActionOnDispose.cs

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

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

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace Sysnet.Common
{
    /// <summary>
    ///     Provides a wrapper over <see cref="IDisposable"/> that
    ///     invokes the provided delegate when <see cref="IDisposable.Dispose()"/>
    ///     is called.
    /// </summary>
    /// <example>
    /// <code>
    /// SqlConnection conn = new SqlConnection(connectionString);
    /// using(new ActionOnDispose(new Action(conn.Close))
    /// {
    ///     // Do work here...
    ///     // For cases where you want the connection closed
    ///     // but not disposed
    /// }
    /// </code>
    /// </example>
    public sealed class ActionOnDispose : IDisposable
    {
        /// <summary>
        ///     Creats a new <see cref="ActionOnDispose"/>
        ///     using the provided <see cref="Action"/>.
        /// </summary>
        /// <param name="unlockAction">
        ///     The <see cref="Action"/> to invoke when <see cref="Dispose"/> is called.
        /// </param>
        /// <exception cref="ArgumentNullException">if <paramref name="unlockAction"/> is null.</exception>
        public ActionOnDispose(Action unlockAction)
        {
            Util.RequireNotNull(unlockAction, "unlockAction");

            m_unlockDelegate = unlockAction;
        }

        /// <summary>
        ///     Calls the provided Action if it has not been called; 
        ///     otherwise, throws an <see cref="Exception"/>.
        /// </summary>
        /// <exception cref="Exception">If <see cref="Dispose()"/> has already been called.</exception>
        public void Dispose()
        {
            Action action = Interlocked.Exchange(ref m_unlockDelegate, null);
            if (action != null)
            {
                action();
            }
            else
            {
                throw new InvalidOperationException("Cannot call Dispose() on this object more than once..");
            }
        }