106 lines
2.9 KiB
C#
106 lines
2.9 KiB
C#
using Raven.Client.Documents;
|
|
using Raven.Client.Documents.Operations;
|
|
using Raven.Client.Documents.Queries;
|
|
using Raven.Client.Documents.Session;
|
|
using Raven.Client;
|
|
|
|
namespace zero.Raven;
|
|
|
|
public class ZeroStore : IZeroStore
|
|
{
|
|
public ZeroStore(IDocumentStore raven, IZeroOptions options) : base()
|
|
{
|
|
Options = options;
|
|
Raven = raven;
|
|
//Database = null;
|
|
}
|
|
|
|
protected IZeroOptions Options { get; set; }
|
|
protected Dictionary<string, IAsyncDocumentSession> ScopedSessions { get; set; } = new();
|
|
private const string NullDb = "__default__";
|
|
|
|
|
|
/// <inheritdoc />
|
|
public IDocumentStore Raven { get; private set; }
|
|
|
|
|
|
/// <inheritdoc />
|
|
public IAsyncDocumentSession Session(ZeroSessionResolution resolution = ZeroSessionResolution.Reuse, SessionOptions options = null)
|
|
{
|
|
options ??= new SessionOptions();
|
|
|
|
if (resolution == ZeroSessionResolution.Create)
|
|
{
|
|
return Raven.OpenAsyncSession(options);
|
|
}
|
|
|
|
if (!ScopedSessions.TryGetValue(Raven.Database, out IAsyncDocumentSession session))
|
|
{
|
|
session = Raven.OpenAsyncSession(options);
|
|
ScopedSessions.TryAdd(Raven.Database, session);
|
|
}
|
|
|
|
return session;
|
|
}
|
|
|
|
|
|
/// <inheritdoc />
|
|
public async Task PurgeAsync<T>(string querySuffix = null, Parameters parameters = null)
|
|
{
|
|
string collectionName = Raven.Conventions.FindCollectionName(typeof(T));
|
|
DeleteByQueryOperation operationQuery = new DeleteByQueryOperation(new IndexQuery()
|
|
{
|
|
Query = $"from {collectionName} c {querySuffix ?? string.Empty}",
|
|
QueryParameters = parameters
|
|
}, new QueryOperationOptions { AllowStale = true });
|
|
|
|
Operation operation = await Raven.Operations.SendAsync(operationQuery);
|
|
|
|
await operation.WaitForCompletionAsync();
|
|
}
|
|
|
|
|
|
/// <inheritdoc />
|
|
public void Purge<T>(string querySuffix = null, Parameters parameters = null)
|
|
{
|
|
string collectionName = Raven.Conventions.FindCollectionName(typeof(T));
|
|
DeleteByQueryOperation operationQuery = new DeleteByQueryOperation(new IndexQuery()
|
|
{
|
|
Query = $"from {collectionName} c {querySuffix ?? string.Empty}",
|
|
QueryParameters = parameters
|
|
}, new QueryOperationOptions { AllowStale = true });
|
|
|
|
Raven.Operations.Send(operationQuery);
|
|
}
|
|
}
|
|
|
|
|
|
public enum ZeroSessionResolution
|
|
{
|
|
Reuse = 0,
|
|
Create = 1
|
|
}
|
|
|
|
|
|
public interface IZeroStore
|
|
{
|
|
/// <summary>
|
|
/// Get underlying raven document store
|
|
/// </summary>
|
|
IDocumentStore Raven { get; }
|
|
|
|
/// <summary>
|
|
/// Use a specific session
|
|
/// </summary>
|
|
IAsyncDocumentSession Session(ZeroSessionResolution resolution = ZeroSessionResolution.Reuse, SessionOptions options = null);
|
|
|
|
/// <summary>
|
|
/// Purges a collection
|
|
/// </summary>
|
|
Task PurgeAsync<T>(string querySuffix = null, Parameters parameters = null);
|
|
|
|
/// <summary>
|
|
/// Purges a collection
|
|
/// </summary>
|
|
void Purge<T>(string querySuffix = null, Parameters parameters = null);
|
|
} |