Changed Shop interface to be more asynchronous.

Implemented changes in Props.

Implemented said changes in AdafruitShop.

Fixed and improved caching in AdafruitShop.
This commit is contained in:
Harrison Deng 2021-08-11 23:54:52 -05:00
parent ff080390f8
commit 8a1e5aca15
12 changed files with 145 additions and 76 deletions

View File

@ -11,7 +11,7 @@ using Props.Shop.Framework;
namespace Props.Shop.Adafruit namespace Props.Shop.Adafruit
{ {
public class AdafruitShop : IShop public class AdafruitShop : IShop, IDisposable
{ {
private string workspaceDir; private string workspaceDir;
private ILoggerFactory loggerFactory; private ILoggerFactory loggerFactory;
@ -34,20 +34,21 @@ namespace Props.Shop.Adafruit
false, false,
true true
); );
public void Initialize(string workspaceDir, ILoggerFactory loggerFactory) public async ValueTask Initialize(string workspaceDir, ILoggerFactory loggerFactory)
{ {
this.workspaceDir = workspaceDir; this.workspaceDir = workspaceDir;
this.loggerFactory = loggerFactory; this.loggerFactory = loggerFactory;
logger = loggerFactory.CreateLogger<AdafruitShop>(); logger = loggerFactory.CreateLogger<AdafruitShop>();
http = new HttpClient(); http = new HttpClient();
http.BaseAddress = new Uri("http://www.adafruit.com/api/"); http.BaseAddress = new Uri("http://www.adafruit.com/api/");
string configPath = Path.Combine(workspaceDir, Configuration.FILE_NAME);
try try
{ {
configuration = JsonSerializer.Deserialize<Configuration>(File.ReadAllText(Path.Combine(workspaceDir, Configuration.FILE_NAME))); configuration = JsonSerializer.Deserialize<Configuration>(File.ReadAllText(configPath));
} }
catch (JsonException) catch (JsonException e)
{ {
logger.LogWarning("Could not read JSON file."); logger.LogWarning("Could not read JSON file \"{0}\": {1}", configPath, e.Message);
} }
catch (ArgumentException) catch (ArgumentException)
{ {
@ -55,11 +56,11 @@ namespace Props.Shop.Adafruit
} }
catch (DirectoryNotFoundException) catch (DirectoryNotFoundException)
{ {
logger.LogWarning("Directory could not be found."); logger.LogWarning("Directory \"{0}\" could not be found.", Path.GetDirectoryName(configPath));
} }
catch (FileNotFoundException) catch (FileNotFoundException)
{ {
logger.LogWarning("File could not be found."); logger.LogWarning("File \"{0}\" could not be found.", configPath);
} }
finally finally
{ {
@ -70,17 +71,17 @@ namespace Props.Shop.Adafruit
} }
ProductListingCacheData listingData = null; ProductListingCacheData listingData = null;
string cachePath = Path.Combine(workspaceDir, ProductListingCacheData.FILE_NAME);
try try
{ {
listingData = JsonSerializer.Deserialize<ProductListingCacheData>(File.ReadAllText(Path.Combine(workspaceDir, ProductListingCacheData.FILE_NAME))); using (Stream fileStream = File.OpenRead(cachePath))
if (listingData.LastUpdatedUtc - DateTime.UtcNow > TimeSpan.FromMilliseconds(configuration.CacheLifespan))
{ {
listingData = null; listingData = await JsonSerializer.DeserializeAsync<ProductListingCacheData>(fileStream);
} }
} }
catch (JsonException) catch (JsonException e)
{ {
logger.LogWarning("Could not read JSON file."); logger.LogWarning("Could not read JSON file \"{0}\": {1}", cachePath, e.Message);
} }
catch (ArgumentException) catch (ArgumentException)
{ {
@ -88,11 +89,11 @@ namespace Props.Shop.Adafruit
} }
catch (DirectoryNotFoundException) catch (DirectoryNotFoundException)
{ {
logger.LogWarning("Directory could not be found."); logger.LogWarning("Directory \"{0}\" could not be found.", Path.GetDirectoryName(cachePath));
} }
catch (FileNotFoundException) catch (FileNotFoundException)
{ {
logger.LogWarning("File could not be found."); logger.LogWarning("File \"{0}\" could not be found.", cachePath);
} }
finally finally
{ {
@ -101,7 +102,7 @@ namespace Props.Shop.Adafruit
configuration = new Configuration(); configuration = new Configuration();
} }
} }
LiveProductListingManager productListingManager = new LiveProductListingManager(http, loggerFactory.CreateLogger<LiveProductListingManager>(), listingData?.ProductListings, configuration.MinDownloadInterval); LiveProductListingManager productListingManager = new LiveProductListingManager(http, loggerFactory.CreateLogger<LiveProductListingManager>(), listingData, configuration.MinDownloadInterval);
this.searchManager = new SearchManager(productListingManager, configuration.Similarity); this.searchManager = new SearchManager(productListingManager, configuration.Similarity);
productListingManager.StartUpdateTimer(delay: 0, configuration.CacheLifespan); productListingManager.StartUpdateTimer(delay: 0, configuration.CacheLifespan);
@ -117,6 +118,31 @@ namespace Props.Shop.Adafruit
return searchManager.Search(query); return searchManager.Search(query);
} }
public async ValueTask SaveData()
{
if (workspaceDir != null)
{
logger.LogDebug("Saving data in \"{0}\"...", workspaceDir);
await File.WriteAllTextAsync(Path.Combine(workspaceDir, Configuration.FILE_NAME), JsonSerializer.Serialize(configuration));
using (Stream fileStream = File.OpenWrite(Path.Combine(workspaceDir, ProductListingCacheData.FILE_NAME)))
{
await JsonSerializer.SerializeAsync(fileStream, new ProductListingCacheData(await searchManager.ProductListingManager.ProductListings));
}
logger.LogDebug("Completed saving data.");
}
}
public async ValueTask DisposeAsync()
{
Dispose(true);
await DisposeAsyncCore();
}
protected virtual async ValueTask DisposeAsyncCore()
{
await SaveData();
}
protected virtual void Dispose(bool disposing) protected virtual void Dispose(bool disposing)
{ {
if (!disposedValue) if (!disposedValue)
@ -126,6 +152,7 @@ namespace Props.Shop.Adafruit
http.Dispose(); http.Dispose();
searchManager.Dispose(); searchManager.Dispose();
} }
disposedValue = true; disposedValue = true;
} }
} }
@ -135,17 +162,5 @@ namespace Props.Shop.Adafruit
Dispose(disposing: true); Dispose(disposing: true);
GC.SuppressFinalize(this); GC.SuppressFinalize(this);
} }
public async ValueTask SaveData()
{
if (workspaceDir != null)
{
await File.WriteAllTextAsync(Path.Combine(workspaceDir, Configuration.FILE_NAME), JsonSerializer.Serialize(configuration));
await File.WriteAllTextAsync(
Path.Combine(workspaceDir, configuration.ProductListingCacheFileName),
JsonSerializer.Serialize(new ProductListingCacheData(await searchManager.ProductListingManager.ProductListings))
);
}
}
} }
} }

