src/DurableSDK/Commands/WaitDurableTaskCommand.cs

//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//

#pragma warning disable 1591 // Missing XML comment for publicly visible type or member 'member'

namespace Microsoft.Azure.Functions.PowerShellWorker.Durable.Commands
{
    using System.Collections;
    using System.Management.Automation;
    using Microsoft.Azure.Functions.PowerShellWorker.Durable.Tasks;

    [Cmdlet("Wait", "DurableTask")]
    public class WaitDurableTaskCommand : PSCmdlet
    {
        [Parameter(Mandatory = true)]
        [ValidateNotNull]
        public DurableTask[] Task { get; set; }

        [Parameter]
        public SwitchParameter Any { get; set; }

        private readonly DurableTaskHandler _durableTaskHandler = new DurableTaskHandler();

        protected override void EndProcessing()
        {
            var privateData = (Hashtable)MyInvocation.MyCommand.Module.PrivateData;
            var context = (OrchestrationContext)privateData[SetFunctionInvocationContextCommand.ContextKey];

            if (Any.IsPresent)
            {
                _durableTaskHandler.WaitAny(Task, context, WriteObject);
            }
            else
            {
                _durableTaskHandler.WaitAll(Task, context, WriteObject);
            }
        }

        protected override void StopProcessing()
        {
            _durableTaskHandler.Stop();
        }
    }
}