How to use

SessionTrackerImpl$SessionImpl

in

org.apache.zookeeper.server

@Override
public synchronized boolean trackSession(long id, int sessionTimeout) {
  boolean added = false;
  SessionImpl session = sessionsById.get(id);
  if (session == null){
    session = new SessionImpl(id, sessionTimeout);
  }
  // findbugs2.0.3 complains about get after put.
  // long term strategy would be use computeIfAbsent after JDK 1.8
  SessionImpl existedSession = sessionsById.putIfAbsent(id, session);
  if (existedSession != null) {
    session = existedSession;
  } else {
    added = true;
    LOG.debug("Adding session 0x" + Long.toHexString(id));
  }
  if (LOG.isTraceEnabled()) {
    String actionStr = added ? "Adding" : "Existing";
    ZooTrace.logTraceMessage(LOG, ZooTrace.SESSION_TRACE_MASK,
        "SessionTrackerImpl --- " + actionStr + " session 0x"
        + Long.toHexString(id) + " " + sessionTimeout);
  }
  updateSessionExpiry(session, sessionTimeout);
  return added;
}

synchronized public void setOwner(long id, Object owner) throws SessionExpiredException {
  SessionImpl session = sessionsById.get(id);
  if (session == null || session.isClosing()) {
    throw new KeeperException.SessionExpiredException();
  }
  session.owner = owner;
}

synchronized public boolean touchSession(long sessionId, int timeout) {
  SessionImpl s = sessionsById.get(sessionId);
  if (s == null) {
    logTraceTouchInvalidSession(sessionId, timeout);
    return false;
  }
  if (s.isClosing()) {
    logTraceTouchClosingSession(sessionId, timeout);
    return false;
  }
  updateSessionExpiry(s, timeout);
  return true;
}

public synchronized void checkSession(long sessionId, Object owner)
    throws KeeperException.SessionExpiredException,
    KeeperException.SessionMovedException,
    KeeperException.UnknownSessionException {
  LOG.debug("Checking session 0x" + Long.toHexString(sessionId));
  SessionImpl session = sessionsById.get(sessionId);
  if (session == null) {
    throw new KeeperException.UnknownSessionException();
  }
  if (session.isClosing()) {
    throw new KeeperException.SessionExpiredException();
  }
  if (session.owner == null) {
    session.owner = owner;
  } else if (session.owner != owner) {
    throw new KeeperException.SessionMovedException();
  }
}

synchronized public void checkSession(long sessionId, Object owner) throws KeeperException.SessionExpiredException, KeeperException.SessionMovedException {
  SessionImpl session = sessionsById.get(sessionId);
  if (session == null || session.isClosing()) {
    throw new KeeperException.SessionExpiredException();
  }
  if (session.owner == null) {
    session.owner = owner;
  } else if (session.owner != owner) {
    throw new KeeperException.SessionMovedException();
  }
}

  synchronized public void setOwner(long id, Object owner) throws SessionExpiredException {
    SessionImpl session = sessionsById.get(id);
    if (session == null || session.isClosing()) {
      throw new KeeperException.SessionExpiredException();
    }
    session.owner = owner;
  }
}

Assert.assertTrue("Session didn't expired", sessionImpl.isClosing());
Assert.assertFalse("Session didn't expired", sessionTrackerImpl
    .touchSession(sessionId, sessionTimeout));

synchronized public void addSession(long id, int sessionTimeout) {
  sessionsWithTimeout.put(id, sessionTimeout);
  if (sessionsById.get(id) == null) {
    SessionImpl s = new SessionImpl(id, sessionTimeout, 0);
    sessionsById.put(id, s);
    if (LOG.isTraceEnabled()) {
      ZooTrace.logTraceMessage(LOG, ZooTrace.SESSION_TRACE_MASK,
          "SessionTrackerImpl --- Adding session 0x"
          + Long.toHexString(id) + " " + sessionTimeout);
    }
  } else {
    if (LOG.isTraceEnabled()) {
      ZooTrace.logTraceMessage(LOG, ZooTrace.SESSION_TRACE_MASK,
          "SessionTrackerImpl --- Existing session 0x"
          + Long.toHexString(id) + " " + sessionTimeout);
    }
  }
  touchSession(id, sessionTimeout);
}

synchronized public boolean touchSession(long sessionId, int timeout) {
  if (LOG.isTraceEnabled()) {
    ZooTrace.logTraceMessage(LOG,
                 ZooTrace.CLIENT_PING_TRACE_MASK,
                 "SessionTrackerImpl --- Touch session: 0x"
        + Long.toHexString(sessionId) + " with timeout " + timeout);
  }
  SessionImpl s = sessionsById.get(sessionId);
  // Return false, if the session doesn't exists or marked as closing
  if (s == null || s.isClosing()) {
    return false;
  }
  long expireTime = roundToInterval(Time.currentElapsedTime() + timeout);
  if (s.tickTime >= expireTime) {
    // Nothing needs to be done
    return true;
  }
  SessionSet set = sessionSets.get(s.tickTime);
  if (set != null) {
    set.sessions.remove(s);
  }
  s.tickTime = expireTime;
  set = sessionSets.get(s.tickTime);
  if (set == null) {
    set = new SessionSet();
    sessionSets.put(expireTime, set);
  }
  set.sessions.add(s);
  return true;
}

synchronized public void addSession(long id, int sessionTimeout) {
  sessionsWithTimeout.put(id, sessionTimeout);
  if (sessionsById.get(id) == null) {
    SessionImpl s = new SessionImpl(id, sessionTimeout, 0);
    sessionsById.put(id, s);
    if (LOG.isTraceEnabled()) {
      ZooTrace.logTraceMessage(LOG, ZooTrace.SESSION_TRACE_MASK,
          "SessionTrackerImpl --- Adding session 0x"
          + Long.toHexString(id) + " " + sessionTimeout);
    }
  } else {
    if (LOG.isTraceEnabled()) {
      ZooTrace.logTraceMessage(LOG, ZooTrace.SESSION_TRACE_MASK,
          "SessionTrackerImpl --- Existing session 0x"
          + Long.toHexString(id) + " " + sessionTimeout);
    }
  }
  touchSession(id, sessionTimeout);
}