-
Notifications
You must be signed in to change notification settings - Fork 7
/
SQLiteMessageJournal.cs
178 lines (162 loc) · 6.87 KB
/
SQLiteMessageJournal.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
// The MIT License (MIT)
//
// Copyright (c) 2017 Jesse Sweetland
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
using System;
using System.IO;
using System.Threading;
using System.Threading.Tasks;
using Platibus.Diagnostics;
using Platibus.Journaling;
using Platibus.SQL;
using Platibus.SQLite.Commands;
#if NET452 || NET461
using System.Configuration;
#endif
#if NETSTANDARD2_0
using Platibus.Config;
#endif
namespace Platibus.SQLite
{
/// <inheritdoc cref="SQLMessageJournal"/>
/// <inheritdoc cref="IDisposable"/>
/// <summary>
/// An <see cref="T:Platibus.IMessageQueueingService" /> implementation that stores queued
/// messages in a SQLite database
/// </summary>
public class SQLiteMessageJournal : SQLMessageJournal, IDisposable
{
protected readonly CancellationTokenSource CancellationTokenSource;
protected readonly ICommandExecutor CommandExecutor = new SynchronizingCommandExecutor();
private bool _disposed;
/// <inheritdoc />
/// <summary>
/// Initializes a new <see cref="T:Platibus.SQLite.SQLiteMessageQueueingService" />
/// </summary>
/// <param name="baseDirectory">The directory in which the SQLite database files will be
/// created</param>
/// <param name="diagnosticService">(Optional) The service through which diagnostic events
/// are reported and processed</param>
/// <remarks>
/// If a base directory is not specified then the base directory will default to a
/// directory named <c>platibus\queues</c> beneath the current app domain base
/// directory. If the base directory does not exist it will be created.
/// </remarks>
public SQLiteMessageJournal(DirectoryInfo baseDirectory, IDiagnosticService diagnosticService = null)
: base(InitConnectionProvider(baseDirectory, diagnosticService), new SQLiteMessageJournalCommandBuilders(), diagnosticService)
{
CancellationTokenSource = new CancellationTokenSource();
}
private static IDbConnectionProvider InitConnectionProvider(DirectoryInfo directory, IDiagnosticService diagnosticService)
{
if (directory == null)
{
var appDomainDirectory = AppDomain.CurrentDomain.BaseDirectory;
directory = new DirectoryInfo(Path.Combine(appDomainDirectory, "platibus", "journal"));
}
directory.Refresh();
if (!directory.Exists)
{
directory.Create();
}
var dbPath = Path.Combine(directory.FullName, "journal.db");
#if NET452 || NET461
var connectionStringSettings = new ConnectionStringSettings
{
Name = dbPath,
ConnectionString = "Data Source=" + dbPath + "; Version=3; BinaryGUID=False; DateTimeKind=Utc",
ProviderName = "System.Data.SQLite"
};
#endif
#if NETSTANDARD2_0
SQLiteProviderFactory.Register();
var connectionStringSettings = new ConnectionStringSettings
{
Name = dbPath,
ConnectionString = "Data Source=" + dbPath + "",
ProviderName = SQLiteProviderFactory.InvariantName
};
#endif
return new SingletonConnectionProvider(connectionStringSettings, diagnosticService);
}
/// <inheritdoc />
public override Task Append(Message message, MessageJournalCategory category,
CancellationToken cancellationToken = new CancellationToken())
{
CheckDisposed();
return CommandExecutor.Execute(
() => base.Append(message, category, cancellationToken),
cancellationToken);
}
/// <inheritdoc />
public override Task<MessageJournalReadResult> Read(MessageJournalPosition start, int count, MessageJournalFilter filter = null,
CancellationToken cancellationToken = new CancellationToken())
{
CheckDisposed();
return CommandExecutor.ExecuteRead(
() => base.Read(start, count, filter, cancellationToken),
cancellationToken);
}
/// <summary>
/// Throws <see cref="ObjectDisposedException"/> if this object has been disposed
/// </summary>
/// <exception cref="ObjectDisposedException">Thrown if this object has been disposed</exception>
protected virtual void CheckDisposed()
{
if (_disposed) throw new ObjectDisposedException(GetType().FullName);
}
/// <summary>
/// Finalizer to ensure that all resources are released
/// </summary>
~SQLiteMessageJournal()
{
Dispose(false);
}
/// <inheritdoc />
/// <summary>
/// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
/// </summary>
/// <filterpriority>2</filterpriority>
public void Dispose()
{
if (_disposed) return;
Dispose(true);
_disposed = true;
GC.SuppressFinalize(this);
}
/// <summary>
/// Called by the <see cref="Dispose()"/> method or by the finalizer to free held resources
/// </summary>
/// <param name="disposing">Indicates whether this method is called from the
/// <see cref="Dispose()"/> method (<c>true</c>) or from the finalizer (<c>false</c>)</param>
protected virtual void Dispose(bool disposing)
{
CancellationTokenSource.Cancel();
if (CommandExecutor is IDisposable disposableCommandExecutor)
{
disposableCommandExecutor.Dispose();
}
if (disposing)
{
CancellationTokenSource.Dispose();
}
}
}
}