Skip to content

Commit cfbf439

Browse files
Use the database configuration to get the server URL
1 parent 784581d commit cfbf439

File tree

1 file changed

+4
-10
lines changed

1 file changed

+4
-10
lines changed

src/ServiceControl.Audit.Persistence.RavenDB/CustomChecks/CheckDirtyMemory.cs

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@ namespace ServiceControl.Audit.Persistence.RavenDB.CustomChecks;
66
using System.Threading.Tasks;
77
using NServiceBus.CustomChecks;
88

9-
class CheckDirtyMemory(IRavenDocumentStoreProvider documentStoreProvider) : CustomCheck("ServiceControl.Audit database", "Dirty memory trends", TimeSpan.FromMinutes(5))
9+
class CheckDirtyMemory(DatabaseConfiguration databaseConfiguration) : CustomCheck("ServiceControl.Audit database", "Dirty memory trends", TimeSpan.FromMinutes(5))
1010
{
1111
readonly List<int> lastDirtyMemoryReads = [];
1212
public override async Task<CheckResult> PerformCheck(CancellationToken cancellationToken = default)
1313
{
14-
var retriever = await GetMemoryRetriever(cancellationToken);
14+
var retriever = await GetMemoryRetriever();
1515
var memoryInfo = await retriever.GetMemoryInformation(cancellationToken);
1616

1717
if (memoryInfo.IsHighDirty)
@@ -37,15 +37,9 @@ public override async Task<CheckResult> PerformCheck(CancellationToken cancellat
3737
}
3838

3939
MemoryInformationRetriever _retriever;
40-
async Task<MemoryInformationRetriever> GetMemoryRetriever(CancellationToken cancellationToken = default)
40+
async Task<MemoryInformationRetriever> GetMemoryRetriever()
4141
{
42-
if (_retriever == null)
43-
{
44-
var documentStore = await documentStoreProvider.GetDocumentStore(cancellationToken);
45-
var serverUrl = documentStore.Urls[0]; //TODO is there a better way to get the RavenDB server URL?
46-
_retriever = new MemoryInformationRetriever(serverUrl);
47-
}
48-
return _retriever;
42+
return _retriever ??= new MemoryInformationRetriever(databaseConfiguration.ServerConfiguration.ServerUrl);
4943
}
5044

5145
static TrendDirection AnalyzeTrendUsingRegression(List<int> values)

0 commit comments

Comments
 (0)