jellyfin/MediaBrowser.Controller/Providers/DirectoryService.cs

88 lines
2.5 KiB
C#
Raw Normal View History

#pragma warning disable CS1591
using System;
using System.Collections.Concurrent;
2018-12-28 00:27:57 +01:00
using System.Collections.Generic;
using System.Linq;
using MediaBrowser.Model.IO;
namespace MediaBrowser.Controller.Providers
{
public class DirectoryService : IDirectoryService
{
private readonly IFileSystem _fileSystem;
2021-12-24 18:28:27 +01:00
private readonly ConcurrentDictionary<string, FileSystemMetadata[]> _cache = new(StringComparer.Ordinal);
2018-12-28 00:27:57 +01:00
2021-12-24 18:28:27 +01:00
private readonly ConcurrentDictionary<string, FileSystemMetadata> _fileCache = new(StringComparer.Ordinal);
2018-12-28 00:27:57 +01:00
2021-12-24 18:28:27 +01:00
private readonly ConcurrentDictionary<string, List<string>> _filePathCache = new(StringComparer.Ordinal);
2018-12-28 00:27:57 +01:00
2019-09-10 22:37:53 +02:00
public DirectoryService(IFileSystem fileSystem)
2018-12-28 00:27:57 +01:00
{
_fileSystem = fileSystem;
}
public FileSystemMetadata[] GetFileSystemEntries(string path)
{
2021-12-19 10:27:57 +01:00
return _cache.GetOrAdd(path, static (p, fileSystem) => fileSystem.GetFileSystemEntries(p).ToArray(), _fileSystem);
2018-12-28 00:27:57 +01:00
}
public List<FileSystemMetadata> GetFiles(string path)
{
var list = new List<FileSystemMetadata>();
var items = GetFileSystemEntries(path);
2021-05-24 00:30:41 +02:00
for (var i = 0; i < items.Length; i++)
2018-12-28 00:27:57 +01:00
{
2021-05-24 00:30:41 +02:00
var item = items[i];
2018-12-28 00:27:57 +01:00
if (!item.IsDirectory)
{
list.Add(item);
}
}
2019-09-10 22:37:53 +02:00
2018-12-28 00:27:57 +01:00
return list;
}
2021-05-07 00:52:06 +02:00
public FileSystemMetadata? GetFile(string path)
2018-12-28 00:27:57 +01:00
{
2021-05-07 00:52:06 +02:00
if (!_fileCache.TryGetValue(path, out var result))
2018-12-28 00:27:57 +01:00
{
2021-05-07 00:52:06 +02:00
var file = _fileSystem.GetFileInfo(path);
2021-05-24 00:30:41 +02:00
if (file.Exists)
2021-05-07 00:52:06 +02:00
{
2021-05-24 00:30:41 +02:00
result = file;
2021-05-07 00:52:06 +02:00
_fileCache.TryAdd(path, result);
}
2018-12-28 00:27:57 +01:00
}
return result;
2018-12-28 00:27:57 +01:00
}
2020-02-23 10:53:51 +01:00
public IReadOnlyList<string> GetFilePaths(string path)
2021-05-31 13:55:54 +02:00
=> GetFilePaths(path, false);
2021-05-24 00:30:41 +02:00
public IReadOnlyList<string> GetFilePaths(string path, bool clearCache, bool sort = false)
2018-12-28 00:27:57 +01:00
{
if (clearCache)
2018-12-28 00:27:57 +01:00
{
2020-09-28 02:24:12 +02:00
_filePathCache.TryRemove(path, out _);
2018-12-28 00:27:57 +01:00
}
2021-12-19 10:27:57 +01:00
var filePaths = _filePathCache.GetOrAdd(path, static (p, fileSystem) => fileSystem.GetFilePaths(p).ToList(), _fileSystem);
2021-05-24 00:30:41 +02:00
if (sort)
{
filePaths.Sort();
}
return filePaths;
2018-12-28 00:27:57 +01:00
}
public bool IsAccessible(string path)
{
return _fileSystem.GetFileSystemEntryPaths(path).Any();
}
2018-12-28 00:27:57 +01:00
}
}