Method Pop

src/brpc/rpc_dump.cpp:322–360

}

SampledRequest* SampleIterator::Pop(butil::IOBuf& buf, bool* format_error) {
    char backing_buf[12];
    const char* p = (const char*)buf.fetch(backing_buf, sizeof(backing_buf));
    if (NULL == p) {  // buf.length() < sizeof(backing_buf)
        return NULL;
    }
    if (*(const uint32_t*)p != *(const uint32_t*)"PRPC") {
        LOG(ERROR) << "Unmatched magic string";
        *format_error = true;
        return NULL;
    }
    uint32_t body_size;
    uint32_t meta_size;
    butil::RawUnpacker(p + 4).unpack32(body_size).unpack32(meta_size);
    if (body_size > FLAGS_max_body_size) {
        LOG(ERROR) << "Too big body=" << body_size;
        *format_error = true;
        return NULL;
    } else if (buf.length() < sizeof(backing_buf) + body_size) {
        return NULL;
    }
    if (meta_size > body_size) {
        LOG(ERROR) << "meta_size=" << meta_size << " is bigger than body_size="
                   << body_size;
        *format_error = true;
        return NULL;
    }
    buf.pop_front(sizeof(backing_buf));
    butil::IOBuf meta_buf;
    buf.cutn(&meta_buf, meta_size);
    std::unique_ptr<SampledRequest> req(new SampledRequest);
    if (!ParsePbFromIOBuf(&req->meta, meta_buf)) {
        LOG(ERROR) << "Fail to parse RpcDumpMeta";
        *format_error = true;
        return NULL;
    }
    buf.cutn(&req->request, body_size - meta_size);
    return req.release();
}

#undef DUMPED_FILE_PREFIX