This commit is contained in:
J. Nick Koston
2025-06-06 22:06:29 -05:00
parent a89c60cf10
commit 32be69f7f2
2 changed files with 5 additions and 5 deletions

View File

@@ -1638,9 +1638,9 @@ bool APIConnection::schedule_batch_() {
return true;
}
ProtoWriteBuffer APIConnection::allocate_single_message_buffer(uint32_t size) { return this->create_buffer(size); }
ProtoWriteBuffer APIConnection::allocate_single_message_buffer(uint16_t size) { return this->create_buffer(size); }
ProtoWriteBuffer APIConnection::allocate_batch_message_buffer(uint32_t size) {
ProtoWriteBuffer APIConnection::allocate_batch_message_buffer(uint16_t size) {
ProtoWriteBuffer result = this->prepare_message_buffer(size, this->batch_first_message_);
this->batch_first_message_ = false;
return result;

View File

@@ -246,7 +246,7 @@ class APIConnection : public APIServerConnection {
}
// Prepare buffer for next message in batch
ProtoWriteBuffer prepare_message_buffer(uint32_t message_size, bool is_first_message) {
ProtoWriteBuffer prepare_message_buffer(uint16_t message_size, bool is_first_message) {
// Get reference to shared buffer (it maintains state between batch messages)
std::vector<uint8_t> &shared_buf = this->parent_->get_shared_buffer_ref();
size_t current_size = shared_buf.size();
@@ -284,8 +284,8 @@ class APIConnection : public APIServerConnection {
std::string get_client_combined_info() const { return this->client_combined_info_; }
// Buffer allocator methods for batch processing
ProtoWriteBuffer allocate_single_message_buffer(uint32_t size);
ProtoWriteBuffer allocate_batch_message_buffer(uint32_t size);
ProtoWriteBuffer allocate_single_message_buffer(uint16_t size);
ProtoWriteBuffer allocate_batch_message_buffer(uint16_t size);
protected:
// Helper function to fill common entity fields