60 lines
2.8 KiB
C#
60 lines
2.8 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
using Umbraco.Cms.Core.DependencyInjection;
|
|
using Umbraco.Cms.Core.Models;
|
|
using Umbraco.Cms.Core.PublishedCache;
|
|
using Umbraco.Cms.Core.Scoping;
|
|
using Umbraco.Cms.Core.Services;
|
|
using Umbraco.Cms.Infrastructure.PublishedCache;
|
|
using Umbraco.Cms.Infrastructure.PublishedCache.Persistence;
|
|
using Umbraco.Core.Services;
|
|
|
|
namespace Umbraco.Extensions
|
|
{
|
|
/// <summary>
|
|
/// Extension methods for <see cref="IUmbracoBuilder"/> for the Umbraco's NuCache
|
|
/// </summary>
|
|
public static class UmbracoBuilderExtensions
|
|
{
|
|
/// <summary>
|
|
/// Adds Umbraco NuCache dependencies
|
|
/// </summary>
|
|
public static IUmbracoBuilder AddNuCache(this IUmbracoBuilder builder)
|
|
{
|
|
// register the NuCache database data source
|
|
builder.Services.TryAddSingleton<INuCacheContentRepository, NuCacheContentRepository>();
|
|
builder.Services.TryAddSingleton<INuCacheContentService, NuCacheContentService>();
|
|
builder.Services.TryAddSingleton<PublishedSnapshotServiceEventHandler>();
|
|
|
|
// register the NuCache published snapshot service
|
|
// must register default options, required in the service ctor
|
|
builder.Services.TryAddTransient(factory => new PublishedSnapshotServiceOptions());
|
|
builder.SetPublishedSnapshotService<PublishedSnapshotService>();
|
|
|
|
// Add as itself
|
|
builder.Services.TryAddSingleton<PublishedSnapshotService>();
|
|
builder.Services.TryAddSingleton<IPublishedSnapshotStatus, PublishedSnapshotStatus>();
|
|
|
|
// replace this service since we want to improve the content/media
|
|
// mapping lookups if we are using nucache.
|
|
// TODO: Gotta wonder how much this does actually improve perf? It's a lot of weird code to make this happen so hope it's worth it
|
|
builder.Services.AddUnique<IIdKeyMap>(factory =>
|
|
{
|
|
var idkSvc = new IdKeyMap(factory.GetRequiredService<IScopeProvider>());
|
|
if (factory.GetRequiredService<IPublishedSnapshotService>() is PublishedSnapshotService publishedSnapshotService)
|
|
{
|
|
idkSvc.SetMapper(UmbracoObjectTypes.Document, id => publishedSnapshotService.GetDocumentUid(id), uid => publishedSnapshotService.GetDocumentId(uid));
|
|
idkSvc.SetMapper(UmbracoObjectTypes.Media, id => publishedSnapshotService.GetMediaUid(id), uid => publishedSnapshotService.GetMediaId(uid));
|
|
}
|
|
|
|
return idkSvc;
|
|
});
|
|
|
|
// add the NuCache health check (hidden from type finder)
|
|
// TODO: no NuCache health check yet
|
|
// composition.HealthChecks().Add<NuCacheIntegrityHealthCheck>();
|
|
return builder;
|
|
}
|
|
}
|
|
}
|