PowerToys/Wox.Infrastructure/Timeit.cs

using System;
using System.Diagnostics;
using Wox.Infrastructure.Logger;

namespace Wox.Infrastructure
{
    public static class Timeit
    {
        /// <summary>
        /// This stopwatch will appear only in Debug mode
        /// </summary>
        public static void StopwatchDebug(string name, Action action)
        {
#if DEBUG
            Stopwatch(name, action);
#else
            action();
#endif
        }

        [Conditional("DEBUG")]
        private static void WriteTimeInfo(string name, long milliseconds)
        {
            string info = $"{name} : {milliseconds}ms";
            Debug.WriteLine(info);
            Log.Info(info);
        }

        /// <summary>
        /// This stopwatch will also appear only in Debug mode
        /// </summary>
        public static long Stopwatch(string name, Action action)
        {
            var stopWatch = new Stopwatch();
            stopWatch.Start();
            action();
            stopWatch.Stop();
            var milliseconds = stopWatch.ElapsedMilliseconds;
            WriteTimeInfo(name, milliseconds);
            return milliseconds;
        }

    }
}