queue with time stamped elements within a time period

后端 未结 3 2242
孤街浪徒
孤街浪徒 2021-02-20 14:29

I want to store in a queue, datastructure does not matter, only the elements that I have inserted within say last 5 minutes from current time. Anything older should get removed

3条回答
  •  慢半拍i
    慢半拍i (楼主)
    2021-02-20 15:01

    You can use a Priority Queue with timestamps as your keys. So that when you call Peek() you always get the oldest timestamp still in the queue. Then each time you go to query for the number of items inside your window size: you cleanup the items outside your window and return the number of items still in the Priority queue.

    For example:

    public class CountInWindow {
    
        /**
         * Adding a main just for testing 
         * @param args
         * @throws InterruptedException 
         */
        public static void main(String[] args) throws InterruptedException {
            System.out.println("test started");
            CountInWindow test = new CountInWindow(5000); //5 seconds for testing
            test.debug = true;
            test.insertTimeStamp(System.currentTimeMillis());
            Thread.sleep(100);//sleep 
            test.insertTimeStamp(System.currentTimeMillis());
            Thread.sleep(100);//sleep 
            test.insertTimeStamp(System.currentTimeMillis());
            Thread.sleep(100);//sleep 
            test.insertTimeStamp(System.currentTimeMillis());
            Thread.sleep(5040);//sleep 5 secs
            test.insertTimeStamp(System.currentTimeMillis());
            Thread.sleep(100);//sleep 
            test.insertTimeStamp(System.currentTimeMillis());
            System.out.println(test.getWindowCount()); //Should be 2 not 6.
            System.out.println("test done");
        }
    
        java.util.PriorityQueue window;
        public static final long FIVE_MINS_IN_MS = 300000l;
        public final long WINDOW_SIZE;
        public boolean debug = false;
    
        //Constructor which defaults to 5mins
        public CountInWindow(){
            WINDOW_SIZE = FIVE_MINS_IN_MS;
            window = new java.util.PriorityQueue();
        }
        //Constructor for any size window
        public CountInWindow(long windowSize){
            WINDOW_SIZE = windowSize;
            window = new java.util.PriorityQueue();
        }
        /**
         * Add a new timestamp to the window's queue
         * @param ts
         */
        public void insertTimeStamp(long ts){
            window.add(ts);
        }
        /**
         * Clean up items outside the window size and then return the count of times still in the window.
         * @return A count of timestamps still inside the 5 mins window.
         */
        public int getWindowCount(){
            long currTime = System.currentTimeMillis();
            //Clean out old Timestamps
            while((currTime - window.peek().longValue()) > WINDOW_SIZE){
                long drop = window.remove().longValue();
                if(debug)System.out.println("dropping item:" + drop);
            }
            return window.size();
        }
    }
    

提交回复
热议问题