protected override void Dispose(bool disposing)
{
_taskTracker.Synchronize().GetAwaiter().GetResult();
_taskTracker.ShutdownAsync(_backgroundContext).Wait();
_taskTracker.Dispose();
base.Dispose(disposing);
}
/// <inheritdoc />
protected override async Task <BoolResult> ShutdownCoreAsync(OperationContext context)
{
if (_taskTracker != null)
{
await _taskTracker.Synchronize();
await _taskTracker.ShutdownAsync(context);
}
return(BoolResult.Success);
}
public async Task AddPinThrowsOnDisposedPinContext()
{
var context = new Context(Logger);
using (var taskTracker = new BackgroundTaskTracker(nameof(PinContextTests), context))
{
try
{
var pinContext = new PinContext(taskTracker, contentHashes => { });
pinContext.AddPin(ContentHash.Random());
await pinContext.DisposeAsync();
Action addPinAction = () => pinContext.AddPin(ContentHash.Random());
addPinAction.Should().Throw <ObjectDisposedException>();
}
finally
{
await taskTracker.ShutdownAsync(context);
}
}
}
public async Task DisposeCallsUnpinHashesAction()
{
var calledOnDispose = false;
var context = new Context(Logger);
using (var taskTracker = new BackgroundTaskTracker(nameof(PinContextTests), context))
{
try
{
using (new PinContext(taskTracker, contentHashes => { calledOnDispose = true; }))
{
}
}
finally
{
await taskTracker.ShutdownAsync(context);
}
}
Assert.True(calledOnDispose);
}
public async Task AddPinAccumulatesDuplicates()
{
var context = new Context(Logger);
using (var taskTracker = new BackgroundTaskTracker(nameof(PinContextTests), context))
{
try
{
ContentHash hash1 = ContentHash.Random();
ContentHash hash2 = ContentHash.Random();
var hashCounts = new ConcurrentDictionary <ContentHash, int>
{
[hash1] = 0,
[hash2] = 0
};
using (var pinContext = new PinContext(taskTracker, pinCounts =>
{
foreach (var pinCount in pinCounts)
{
hashCounts[pinCount.Key] = pinCount.Value;
}
}))
{
pinContext.AddPin(hash2);
pinContext.AddPin(hash1);
pinContext.AddPin(hash2);
}
hashCounts[hash1].Should().Be(1);
hashCounts[hash2].Should().Be(2);
}
finally
{
await taskTracker.ShutdownAsync(context);
}
}
}
/// <inheritdoc />
public Task <BoolResult> ShutdownAsync(Context context)
{
ShutdownStarted = true;
return(ShutdownCall <MemoizationStoreTracer> .RunAsync(Tracer.MemoizationStoreTracer, context, async() =>
{
Tracer.Debug(context, "IncorporateOnShutdown start");
Tracer.Debug(context, $"Incorporate fingerprints feature enabled:[{_fingerprintIncorporationEnabled}]");
Tracer.Debug(context, $"Total fingerprints to be incorporated:[{FingerprintTracker.Count}]");
Tracer.Debug(context, $"Max fingerprints per incorporate request(=chunk size):[{_maxFingerprintsPerIncorporateRequest}]");
Tracer.Debug(context, $"Max incorporate requests allowed in parallel:[{_maxDegreeOfParallelismForIncorporateRequests}]");
if (_fingerprintIncorporationEnabled)
{
// Incorporating all of the fingerprints for a build, in one request, to a single endpoint causes pain. Incorporation involves
// extending the lifetime of all fingerprints *and* content/s mapped to each fingerprint. Processing a large request payload
// results in, potentially, fanning out a massive number of "lifetime extend" requests to itemstore and blobstore, which can
// bring down the endpoint. Break this down into chunks so that multiple, load-balanced endpoints can share the burden.
List <StrongFingerprint> fingerprintsToBump = FingerprintTracker.StaleFingerprints.ToList();
Tracer.Debug(context, $"Total fingerprints to be sent in incorporation requeststo the service: {fingerprintsToBump.Count}");
List <List <StrongFingerprintAndExpiration> > chunks = fingerprintsToBump.Select(
strongFingerprint => new StrongFingerprintAndExpiration(strongFingerprint, FingerprintTracker.GenerateNewExpiration())
).GetPages(_maxFingerprintsPerIncorporateRequest).ToList();
Tracer.Debug(context, $"Total fingerprint incorporation requests to be issued(=number of fingerprint chunks):[{chunks.Count}]");
var incorporateBlock = new ActionBlock <IEnumerable <StrongFingerprintAndExpiration> >(
async chunk =>
{
await ContentHashListAdapter.IncorporateStrongFingerprints(
context,
CacheNamespace,
new IncorporateStrongFingerprintsRequest(chunk.ToList().AsReadOnly())
).ConfigureAwait(false);
},
new ExecutionDataflowBlockOptions {
MaxDegreeOfParallelism = _maxDegreeOfParallelismForIncorporateRequests
});
foreach (var chunk in chunks)
{
await incorporateBlock.SendAsync(chunk);
}
incorporateBlock.Complete();
await incorporateBlock.Completion.ConfigureAwait(false); // TODO: Gracefully handle exceptions so that the rest of shutdown can happen (bug 1365340)
Tracer.Debug(context, "IncorporateOnShutdown stop");
}
if (_taskTracker != null)
{
await _taskTracker.Synchronize().ConfigureAwait(false);
await _taskTracker.ShutdownAsync(context).ConfigureAwait(false);
}
var backingContentSessionTask = Task.Run(async() => await BackingContentSession.ShutdownAsync(context).ConfigureAwait(false));
var writeThroughContentSessionResult = WriteThroughContentSession != null
? await WriteThroughContentSession.ShutdownAsync(context).ConfigureAwait(false)
: BoolResult.Success;
var backingContentSessionResult = await backingContentSessionTask.ConfigureAwait(false);
BoolResult result;
if (backingContentSessionResult.Succeeded && writeThroughContentSessionResult.Succeeded)
{
if (_sealingErrorsToPrintOnShutdown.Any())
{
var sb = new StringBuilder();
sb.AppendLine("Error(s) during background sealing:");
foreach (var sealingError in _sealingErrorsToPrintOnShutdown)
{
sb.AppendLine($"[{sealingError}]");
}
if (_sealingErrorCount > MaxSealingErrorsToPrintOnShutdown)
{
sb.AppendLine($"See log for the other {MaxSealingErrorsToPrintOnShutdown - _sealingErrorCount} error(s).");
}
result = new BoolResult(sb.ToString());
}
else
{
result = BoolResult.Success;
}
}
else
{
var sb = new StringBuilder();
if (!backingContentSessionResult.Succeeded)
{
sb.Append($"Backing content session shutdown failed, error=[{backingContentSessionResult}]");
}
if (!writeThroughContentSessionResult.Succeeded)
{
sb.Append(sb.Length > 0 ? ", " : string.Empty);
sb.Append($"Write-through content session shutdown failed, error=[{writeThroughContentSessionResult}]");
}
result = new BoolResult(sb.ToString());
}
ShutdownCompleted = true;
return result;
}));
}
