logos-storage-nim-cs-dist-tests/Tools/AutoClient/Modes/FolderStoreMode.cs

using AutoClient.Modes.FolderStore;

namespace AutoClient.Modes
{
    public class FolderStoreMode
    {
        private readonly App app;
        private readonly CancellationTokenSource cts = new CancellationTokenSource();
        private Task checkTask = Task.CompletedTask;
        private readonly LoadBalancer loadBalancer;

        public FolderStoreMode(App app, LoadBalancer loadBalancer)
        {
            this.app = app;
            this.loadBalancer = loadBalancer;
        }

        public void Start()
        {
            checkTask = Task.Run(() =>
            {
                try
                {
                    var saver = new FolderSaver(app, loadBalancer);
                    while (!cts.IsCancellationRequested)
                    {
                        saver.Run(cts);
                    }
                }
                catch (Exception ex)
                {
                    app.Log.Error("Exception in FolderStoreMode: " + ex);
                    Environment.Exit(1);
                }
            });
        }

        public void Stop()
        {
            cts.Cancel();
            checkTask.Wait();
        }
    }
}