LoggerTask.cs

Go to the documentation of this file.

using System;
using System.Collections.Concurrent;
using System.IO;

namespace Logging
{
    class LoggerTask
    {
        private readonly StreamWriter streamWriter;
        private readonly object lockObject = new Object();
        private bool stop;

        public bool IsStopped { get; private set; }

        private readonly BlockingCollection<LoggerInfo> logEntries = new BlockingCollection<LoggerInfo>();

        public LoggerTask(string folder, string fileName)
        {
            fileName = String.IsNullOrWhiteSpace(fileName) 
                ? $"{DateTime.Now.ToFileTime()}_DefaultLog.log"
                : $"{DateTime.Now.ToFileTime()}_{Path.GetFileName(fileName)}.log";
            this.streamWriter = new StreamWriter(Path.Combine(folder, fileName));
        }

        public void DoWork()
        {
            while (!this.stop)
            {
                LoggerInfo logInfo = this.logEntries.Take();
                if (null != logInfo)
                    this.Log(logInfo);
            }

            if (this.streamWriter == null) 
                return;

            lock (this.lockObject)
            {
                this.streamWriter.Close();
            }

            this.IsStopped = true;
        }

        private void Log(LoggerInfo logInfo)
        {
            lock (this.lockObject)
            {
                this.streamWriter.WriteLine(logInfo.ToString());
                this.streamWriter.Flush();
            }
        }

        internal void EnqueueObject(LoggerInfo loggerInfo)
        {
            this.logEntries.Add(loggerInfo);
        }

        internal void Stop()
        {
            this.stop = true;
            this.logEntries.Add(null);
        }
    }
}