DotnetDumpService.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Diagnostics.NETCore.Client;
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Jobs;
 
namespace Tgstation.Server.Host.System
{
    sealed class DotnetDumpService : IDotnetDumpService
    {
        readonly ILogger<DotnetDumpService> logger;
 
        public DotnetDumpService(
            ILogger<DotnetDumpService> logger)
        {
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }
 
        public async ValueTask Dump(IProcess process, string outputFile, bool minidump, CancellationToken cancellationToken)
        {
            // need to use an extra timeout here because if the process is truly deadlocked. A cooperative dump will hang forever
            using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
 
            const int TimeoutMinutes = 5;
            cts.CancelAfter(TimeSpan.FromMinutes(TimeoutMinutes));
            cts.Token.Register(() =>
            {
                if (!cancellationToken.IsCancellationRequested)
                    logger.LogError("dotnet-dump timed out after {minutes} minutes!", TimeoutMinutes);
            });
 
            var pid = process.Id;
            logger.LogDebug("dotnet-dump requested for PID {pid}...", pid);
            var client = new DiagnosticsClient(pid);
            try
            {
                await client.WriteDumpAsync(
                    minidump
                        ? DumpType.Normal
                        : DumpType.Full,
                    outputFile,
                    false,
                    cts.Token);
            }
            catch (Exception ex)
            {
                throw new JobException(ErrorCode.GCoreFailure, ex);
            }
        }
    }
}