Если бы вы выбирали сообщения с наивысшим приоритетом, то очередь с простым приоритетом работала бы, в памяти или из базы данных. Но вы намереваетесь получить некоторые сообщения из всей очереди. Вот грубая реализация, чтобы дать вам идею. Вы можете экстраполировать его на массовый сценарий.
import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;
import java.util.Optional;
public final class NQueue
{
private final List<Deque<String>> queues = new ArrayList<>();
private final int[] poll;
private int currentIndex;
private int queueIndex;
private final int pollWidth;
public NQueue(int[] poll)
{
this.poll = new int[poll.length];
int sum = 0;
for (int i = 0; i < poll.length; i++)
{
int val = poll[i];
sum += val;
this.poll[i] = sum;
queues.add(new LinkedList<>());
}
this.pollWidth = sum;
}
public synchronized void addMessage(String msg, int priority)
{
queues.get(priority).add(msg);
}
public synchronized Optional<String> removeMessage()
{
for (int i = 0; i < this.pollWidth; i++)
{
updateQueueIndex();
String msg = queues.get(queueIndex).poll();
currentIndex++;
if (msg != null)
{
return Optional.of(msg);
}
}
return Optional.empty();
}
private void updateQueueIndex()
{
if (currentIndex == pollWidth)
{
currentIndex = 0;
queueIndex = 0;
}
else if (currentIndex == this.poll[this.queueIndex])
{
queueIndex++;
}
}
}
Пример клиента.
//in every run, poll 3 from priority 0, 2 from priority 1 and 1 from priority 2
NQueue nQueue = new NQueue(new int[]{3,2,1});
nQueue.addMessage("a",0);
nQueue.addMessage("b",1);
nQueue.addMessage("c",2);
nQueue.addMessage("d",1);
nQueue.addMessage("e",0);
nQueue.addMessage("f",1);
nQueue.addMessage("g",1);
for (int i = 0; i < 10; i++)
{
Optional<String> s = nQueue.removeMessage();
System.out.println("s = " + s);
}
печатает
s = Optional[a]
s = Optional[e]
s = Optional[b]
s = Optional[d]
s = Optional[c]
s = Optional[f]
s = Optional[g]
s = Optional.empty
s = Optional.empty
s = Optional.empty