Mysql storage

src/BaGet.Core/Entities/IPackageContentsContext.cs

@@ -0,0 +1,13 @@
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+
+namespace BaGet.Core
+{
+ public interface IPackageContentsContext
+ {
+ DbSet<PackageContents> PackageContents { get; set; }
+
+ Task<int> SaveChangesAsync(CancellationToken cancellationToken);
+ }
+}

src/BaGet.Core/Entities/PackageContents.cs

@@ -0,0 +1,11 @@
+namespace BaGet.Core
+{
+ public class PackageContents
+ {
+ public int Key { get; set; }
+
+ public string Path { get; set; }
+
+ public byte[] Data { get; set; }
+ }
+}

src/BaGet.Core/Extensions/BaGetApplicationExtensions.cs

@@ -22,6 +22,12 @@ public static BaGetApplication AddFileStorage(this BaGetApplication app)
  return app;
  }
 
+ public static BaGetApplication AddMySqlStorage(this BaGetApplication app)
+ {
+ app.Services.TryAddTransient<IStorageService>(provider => provider.GetRequiredService<DatabaseStorageService>());
+ return app;
+ }
+
  public static BaGetApplication AddNullStorage(this BaGetApplication app)
  {
  app.Services.TryAddTransient<IStorageService>(provider => provider.GetRequiredService<NullStorageService>());

src/BaGet.Core/Extensions/DependencyInjectionExtensions.cs

@@ -100,6 +100,7 @@ private static void AddBaGetServices(this IServiceCollection services)
 
  services.TryAddTransient<DatabaseSearchService>();
  services.TryAddTransient<FileStorageService>();
+ services.TryAddTransient<DatabaseStorageService>();
  services.TryAddTransient<PackageService>();
  services.TryAddTransient<V2UpstreamClient>();
  services.TryAddTransient<V3UpstreamClient>();
@@ -133,6 +134,11 @@ private static void AddDefaultProviders(this IServiceCollection services)
  return provider.GetRequiredService<FileStorageService>();
  }
 
+ if (configuration.HasStorageType("mysql"))
+ {
+ return provider.GetRequiredService<DatabaseStorageService>();
+ }
+
  if (configuration.HasStorageType("null"))
  {
  return provider.GetRequiredService<NullStorageService>();

src/BaGet.Core/Storage/DatabaseStorageService.cs

@@ -0,0 +1,73 @@
+using System.Linq;
+using System;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+
+namespace BaGet.Core
+{
+ public class DatabaseStorageService : IStorageService
+ {
+ private readonly IPackageContentsContext _context;
+
+ public DatabaseStorageService(IPackageContentsContext context)
+ {
+ if (context == null) throw new ArgumentException(nameof(context));
+
+ _context = context;
+ }
+
+ public async Task DeleteAsync(string path, CancellationToken cancellationToken = default)
+ {
+ var contents = await _context.PackageContents.SingleOrDefaultAsync(p => p.Path == path, cancellationToken);
+ if (contents != null)
+ {
+ _context.PackageContents.Remove(contents);
+ await _context.SaveChangesAsync(cancellationToken);
+ }
+ }
+
+ public async Task<Stream> GetAsync(string path, CancellationToken cancellationToken = default)
+ {
+ var contents = await _context.PackageContents.SingleOrDefaultAsync(p => p.Path == path, cancellationToken);
+ if (contents == null)
+ {
+ throw new Exception($"PackageContents record not found for path: {path}");
+ }
+ var ms = new MemoryStream(contents.Data);
+ return ms;
+ }
+
+ public Task<Uri> GetDownloadUriAsync(string path, CancellationToken cancellationToken = default)
+ {
+ throw new NotImplementedException();
+ }
+
+ public async Task<StoragePutResult> PutAsync(string path, Stream content, string contentType, CancellationToken cancellationToken = default)
+ {
+ byte[] newData;
+ using (var binaryReader = new BinaryReader(content))
+ {
+ newData = binaryReader.ReadBytes((int)content.Length);
+ }
+
+ var existingContents = await _context.PackageContents.SingleOrDefaultAsync(p => p.Path == path, cancellationToken);
+ if (existingContents != null)
+ {
+ return existingContents.Data.SequenceEqual(newData)
+ ? StoragePutResult.AlreadyExists
+ : StoragePutResult.Success;
+ }
+
+ _context.PackageContents.Add(new PackageContents
+ {
+ Path = path,
+ Data = newData,
+ });
+ await _context.SaveChangesAsync(cancellationToken);
+
+ return StoragePutResult.Success;
+ }
+ }
+}