// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

namespace Microsoft.Build.BackEnd
{
    /// <summary>
    /// Packet sent from TaskHost to owning worker node for RequestCores/ReleaseCores operations.
    /// When <see cref="IsRelease"/> is false, this is a RequestCores call with <see cref="RequestedCores"/> cores requested.
    /// When <see cref="IsRelease"/> is true, this is a ReleaseCores call with <see cref="RequestedCores"/> cores to release.
    /// </summary>
    internal class TaskHostCoresRequest : INodePacket, ITaskHostCallbackPacket
    {
        private int _requestId;
        private int _requestedCores;
        private bool _isRelease;

        public TaskHostCoresRequest()
        {
        }

        public TaskHostCoresRequest(int requestedCores, bool isRelease)
        {
            _requestedCores = requestedCores;
            _isRelease = isRelease;
        }

        public NodePacketType Type => NodePacketType.TaskHostCoresRequest;

        public int RequestId
        {
            get => _requestId;
            set => _requestId = value;
        }

        /// <summary>
        /// The number of cores requested (for RequestCores) or cores to release (for ReleaseCores).
        /// </summary>
        public int RequestedCores => _requestedCores;

        /// <summary>
        /// True if this is a ReleaseCores operation, false if RequestCores.
        /// </summary>
        public bool IsRelease => _isRelease;

        public void Translate(ITranslator translator)
        {
            translator.Translate(ref _requestId);
            translator.Translate(ref _requestedCores);
            translator.Translate(ref _isRelease);
        }

        internal static INodePacket FactoryForDeserialization(ITranslator translator)
        {
            var packet = new TaskHostCoresRequest();
            packet.Translate(translator);
            return packet;
        }
    }
}