View File

@ -12,6 +12,8 @@ namespace Props.Shop.Adafruit.Api
public void StartUpdateTimer(int delay = 1000 * 60 * 5, int period = 1000 * 60 * 5); public void StartUpdateTimer(int delay = 1000 * 60 * 5, int period = 1000 * 60 * 5);
public void StopUpdateTimer(); public void StopUpdateTimer();
public DateTime? LastDownload { get; }
public Task<ProductListing> GetProductListingFromIdentifier(string url); public Task<ProductListing> GetProductListingFromIdentifier(string url);
} }
} }

View File

@ -5,6 +5,7 @@ using System.Net.Http;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
using Microsoft.Extensions.Logging; using Microsoft.Extensions.Logging;
using Props.Shop.Adafruit.Persistence;
using Props.Shop.Framework; using Props.Shop.Framework;
namespace Props.Shop.Adafruit.Api namespace Props.Shop.Adafruit.Api
@ -14,7 +15,7 @@ namespace Props.Shop.Adafruit.Api
private ILogger<LiveProductListingManager> logger; private ILogger<LiveProductListingManager> logger;
private bool disposedValue; private bool disposedValue;
private int minDownloadInterval; private int minDownloadInterval;
private DateTime? lastDownload; public DateTime? LastDownload { get; private set; }
private object refreshLock = new object(); private object refreshLock = new object();
private volatile Task<IReadOnlyDictionary<string, IList<ProductListing>>> productListingsTask; private volatile Task<IReadOnlyDictionary<string, IList<ProductListing>>> productListingsTask;
@ -25,12 +26,17 @@ namespace Props.Shop.Adafruit.Api
private HttpClient httpClient; private HttpClient httpClient;
private Timer updateTimer; private Timer updateTimer;
public LiveProductListingManager(HttpClient httpClient, ILogger<LiveProductListingManager> logger, IReadOnlyDictionary<string, IList<ProductListing>> productListings = null, int minDownloadInterval = 5 * 60 * 1000) public LiveProductListingManager(HttpClient httpClient, ILogger<LiveProductListingManager> logger, ProductListingCacheData productListingCacheData = null, int minDownloadInterval = 5 * 60 * 1000)
{ {
this.logger = logger; this.logger = logger;
this.minDownloadInterval = minDownloadInterval; this.minDownloadInterval = minDownloadInterval;
this.httpClient = httpClient; this.httpClient = httpClient;
productListingsTask = Task.FromResult(productListings); if (productListingCacheData != null)
{
productListingsTask = Task.FromResult(productListingCacheData.ProductListings);
LastDownload = productListingCacheData.LastUpdatedUtc;
logger.LogInformation("{0} Cached product listings loaded. Listing saved at {1}", productListingCacheData.ProductListings.Count, productListingCacheData.LastUpdatedUtc);
}
} }
public void RefreshProductListings() public void RefreshProductListings()
@ -38,9 +44,9 @@ namespace Props.Shop.Adafruit.Api
lock (refreshLock) lock (refreshLock)
{ {
if (disposedValue) throw new ObjectDisposedException("ProductListingManager"); if (disposedValue) throw new ObjectDisposedException("ProductListingManager");
if ((lastDownload != null && DateTime.UtcNow - lastDownload <= TimeSpan.FromMilliseconds(minDownloadInterval)) || (productListingsTask != null && !productListingsTask.IsCompleted)) return; if ((LastDownload != null && DateTime.UtcNow - LastDownload <= TimeSpan.FromMilliseconds(minDownloadInterval)) || (productListingsTask != null && !productListingsTask.IsCompleted)) return;
lastDownload = DateTime.UtcNow; LastDownload = DateTime.UtcNow;
logger.LogDebug("Refreshing listings ({0}).", lastDownload); logger.LogDebug("Refreshing listings ({0}).", LastDownload);
productListingsTask = DownloadListings(); productListingsTask = DownloadListings();
} }
} }

