首页 > CShap > 让您知道您的方法是被何“人”调用

让您知道您的方法是被何“人”调用

也许在某些场合下我们想知道自己的某个方法是被谁(哪个方法)调用的?比如下面的例子:

/**//// <summary>
        /// 正常方法
        /// </summary>
        static void Method1()
        {
            DisabledObsoleteMethod();
        }

        /**//// <summary>
        /// 过期方法
        /// </summary>
        [Obsolete]
        static void Method2()
        {
            DisabledObsoleteMethod();
        }

        /**//// <summary>
        /// 禁止过期方法调用此方法
        /// </summary>
        static void DisabledObsoleteMethod()
        {
            //如果调用此方法的方法中有"Obsolete"标记则不允许继续运行
        }

在上面代码中,我们要在DisabledObsoleteMethod函数里限制具有“Obsolete”属性的方法调用,我们如何去做呢?

在.Net中提供了一个"StackFrame"类用于表示当前线程上的函数调用堆栈中的某个具体函数,所以我们通过它就可继续编写我们的DisabledObsoleteMethod函数,代码如下:

        /**//// <summary>
        /// 禁止过期方法调用此方法
        /// </summary>
        static void DisabledObsoleteMethod()
        {
            StackFrame frame = new StackFrame(1);       //偏移一个函数位,也即是获取当前函数的前一个调用函数
            MethodBase method = frame.GetMethod();      //取得调用函数
            //反射获取其特性
            object[] attributes = method.GetCustomAttributes(typeof(ObsoleteAttribute), false);
            if (attributes.Length > 0)
            {
                //包含有"Obsolete"标记抛出错误或做其它处理
                throw new Exception(string.Format("方法{0}包括有Obsolete属性已被禁止调用",method.Name));
            }

            //继续做其它操作
        }

到此,当运行Method1时我们的DisabledObsoleteMethod函数就可以正常运作,而Method2就会被抛出异常警告了

因为StackFrame的构造函数可以指定偏移量,所以我们可以使用它获取调用我们的函数时函数调用堆栈里都有些什么函数,也即是可以了解到当前程序的一个流程是如何的,示例代码如下:

class Test
    {
        static void Main()
        {
            int offset = 0;
            do
            {
                StackFrame frame = new StackFrame(offset++);
                MethodBase method = frame.GetMethod();
                if (method == null) break;       //如果偏移位置没有函数时,则GetMethod方法返回null
                Console.WriteLine(method.Name);

            } while (true);
            Console.Read();
        }
    }

其实.NET已经为我们提供了一个StackTrace类,其可以获取函数调用堆栈里的所有函数的有序集合,通过它我们就能将上面的代码简化为下面的代码了,如下:

   class Test
    {
        static void Main()
        {
            StackTrace trace = new StackTrace();
            foreach (StackFrame frame in trace.GetFrames())
            {
                Console.WriteLine(frame.GetMethod().Name);
            }
            Console.Read();
        }
    }

两者输出的结果还是一样的,如下

Main
_nExecuteAssembly
ExecuteAssembly
RunUsersAssembly
ThreadStart_Context
Run
ThreadStart

看来在控制台程序中也是由某个线程委托开始运作的。

其时,.NET里还有一个东东可以方便的得到当前调用方法的引用,这东东就是MethodBase,其有一个静态方法GetCurrentMethod 可直接获取当前执行的方法引用,所以我们可以将上面的StackFrame处的两句代码改为如下:

return (long)(MethodBase.GetCurrentMethod().Invoke(num, new object[] { v2, (v1 + v2), num, cout }));

原文地址

分类: CShap 标签: , ,
  1. ert
    2011年2月1日13:00 | #1

    恩,对反射不怎么了解,蛮好玩的。

    我记得这样子的话,trace.GetFrames()会被执行N次。
    foreach (StackFrame frame in trace.GetFrames())

  1. 本文目前尚无任何 trackbacks 和 pingbacks.

注意: 评论者允许使用'@user空格'的方式将自己的评论通知另外评论者。例如, ABC是本文的评论者之一,则使用'@ABC '(不包括单引号)将会自动将您的评论发送给ABC。使用'@all ',将会将评论发送给之前所有其它评论者。请务必注意user必须和评论者名相匹配(大小写一致)。直接点击评论上方的回复实现此功能