Files
Umbraco-CMS/src/Umbraco.Core/IO/FileSystemExtensions.cs
2016-01-26 14:18:30 +01:00

94 lines
3.0 KiB
C#

using System;
using System.IO;
using System.Threading;
namespace Umbraco.Core.IO
{
public static class FileSystemExtensions
{
/// <summary>
/// Attempts to open the file at <code>filePath</code> up to <code>maxRetries</code> times,
/// with a thread sleep time of <code>sleepPerRetryInMilliseconds</code> between retries.
/// </summary>
public static FileStream OpenReadWithRetry(this FileInfo file, int maxRetries = 5, int sleepPerRetryInMilliseconds = 50)
{
var retries = maxRetries;
while (retries > 0)
{
try
{
return File.OpenRead(file.FullName);
}
catch(IOException)
{
retries--;
if (retries == 0)
{
throw;
}
Thread.Sleep(sleepPerRetryInMilliseconds);
}
}
throw new ArgumentException("Retries must be greater than zero");
}
public static long GetSize(this IFileSystem fs, string path)
{
// unwrap, eg MediaFileSystem is wrapping an IFileSystem
FileSystemWrapper w;
while ((w = fs as FileSystemWrapper) != null)
fs = w.Wrapped;
// no idea why GetSize is not part of IFileSystem, but
// for physical file system we have way better & faster ways
// to get the size, than to read the entire thing in memory!
var physical = fs as PhysicalFileSystem;
if (physical != null)
return physical.GetSize(path);
// other filesystems... bah...
using (var file = fs.OpenFile(path))
using (var sr = new StreamReader(file))
{
var str = sr.ReadToEnd();
return str.Length;
}
}
public static void CopyFile(this IFileSystem fs, string path, string newPath)
{
using (var stream = fs.OpenFile(path))
{
fs.AddFile(newPath, stream);
}
}
public static string GetExtension(this IFileSystem fs, string path)
{
return Path.GetExtension(fs.GetFullPath(path));
}
public static string GetFileName(this IFileSystem fs, string path)
{
return Path.GetFileName(fs.GetFullPath(path));
}
//TODO: Currently this is the only way to do this
internal static void CreateFolder(this IFileSystem fs, string folderPath)
{
var path = fs.GetRelativePath(folderPath);
var tempFile = Path.Combine(path, Guid.NewGuid().ToString("N") + ".tmp");
using (var s = new MemoryStream())
{
fs.AddFile(tempFile, s);
}
fs.DeleteFile(tempFile);
}
}
}