在优先阻塞队列中设置排序

Question

我有一个PriorityBlockingQueue，我在其中添加SocketHolder，如下所示。假设当前运行此代码的数据中心是abc。我们有三个数据中心abc、def 和pqr。

private static final PriorityBlockingQueue<SocketHolder> liveQueue = new PriorityBlockingQueue<>();

for (SocketHolder socket : listOfSockets) {
    liveQueue.add(socket);
}

现在我需要做的是：

• 我必须在liveQueue 上设置某种排序，以便所有SocketHolder 的数据中心为abc（这是当前运行代码的数据中心），它们应该位于顶部，这意味着当我们从这个liveQueue 中检索它，所有SocketHolder 和abc 数据中心应该首先出现，然后在所有其他SocketHolder 和其他dc 上。对于当前数据中心为 pqr 或 def 的情况也是如此。

下面是我的 SocketHolder 类，它实现了 Comparable 接口，但我很困惑我需要在我的 compareTo 方法中做什么，所以在 liveQueue 的顶部（在向它添加元素之后）我们拥有所有 @987654340 @ 其数据中心是当前数据中心，然后是我没有优先权的其他数据中心。

public final class SocketHolder implements Comparable<SocketHolder> {
  private final Socket socket;
  private final Datacenter dc;
  private final Context context;

  public SocketHolder(Socket socket, Datacenter dc, Context context) {
    super();
    this.socket = socket;
    this.dc = dc;
    this.context = context;
  }

  @Override
  public int compareTo(SocketHolder o) {
    // String currentDC = Utils.CURRENT_DATACENTER.get().name();
    return 0;
  }
}

Answer 1

实时数据中心将被分配优先级 1，其他数据中心优先级 2。队列将首先处理最低优先级。

public final class SocketHolder implements Comparable<SocketHolder> {
   private final Socket socket;
   private final Datacenter dc;
   private final Context context;
   private final int priority;

   public SocketHolder(Socket socket, Datacenter dc, Context context) {
      super();
      this.socket = socket;
      this.dc = dc;
      this.context = context;
      this.priority = this.dc.getName().equals(Utils.CURRENT_DATACENTER.get().name()) ? 1 : 2;
   }

   @Override
   public int compareTo(SocketHolder o) {
       return Integer.compare(this.priority, o.priority);
   }
}