pepperdash/Essentials
1
0
Fork 0
mirror of https://github.com/PepperDash/Essentials.git synced 2026-02-06 16:25:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

File IO Rate Limiting

Browse Source 
feature: adds rate limiting for FileIO
This commit is contained in:
jta
2022-06-07 12:54:28 -04:00
parent 96dbd44a04
commit 50e94eb373
1 changed files with 46 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
essentials-framework/Essentials Core/PepperDashEssentialsBase/File/FileIO.cs
View File

@@ -15,14 +15,19 @@ namespace PepperDash.Essentials.Core
static CCriticalSection fileLock = new CCriticalSection();
public delegate void GotFileEventHandler(object sender, FileEventArgs e);
public static event GotFileEventHandler GotFileEvent;
private static Dictionary<string, CTimer> WriteTimers;
public const long WriteTimeout = 5000;
/// <summary>
/// Get the full file info from a path/filename, can include wildcards.
/// </summary>
/// <param name="fileName"></param>
/// <returns></returns>
///
static FileIO()
{
WriteTimers = new Dictionary<string, CTimer>();
}
public static FileInfo[] GetFiles(string fileName)
{
DirectoryInfo dirInfo = new DirectoryInfo(Path.GetDirectoryName(fileName));
@@ -170,7 +175,8 @@ namespace PepperDash.Essentials.Core
}
else
{
Debug.Console(0, Debug.ErrorLogLevel.Error, "FileIO Unable to enter FileLock");
Debug.Console(0, Debug.ErrorLogLevel.Error, "FileIO Unable to enter FileLock retrying");
}
}
@@ -190,6 +196,41 @@ namespace PepperDash.Essentials.Core
}
/// <summary>
/// Resets (or starts) the write timer
/// </summary>
static void ResetTimer(string data, string filePath)
{
try
{
if (WriteTimers.ContainsKey(filePath))
{
WriteTimers[filePath].Stop();
WriteTimers[filePath] = null;
WriteTimers.Remove(filePath);
}
WriteTimers.Add(filePath, new CTimer((o) =>
{
Thread _WriteFileThread;
_WriteFileThread = new Thread((O) => _WriteFileMethod(data, filePath), null, Thread.eThreadStartOptions.CreateSuspended);
_WriteFileThread.Priority = Thread.eThreadPriority.LowestPriority;
_WriteFileThread.Start();
Debug.Console(0, Debug.ErrorLogLevel.Notice, "New WriteFile Thread");
WriteTimers.Remove(filePath);
}, WriteTimeout));
Debug.Console(0, "FileIO write timer has been reset.");
}
catch (Exception e)
{
Debug.Console(0, Debug.ErrorLogLevel.Error, "Error: FileIO ResetTimer failed: \r{0}", e);
data = "";
}
}
/// <summary>
///
/// </summary>
@@ -202,11 +243,7 @@ namespace PepperDash.Essentials.Core
/// <param name="filePath"></param>
public static void WriteDataToFile(string data, string filePath)
{
Thread _WriteFileThread;
_WriteFileThread = new Thread((O) => _WriteFileMethod(data, filePath), null, Thread.eThreadStartOptions.CreateSuspended);
_WriteFileThread.Priority = Thread.eThreadPriority.LowestPriority;
_WriteFileThread.Start();
Debug.Console(0, Debug.ErrorLogLevel.Notice, "New WriteFile Thread");
ResetTimer(data, filePath);
}
@@ -241,6 +278,7 @@ namespace PepperDash.Essentials.Core
fileLock.Leave();
}
Debug.Console(0, Debug.ErrorLogLevel.Notice, "File Written to : '{0}'", filePath);
return null;
}