@@ -6,12 +6,12 @@ namespace ServiceControl.Audit.Persistence.RavenDB.CustomChecks;
66using System . Threading . Tasks ;
77using 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