Method HandleFileRsyncRequest

src/rsync_server.cc:159–223

}

void RsyncServerConn::HandleFileRsyncRequest(void* arg) {
  std::unique_ptr<RsyncServerTaskArg> task_arg(static_cast<RsyncServerTaskArg*>(arg));
  const std::shared_ptr<RsyncService::RsyncRequest> req = task_arg->req;
  std::shared_ptr<RsyncServerConn> conn = task_arg->conn;

  std::string db_name = req->db_name();
  std::string filename = req->file_req().filename();
  size_t offset = req->file_req().offset();
  size_t count = req->file_req().count();

  RsyncService::RsyncResponse response;
  response.set_reader_index(req->reader_index());
  response.set_code(RsyncService::kOk);
  response.set_type(RsyncService::kRsyncFile);
  response.set_db_name(db_name);
  /*
   * Since the slot field is written in protobuffer,
   * slot_id is set to the default value 0 for compatibility
   * with older versions, but slot_id is not used
   */
  response.set_slot_id(0);

  std::string snapshot_uuid;
  Status s = g_pika_server->GetDumpUUID(db_name, &snapshot_uuid);
  response.set_snapshot_uuid(snapshot_uuid);
  if (!s.ok()) {
    LOG(WARNING) << "rsyncserver get snapshotUUID failed";
    response.set_code(RsyncService::kErr);
    RsyncWriteResp(response, conn);
    return;
  }

  std::shared_ptr<DB> db = g_pika_server->GetDB(db_name);
  if (!db) {
   LOG(WARNING) << "cannot find db for db_name: " << db_name;
   response.set_code(RsyncService::kErr);
   RsyncWriteResp(response, conn);
  }

  const std::string filepath = db->bgsave_info().path + "/" + filename;
  char* buffer = new char[req->file_req().count() + 1];
  size_t bytes_read{0};
  std::string checksum = "";
  bool is_eof = false;
  std::shared_ptr<RsyncReader> reader = conn->readers_[req->reader_index()];
  s = reader->Read(filepath, offset, count, buffer,
                   &bytes_read, &checksum, &is_eof);
  if (!s.ok()) {
    response.set_code(RsyncService::kErr);
    RsyncWriteResp(response, conn);
    delete []buffer;
    return;
  }

  RsyncService::FileResponse* file_resp = response.mutable_file_resp();
  file_resp->set_data(buffer, bytes_read);
  file_resp->set_eof(is_eof);
  file_resp->set_checksum(checksum);