aspnetcore/src/Microsoft.AspNet.DataProtec.../Repositories/EphemeralXmlRepository.cs

61 lines
2.0 KiB
C#

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Xml.Linq;
using Microsoft.Extensions.Logging;
namespace Microsoft.AspNet.DataProtection.Repositories
{
/// <summary>
/// An ephemeral XML repository backed by process memory. This class must not be used for
/// anything other than dev scenarios as the keys will not be persisted to storage.
/// </summary>
internal class EphemeralXmlRepository : IXmlRepository
{
private readonly List<XElement> _storedElements = new List<XElement>();
public EphemeralXmlRepository(IServiceProvider services)
{
var logger = services?.GetLogger<EphemeralXmlRepository>();
logger?.UsingInmemoryRepository();
}
public virtual IReadOnlyCollection<XElement> GetAllElements()
{
// force complete enumeration under lock for thread safety
lock (_storedElements)
{
return GetAllElementsCore().ToList().AsReadOnly();
}
}
private IEnumerable<XElement> GetAllElementsCore()
{
// this method must be called under lock
foreach (XElement element in _storedElements)
{
yield return new XElement(element); // makes a deep copy so caller doesn't inadvertently modify it
}
}
public virtual void StoreElement(XElement element, string friendlyName)
{
if (element == null)
{
throw new ArgumentNullException(nameof(element));
}
XElement cloned = new XElement(element); // makes a deep copy so caller doesn't inadvertently modify it
// under lock for thread safety
lock (_storedElements)
{
_storedElements.Add(cloned);
}
}
}
}