using System;
using System.Threading;
using System.Threading.Tasks;
using MxGateway.Contracts.Proto;
using MxGateway.Worker.Sta;

namespace MxGateway.Worker.MxAccess;

public interface IWorkerRuntimeSession : IDisposable
{
    Task<WorkerReady> StartAsync(
        string sessionId,
        int workerProcessId,
        CancellationToken cancellationToken = default);

    Task<MxCommandReply> DispatchAsync(StaCommand command);

    WorkerRuntimeHeartbeatSnapshot CaptureHeartbeat();

    void RequestShutdown();

    Task<MxAccessShutdownResult> ShutdownGracefullyAsync(
        TimeSpan timeout,
        CancellationToken cancellationToken = default);
}