WorkerQueue.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;


namespace kinetica.Utils;

internal sealed class WorkerQueue<T>
    {
        public System.Uri url { get; private set; }
        private readonly int capacity;
        private List<T> queue;


        public WorkerQueue( System.Uri url )
        {
            this.url = url;
            this.capacity = 1;

            queue = [];
        }  // end constructor WorkerQueue<T>



        public WorkerQueue(System.Uri url, int capacity, bool hasPrimaryKey, bool updateOnExistingPk)
        {
            this.url = url;
            this.capacity = capacity;

            queue = [];

            // Note: hasPrimaryKey and updateOnExistingPk parameters are reserved for future
            // primary key deduplication functionality but not currently implemented.
        }  // end constructor WorkerQueue<T>



        public IList<T> flush()
        {
            IList<T> old_queue = this.queue;
            queue = new List<T>(this.capacity);

            return old_queue;
        }  // end flush



        public IList<T>? insert(T record, RecordKey key)
        {
            queue.Add(record);
            // If the queue is full, then flush and return the 'old' queue
            if (queue.Count == capacity)
                return flush();
            else // no records to return
                return null;
        }  // end insert
    }  // end class WorkerQueue