| 95 | } |
| 96 | |
| 97 | void Worker::DeregisterGraphAsync(const DeregisterGraphRequest* request, |
| 98 | DeregisterGraphResponse* response, |
| 99 | StatusCallback done) { |
| 100 | std::shared_ptr<WorkerSession> session; |
| 101 | Status s; |
| 102 | if (request->create_worker_session_called()) { |
| 103 | s = env_->session_mgr->WorkerSessionForSession(request->session_handle(), |
| 104 | &session); |
| 105 | } else { |
| 106 | session = env_->session_mgr->LegacySession(); |
| 107 | } |
| 108 | if (s.ok()) { |
| 109 | s = session->graph_mgr->Deregister(request->graph_handle()); |
| 110 | } |
| 111 | |
| 112 | done(s); |
| 113 | } |
| 114 | |
| 115 | void Worker::AbortStep(int64 step_id) { |
| 116 | Rendezvous* rendez = env_->rendezvous_mgr->Find(step_id); |
no test coverage detected