57 lines
1.8 KiB
C#
57 lines
1.8 KiB
C#
using System;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using IServiceProvider = NzbDrone.Common.IServiceProvider;
|
|
|
|
namespace NzbDrone.Update.UpdateEngine
|
|
{
|
|
public interface ITerminateNzbDrone
|
|
{
|
|
void Terminate();
|
|
}
|
|
|
|
public class TerminateNzbDrone : ITerminateNzbDrone
|
|
{
|
|
private readonly IServiceProvider _serviceProvider;
|
|
private readonly IProcessProvider _processProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public TerminateNzbDrone(IServiceProvider serviceProvider, IProcessProvider processProvider, Logger logger)
|
|
{
|
|
_serviceProvider = serviceProvider;
|
|
_processProvider = processProvider;
|
|
_logger = logger;
|
|
}
|
|
|
|
public void Terminate()
|
|
{
|
|
_logger.Info("Stopping all running services");
|
|
|
|
if (_serviceProvider.ServiceExist(ServiceProvider.NZBDRONE_SERVICE_NAME)
|
|
&& _serviceProvider.IsServiceRunning(ServiceProvider.NZBDRONE_SERVICE_NAME))
|
|
{
|
|
try
|
|
{
|
|
_serviceProvider.Stop(ServiceProvider.NZBDRONE_SERVICE_NAME);
|
|
|
|
}
|
|
catch (InvalidOperationException e)
|
|
{
|
|
_logger.WarnException("couldn't stop service", e);
|
|
}
|
|
}
|
|
|
|
//TODO:Should be able to restart service if anything beyond this point fails
|
|
_logger.Info("Killing all running processes");
|
|
|
|
if (_processProvider.GetProcessByName(ProcessProvider.NzbDroneConsoleProcessName).Any())
|
|
{
|
|
_processProvider.KillAll(ProcessProvider.NzbDroneConsoleProcessName);
|
|
}
|
|
|
|
_processProvider.KillAll(ProcessProvider.NzbDroneProcessName);
|
|
|
|
}
|
|
}
|
|
} |