How to use

OfflinePersistentWorkerContext

in

org.keycloak.models.sessions.infinispan.initializer

@Override
public OfflinePersistentWorkerResult createFailedWorkerResult(OfflinePersistentLoaderContext loaderContext, OfflinePersistentWorkerContext workerContext) {
  return new OfflinePersistentWorkerResult(false, workerContext.getSegment(), workerContext.getWorkerId(), -1, FIRST_SESSION_ID);
}

@Override
public OfflinePersistentWorkerResult loadSessions(KeycloakSession session, OfflinePersistentLoaderContext loaderContext, OfflinePersistentWorkerContext ctx) {
  int first = ctx.getWorkerId() * sessionsPerSegment;
  log.tracef("Loading sessions for segment: %d", ctx.getSegment());
  UserSessionPersisterProvider persister = session.getProvider(UserSessionPersisterProvider.class);
  List<UserSessionModel> sessions = persister.loadUserSessions(first, sessionsPerSegment, true, ctx.getLastCreatedOn(), ctx.getLastSessionId());
  log.tracef("Sessions loaded from DB - segment: %d", ctx.getSegment());
  UserSessionModel lastSession = null;
  if (!sessions.isEmpty()) {
    lastSession = sessions.get(sessions.size() - 1);
    // Save to memory/infinispan
    session.sessions().importUserSessions(sessions, true);
  }
  int lastCreatedOn = lastSession==null ? Time.currentTime() + 100000 : lastSession.getStarted();
  String lastSessionId = lastSession==null ? FIRST_SESSION_ID : lastSession.getId();
  log.tracef("Sessions imported to infinispan - segment: %d, lastCreatedOn: %d, lastSessionId: %s", ctx.getSegment(), lastCreatedOn, lastSessionId);
  return new OfflinePersistentWorkerResult(true, ctx.getSegment(), ctx.getWorkerId(), lastCreatedOn, lastSessionId);
}

@Override
public OfflinePersistentWorkerContext computeWorkerContext(OfflinePersistentLoaderContext loaderCtx, int segment, int workerId, List<OfflinePersistentWorkerResult> previousResults) {
  int lastCreatedOn;
  String lastSessionId;
  if (previousResults.isEmpty()) {
    lastCreatedOn = 0;
    lastSessionId = FIRST_SESSION_ID;
  } else {
    OfflinePersistentWorkerResult lastResult = previousResults.get(previousResults.size() - 1);
    lastCreatedOn = lastResult.getLastCreatedOn();
    lastSessionId = lastResult.getLastSessionId();
  }
  // We know the last loaded session. New workers iteration will start from this place
  return new OfflinePersistentWorkerContext(segment, workerId, lastCreatedOn, lastSessionId);
}