/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
package org.apache.qpid.server.protocol.v0_8;
import org.apache.qpid.AMQException;
import org.apache.qpid.server.queue.QueueEntry;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
public class UnacknowledgedMessageMapImpl implements UnacknowledgedMessageMap
{
private final Object _lock = new Object();
private long _unackedSize;
private Map<Long, QueueEntry> _map;
private long _lastDeliveryTag;
private final int _prefetchLimit;
public UnacknowledgedMessageMapImpl(int prefetchLimit)
{
_prefetchLimit = prefetchLimit;
_map = new LinkedHashMap<Long, QueueEntry>(prefetchLimit);
}
public void collect(long deliveryTag, boolean multiple, Map<Long, QueueEntry> msgs)
{
if (multiple)
{
collect(deliveryTag, msgs);
}
else
{
final QueueEntry entry = get(deliveryTag);
if(entry != null)
{
msgs.put(deliveryTag, entry);
}
}
}
public void remove(Map<Long,QueueEntry> msgs)
{
synchronized (_lock)
{
for (Long deliveryTag : msgs.keySet())
{
remove(deliveryTag);
}
}
}
public QueueEntry remove(long deliveryTag)
{
synchronized (_lock)
{
QueueEntry message = _map.remove(deliveryTag);
if(message != null)
{
_unackedSize -= message.getMessage().getSize();
}
return message;
}
}
public void visit(Visitor visitor) throws AMQException
{
synchronized (_lock)
{
Set<Map.Entry<Long, QueueEntry>> currentEntries = _map.entrySet();
for (Map.Entry<Long, QueueEntry> entry : currentEntries)
{
visitor.callback(entry.getKey().longValue(), entry.getValue());
}
visitor.visitComplete();
}
}
public void add(long deliveryTag, QueueEntry message)
{
synchronized (_lock)
{
_map.put(deliveryTag, message);
_unackedSize += message.getMessage().getSize();
_lastDeliveryTag = deliveryTag;
}
}
public Collection<QueueEntry> cancelAllMessages()
{
synchronized (_lock)
{
Collection<QueueEntry> currentEntries = _map.values();
_map = new LinkedHashMap<Long, QueueEntry>(_prefetchLimit);
_unackedSize = 0l;
return currentEntries;
}
}
public int size()
{
synchronized (_lock)
{
return _map.size();
}
}
public void clear()
{
synchronized (_lock)
{
_map.clear();
_unackedSize = 0l;
}
}
public QueueEntry get(long key)
{
synchronized (_lock)
{
return _map.get(key);
}
}
public Set<Long> getDeliveryTags()
{
synchronized (_lock)
{
return _map.keySet();
}
}
public Collection<QueueEntry> acknowledge(long deliveryTag, boolean multiple)
{
Map<Long, QueueEntry> ackedMessageMap = new LinkedHashMap<Long,QueueEntry>();
collect(deliveryTag, multiple, ackedMessageMap);
remove(ackedMessageMap);
return ackedMessageMap.values();
}
private void collect(long key, Map<Long, QueueEntry> msgs)
{
synchronized (_lock)
{
for (Map.Entry<Long, QueueEntry> entry : _map.entrySet())
{
msgs.put(entry.getKey(),entry.getValue());
if (entry.getKey() == key)
{
break;
}
}
}
}
}