com.avaje.ebeaninternal.server.cluster.mcast.OutgoingPacketsCache Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2009 Authors
 * 
 * This file is part of Ebean.
 * 
 * Ebean is free software; you can redistribute it and/or modify it 
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *  
 * Ebean is distributed in the hope that it will be useful, but 
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with Ebean; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA  
 */
package com.avaje.ebeaninternal.server.cluster.mcast;

import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import com.avaje.ebeaninternal.server.cluster.Packet;

/**
 * Cache of the outgoing packets.
 * 
 * These are held until we receive ACKs from the other members of the cluster to
 * say they have received the packets.
 * 
 * 
 * @author rbygrave
 * 
 */
public class OutgoingPacketsCache {

    private final Map packetMap = new TreeMap();

    public int size() {
        return packetMap.size();
    }

    public Packet getPacket(Long packetId) {
        return packetMap.get(packetId);
    }

    public String toString() {
        return packetMap.keySet().toString();
    }

    /**
     * Remove the packet when we give up trying to send it out.
     */
    public void remove(Packet packet) {
        packetMap.remove(packet.getPacketId());
    }

    public void registerPackets(List packets) {
        for (int i = 0; i < packets.size(); i++) {
            Packet p = packets.get(i);
            packetMap.put(p.getPacketId(), p);
        }
    }

    public int trimAll() {
        int size = packetMap.size();
        packetMap.clear();
        return size;
    }

    public void trimAcknowledgedMessages(long minAcked) {
        Iterator it = packetMap.keySet().iterator();
        while (it.hasNext()) {
            Long pktId = it.next();
            if (minAcked >= pktId.longValue()) {
                it.remove();
            }
        }
    }

}