コード例 #1
0
  /**
   * The implementation for getPackets() function. This is a blocking API.
   *
   * @param sessionid
   * @return A list of packets associated with the session
   */
  @Override
  public List<ByteBuffer> getPackets(String sessionid) throws org.apache.thrift.TException {
    List<ByteBuffer> packets = new ArrayList<ByteBuffer>();
    int count = 0;

    while (!msgQueues.containsKey(sessionid) && count++ < 100) {
      log.debug("Queue for session {} doesn't exist yet.", sessionid);
      try {
        Thread.sleep(100); // Wait 100 ms to check again.
      } catch (InterruptedException e) {
        log.error(e.toString());
      }
    }

    if (count < 100) {
      SessionQueue pQueue = msgQueues.get(sessionid);
      BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
      // Block if queue is empty
      try {
        packets.add(queue.take());
        queue.drainTo(packets);
      } catch (InterruptedException e) {
        log.error(e.toString());
      }
    }

    return packets;
  }
コード例 #2
0
  /**
   * The implementation for terminateSession() function. It removes the session to queue
   * association.
   *
   * @param sessionid
   * @throws TException
   */
  @Override
  public void terminateSession(String sessionid) throws org.apache.thrift.TException {
    if (!msgQueues.containsKey(sessionid)) {
      return;
    }

    SessionQueue pQueue = msgQueues.get(sessionid);

    log.debug("terminateSession: SessionId: " + sessionid + "\n");
    String data = "FilterTimeout";
    ByteBuffer bb = ByteBuffer.wrap(data.getBytes());
    BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
    if (queue != null) {
      if (!queue.offer(bb)) {
        log.error("Failed to queue message for session: " + sessionid);
      }
      msgQueues.remove(sessionid);
    } else {
      log.error("queue for session {} is null", sessionid);
    }
  }
コード例 #3
0
  /**
   * The implementation for pushMessageSync() function.
   *
   * @param msg
   * @return 1 for success, 0 for failure
   * @throws TException
   */
  @Override
  public int pushMessageSync(Message msg) throws org.apache.thrift.TException {

    if (msg == null) {
      log.error("Error, pushMessageSync: empty message ");
      return 0;
    }

    List<String> sessionids = msg.getSessionIDs();
    for (String sid : sessionids) {
      SessionQueue pQueue = null;

      if (!msgQueues.containsKey(sid)) {
        pQueue = new SessionQueue();
        msgQueues.put(sid, pQueue);
      } else {
        pQueue = msgQueues.get(sid);
      }

      log.debug(
          "pushMessageSync: SessionId: " + sid + " Receive a message, " + msg.toString() + "\n");
      ByteBuffer bb = ByteBuffer.wrap(msg.getPacket().getData());
      // ByteBuffer dst = ByteBuffer.wrap(msg.getPacket().toString().getBytes());
      BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
      if (queue != null) {
        if (!queue.offer(bb)) {
          log.error("Failed to queue message for session: " + sid);
        } else {
          log.debug("insert a message to session: " + sid);
        }
      } else {
        log.error("queue for session {} is null", sid);
      }
    }

    return 1;
  }