CallSiteInformation - Prepare for fast classname lookup from filename…

… and linenumber
NLog · Dec 3, 2017 · 9038a7e · 9038a7e
1 parent 46d55f4
commit 9038a7e
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 10 deletions.
diff --git a/src/NLog/Fluent/LogBuilder.cs b/src/NLog/Fluent/LogBuilder.cs
@@ -306,7 +306,7 @@ public LogBuilder StackTrace(StackTrace stackTrace, int userStackFrame)
             if (callerLineNumber != 0)
                 Property("CallerLineNumber", callerLineNumber);
 
-            _logEvent.SetCallerInfo(callerMemberName, callerFilePath, callerLineNumber);
+            _logEvent.SetCallerInfo(null, callerMemberName, callerFilePath, callerLineNumber);
 
             _logger.Log(_logEvent);
         }

diff --git a/src/NLog/Internal/CallSiteInformation.cs b/src/NLog/Internal/CallSiteInformation.cs
@@ -55,11 +55,13 @@ public void SetStackTrace(StackTrace stackTrace, int userStackFrame, int? userSt
         /// <summary>
         /// Sets the details retrieved from the Caller Information Attributes
         /// </summary>
+        /// <param name="callerClassName"></param>
         /// <param name="callerMemberName"></param>
         /// <param name="callerFilePath"></param>
         /// <param name="callerLineNumber"></param>
-        public void SetCallerInfo(string callerMemberName, string callerFilePath, int callerLineNumber)
+        public void SetCallerInfo(string callerClassName, string callerMemberName, string callerFilePath, int callerLineNumber)
         {
+            CallerClassName = callerClassName;
             CallerMemberName = callerMemberName;
             CallerFilePath = callerFilePath;
             CallerLineNumber = callerLineNumber;
@@ -94,6 +96,22 @@ public MethodBase GetCallerStackFrameMethod(int skipFrames)
 
         public string GetCallerClassName(MethodBase method, bool includeNameSpace, bool cleanAsyncMoveNext, bool cleanAnonymousDelegates)
         {
+            if (!string.IsNullOrEmpty(CallerClassName))
+            {
+                if (includeNameSpace)
+                {
+                    return CallerClassName;
+                }
+                else
+                {
+                    int lastDot = CallerClassName.LastIndexOf('.');
+                    if (lastDot < 0 || lastDot >= CallerClassName.Length - 1)
+                        return CallerClassName;
+                    else
+                        return CallerClassName.Substring(lastDot + 1);
+                }
+            }
+
             method = method ?? GetCallerStackFrameMethod(0);
             if (method == null)
                 return string.Empty;
@@ -139,6 +157,7 @@ public int GetCallerLineNumber(int skipFrames)
             return frame?.GetFileLineNumber() ?? 0;
         }
 
+        public string CallerClassName { get; private set; }
         public string CallerMemberName { get; private set; }
         public string CallerFilePath { get; private set; }
         public int? CallerLineNumber { get; private set; }

diff --git a/src/NLog/LayoutRenderers/CallSiteLayoutRenderer.cs b/src/NLog/LayoutRenderers/CallSiteLayoutRenderer.cs
@@ -31,17 +31,14 @@
 // THE POSSIBILITY OF SUCH DAMAGE.
 // 
 
-using System;
-
 namespace NLog.LayoutRenderers
 {
+    using System;
     using System.ComponentModel;
-    using System.Diagnostics;
     using System.IO;
-    using System.Reflection;
     using System.Text;
-    using Config;
-    using Internal;
+    using NLog.Config;
+    using NLog.Internal;
 
     /// <summary>
     /// The call site (class name, method name and source information).

diff --git a/src/NLog/LogEventInfo.cs b/src/NLog/LogEventInfo.cs
@@ -193,6 +193,11 @@ public LogEventInfo(LogLevel level, string loggerName, IFormatProvider formatPro
         /// </summary>
         public StackTrace StackTrace => CallSiteInformation?.StackTrace;
 
+        /// <summary>
+        /// Gets the callsite class name
+        /// </summary>
+        public string CallerClassName => CallSiteInformation?.CallerClassName;
+
         /// <summary>
         /// Gets the callsite member function name
         /// </summary>
@@ -516,12 +521,13 @@ public void SetStackTrace(StackTrace stackTrace, int userStackFrame)
         /// <summary>
         /// Sets the details retrieved from the Caller Information Attributes
         /// </summary>
+        /// <param name="callerClassName"></param>
         /// <param name="callerMemberName"></param>
         /// <param name="callerFilePath"></param>
         /// <param name="callerLineNumber"></param>
-        public void SetCallerInfo(string callerMemberName, string callerFilePath, int callerLineNumber)
+        public void SetCallerInfo(string callerClassName, string callerMemberName, string callerFilePath, int callerLineNumber)
         {
-            GetCallSiteInformationInternal().SetCallerInfo(callerMemberName, callerFilePath, callerLineNumber);
+            GetCallSiteInformationInternal().SetCallerInfo(callerClassName, callerMemberName, callerFilePath, callerLineNumber);
         }
 
         internal string AddCachedLayoutValue(Layout layout, string value)