View File

@ -6,14 +6,12 @@ namespace Props.Shop.Adafruit.Persistence
public int MinDownloadInterval { get; set; } public int MinDownloadInterval { get; set; }
public int CacheLifespan { get; set; } public int CacheLifespan { get; set; }
public float Similarity { get; set; } public float Similarity { get; set; }
public string ProductListingCacheFileName { get; set; }
public Configuration() public Configuration()
{ {
MinDownloadInterval = 5 * 60 * 1000; MinDownloadInterval = 5 * 60 * 1000;
Similarity = 0.8f; Similarity = 0.8f;
CacheLifespan = 5 * 60 * 1000; CacheLifespan = 5 * 60 * 1000;
ProductListingCacheFileName = "ProductListings.json";
} }
} }
} }

View File

@ -7,7 +7,7 @@ namespace Props.Shop.Adafruit.Persistence
public class ProductListingCacheData public class ProductListingCacheData
{ {
public const string FILE_NAME = "Product-listing-cache.json"; public const string FILE_NAME = "Product-listing-cache.json";
public DateTime LastUpdatedUtc { get; private set; } public DateTime LastUpdatedUtc { get; set; }
public IReadOnlyDictionary<string, IList<ProductListing>> ProductListings { get; set; } public IReadOnlyDictionary<string, IList<ProductListing>> ProductListings { get; set; }
public ProductListingCacheData(IReadOnlyDictionary<string, IList<ProductListing>> productListings) public ProductListingCacheData(IReadOnlyDictionary<string, IList<ProductListing>> productListings)

View File

@ -7,7 +7,7 @@ using Microsoft.Extensions.Logging;
namespace Props.Shop.Framework namespace Props.Shop.Framework
{ {
public interface IShop : IDisposable public interface IShop : IAsyncDisposable
{ {
string ShopName { get; } string ShopName { get; }
string ShopDescription { get; } string ShopDescription { get; }
@ -17,8 +17,7 @@ namespace Props.Shop.Framework
public Task<ProductListing> GetProductFromIdentifier(string identifier); public Task<ProductListing> GetProductFromIdentifier(string identifier);
void Initialize(string workspaceDir, ILoggerFactory loggerFactory); ValueTask Initialize(string workspaceDir, ILoggerFactory loggerFactory);
ValueTask SaveData();
public SupportedFeatures SupportedFeatures { get; } public SupportedFeatures SupportedFeatures { get; }
} }
} }

View File

@ -12,7 +12,7 @@ namespace Props.Shop.Adafruit.Tests.Api
public class FakeProductListingManager : IProductListingManager public class FakeProductListingManager : IProductListingManager
{ {
private bool disposedValue; private bool disposedValue;
private DateTime? lastDownload; public DateTime? LastDownload { get; private set; }
private ProductListingsParser parser = new ProductListingsParser(); private ProductListingsParser parser = new ProductListingsParser();
private readonly ConcurrentDictionary<string, ProductListing> activeProductListingUrls = new ConcurrentDictionary<string, ProductListing>(); private readonly ConcurrentDictionary<string, ProductListing> activeProductListingUrls = new ConcurrentDictionary<string, ProductListing>();
@ -29,21 +29,27 @@ namespace Props.Shop.Adafruit.Tests.Api
public void RefreshProductListings() public void RefreshProductListings()
{ {
if (disposedValue) throw new ObjectDisposedException("ProductListingManager"); if (disposedValue) throw new ObjectDisposedException("ProductListingManager");
if ((lastDownload != null && DateTime.UtcNow - lastDownload <= TimeSpan.FromMilliseconds(5 * 60 * 1000)) || (ProductListings != null && !ProductListings.IsCompleted)) return; if ((LastDownload != null && DateTime.UtcNow - LastDownload <= TimeSpan.FromMilliseconds(5 * 60 * 1000)) || (ProductListings != null && !ProductListings.IsCompleted)) return;
ProductListings = DownloadListings(); ProductListings = DownloadListings();
} }
private Task<IReadOnlyDictionary<string, IList<ProductListing>>> DownloadListings() { private Task<IReadOnlyDictionary<string, IList<ProductListing>>> DownloadListings()
{
if (disposedValue) throw new ObjectDisposedException("ProductListingManager"); if (disposedValue) throw new ObjectDisposedException("ProductListingManager");
lastDownload = DateTime.UtcNow; LastDownload = DateTime.UtcNow;
parser.BuildProductListings(File.OpenRead("./Assets/products.json")); using (Stream stream = File.OpenRead("./Assets/products.json"))
{
parser.BuildProductListings(stream);
}
Dictionary<string, IList<ProductListing>> listingNames = new Dictionary<string, IList<ProductListing>>(); Dictionary<string, IList<ProductListing>> listingNames = new Dictionary<string, IList<ProductListing>>();
activeProductListingUrls.Clear(); activeProductListingUrls.Clear();
foreach (ProductListing product in parser.ProductListings) foreach (ProductListing product in parser.ProductListings)
{ {
activeProductListingUrls.TryAdd(product.URL, product); activeProductListingUrls.TryAdd(product.URL, product);
IList<ProductListing> sameProducts = listingNames.GetValueOrDefault(product.Name); IList<ProductListing> sameProducts = listingNames.GetValueOrDefault(product.Name);
if (sameProducts == null) { if (sameProducts == null)
{
sameProducts = new List<ProductListing>(); sameProducts = new List<ProductListing>();
listingNames.Add(product.Name, sameProducts); listingNames.Add(product.Name, sameProducts);
} }

View File

@ -0,0 +1,47 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using Props.Shop.Adafruit.Api;
using Props.Shop.Adafruit.Persistence;
using Props.Shop.Framework;
using Xunit;
namespace Props.Shop.Adafruit.Tests.Api
{
public class LiveProductListingManagerTest
{
[Fact]
public async Task CacheTest()
{
// TODO: Improve testability of caching system, IProductListingManager, and implement here.
//Given
ProductListingsParser parser = new ProductListingsParser();
using (Stream stream = File.OpenRead("./Assets/products.json"))
{
parser.BuildProductListings(stream);
}
Dictionary<string, IList<ProductListing>> listingNames = new Dictionary<string, IList<ProductListing>>();
foreach (ProductListing product in parser.ProductListings)
{
IList<ProductListing> sameProducts = listingNames.GetValueOrDefault(product.Name);
if (sameProducts == null)
{
sameProducts = new List<ProductListing>();
listingNames.Add(product.Name, sameProducts);
}
sameProducts.Add(product);
}
ProductListingCacheData cache = new ProductListingCacheData(listingNames);
await Task.Delay(500);
LiveProductListingManager mockLiveProductListingManager = new LiveProductListingManager(null, new Logger<LiveProductListingManager>(LoggerFactory.Create((builder) => builder.AddXUnit())), cache);
//When
mockLiveProductListingManager.RefreshProductListings();
//Then
Assert.True(cache.LastUpdatedUtc.Equals(mockLiveProductListingManager.LastDownload));
Assert.NotEmpty(await mockLiveProductListingManager.ProductListings);
}
}
}

View File

@ -7,10 +7,10 @@ using Props.Shop.Framework;
namespace Props.Services.Modules namespace Props.Services.Modules
{ {
public interface IShopManager : IDisposable public interface IShopManager : IAsyncDisposable
{ {
public IEnumerable<string> GetAllShopNames(); public ValueTask<IEnumerable<string>> GetAllShopNames();
public IShop GetShop(string name); public ValueTask<IShop> GetShop(string name);
public IEnumerable<IShop> GetAllShops(); public ValueTask<IEnumerable<IShop>> GetAllShops();
} }
} }

View File

@ -32,13 +32,13 @@ namespace Props.Services.Modules
metricsManager.RegisterSearchQuery(query); metricsManager.RegisterSearchQuery(query);
logger.LogDebug("Searching for \"{0}\".", query); logger.LogDebug("Searching for \"{0}\".", query);
foreach (string shopName in ShopManager.GetAllShopNames()) foreach (string shopName in await ShopManager.GetAllShopNames())
{ {
if (searchOutline.Enabled[shopName]) if (searchOutline.Enabled[shopName])
{ {
logger.LogDebug("Checking \"{0}\".", shopName); logger.LogDebug("Checking \"{0}\".", shopName);
int amount = 0; int amount = 0;
await foreach (ProductListing product in ShopManager.GetShop(shopName).Search(query, searchOutline.Filters)) await foreach (ProductListing product in (await ShopManager.GetShop(shopName)).Search(query, searchOutline.Filters))
{ {
if (searchOutline.Filters.Validate(product)) if (searchOutline.Filters.Validate(product))
{ {

View File

@ -19,12 +19,12 @@ namespace Props.Services.Modules
{ {
public class ModularShopManager : IShopManager public class ModularShopManager : IShopManager
{ {
private Task ShopLoadingTask;
private ILoggerFactory loggerFactory; private ILoggerFactory loggerFactory;
private ILogger<ModularShopManager> logger; private ILogger<ModularShopManager> logger;
private Dictionary<string, IShop> shops; private Dictionary<string, IShop> shops;
private ModulesOptions options; private ModulesOptions options;
private IConfiguration configuration; private IConfiguration configuration;
private bool disposedValue;
public ModularShopManager(IConfiguration configuration, ILogger<ModularShopManager> logger, ILoggerFactory loggerFactory) public ModularShopManager(IConfiguration configuration, ILogger<ModularShopManager> logger, ILoggerFactory loggerFactory)
{ {
@ -34,26 +34,29 @@ namespace Props.Services.Modules
options = configuration.GetSection(ModulesOptions.Modules).Get<ModulesOptions>(); options = configuration.GetSection(ModulesOptions.Modules).Get<ModulesOptions>();
Directory.CreateDirectory(options.ModuleDataDir); Directory.CreateDirectory(options.ModuleDataDir);
shops = new Dictionary<string, IShop>(); shops = new Dictionary<string, IShop>();
LoadShops(); ShopLoadingTask = LoadShops();
} }
public IEnumerable<string> GetAllShopNames() public async ValueTask<IEnumerable<string>> GetAllShopNames()
{ {
await ShopLoadingTask;
return shops.Keys; return shops.Keys;
} }
public IShop GetShop(string name) public async ValueTask<IShop> GetShop(string name)
{ {
await ShopLoadingTask;
return shops[name]; return shops[name];
} }
public IEnumerable<IShop> GetAllShops() public async ValueTask<IEnumerable<IShop>> GetAllShops()
{ {
await ShopLoadingTask;
return shops.Values; return shops.Values;
} }
public void LoadShops() private async Task LoadShops()
{ {
string shopsDir = options.ModulesDir; string shopsDir = options.ModulesDir;
string shopRegex = options.ShopRegex; string shopRegex = options.ShopRegex;
@ -88,7 +91,7 @@ namespace Props.Services.Modules
{ {
DirectoryInfo dataDir = Directory.CreateDirectory(Path.Combine(options.ModuleDataDir, file.Substring(file.IndexOf(Path.DirectorySeparatorChar) + 1))); DirectoryInfo dataDir = Directory.CreateDirectory(Path.Combine(options.ModuleDataDir, file.Substring(file.IndexOf(Path.DirectorySeparatorChar) + 1)));
logger.LogDebug("Checking data directory for \"{0}\" at \"{1}\"", Path.GetFileName(file), dataDir.FullName); logger.LogDebug("Checking data directory for \"{0}\" at \"{1}\"", Path.GetFileName(file), dataDir.FullName);
shop.Initialize(dataDir.FullName, loggerFactory); await shop.Initialize(dataDir.FullName, loggerFactory);
success += 1; success += 1;
if (!shops.TryAdd(shop.ShopName, shop)) if (!shops.TryAdd(shop.ShopName, shop))
{ {
@ -107,27 +110,20 @@ namespace Props.Services.Modules
} }
} }
protected virtual void Dispose(bool disposing) public async ValueTask DisposeAsync()
{ {
if (!disposedValue) logger.LogDebug("Disposing...");
{ await DisposeAsyncCore();
if (disposing) }
protected virtual async ValueTask DisposeAsyncCore()
{ {
await ShopLoadingTask;
foreach (string shopName in shops.Keys) foreach (string shopName in shops.Keys)
{ {
shops[shopName].SaveData().AsTask().Wait(); await shops[shopName].DisposeAsync();
shops[shopName].Dispose(); logger.LogDebug("Completed dispose task for \"{0}\".", shopName);
} }
}
disposedValue = true;
}
}
public void Dispose()
{
Dispose(disposing: true);
GC.SuppressFinalize(this);
} }
} }
} }

Binary file not shown.