Method GetIdState

oneflow/core/framework/multi_client_session_context.cpp:209–219

}

IdState MultiClientSessionContext::GetIdState() {
  CHECK(Singleton<IDMgr>::Get() != nullptr);
  CHECK(Singleton<TaskStreamIndexManager>::Get() != nullptr);
  CHECK(Singleton<LazyJobBuildAndInferCtxMgr>::Get() != nullptr);
  IdState id_state;

  id_state.job_id_state_ = Singleton<LazyJobBuildAndInferCtxMgr>::Get()->GetJobIdCount();
  Singleton<IDMgr>::Get()->SaveIdAndTaskIndex(&id_state);
  Singleton<TaskStreamIndexManager>::Get()->GetTaskStreamIndex(&id_state.stream_index_state_);
  return id_state;
}

void MultiClientSessionContext::SetIdState(const IdState& id_state) {
  CHECK(Singleton<IDMgr>::Get() != nullptr);