diff options
Diffstat (limited to 'ext/kv/lib.rs')
| -rw-r--r-- | ext/kv/lib.rs | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/ext/kv/lib.rs b/ext/kv/lib.rs index d357a2927..c0091d75d 100644 --- a/ext/kv/lib.rs +++ b/ext/kv/lib.rs @@ -60,7 +60,7 @@ const MAX_READ_KEY_SIZE_BYTES: usize = MAX_WRITE_KEY_SIZE_BYTES + 1; const MAX_VALUE_SIZE_BYTES: usize = 65536; const MAX_READ_RANGES: usize = 10; const MAX_READ_ENTRIES: usize = 1000; -const MAX_CHECKS: usize = 10; +const MAX_CHECKS: usize = 100; const MAX_MUTATIONS: usize = 1000; const MAX_TOTAL_MUTATION_SIZE_BYTES: usize = 800 * 1024; const MAX_TOTAL_KEY_SIZE_BYTES: usize = 80 * 1024; @@ -669,13 +669,16 @@ where return Err(type_error("key cannot be empty")); } - let checked_size = check_write_key_size(key)?; - total_payload_size += checked_size; - total_key_size += checked_size; + total_payload_size += check_write_key_size(key)?; } - for value in mutations.iter().flat_map(|m| m.kind.value()) { - total_payload_size += check_value_size(value)?; + for (key, value) in mutations + .iter() + .flat_map(|m| m.kind.value().map(|x| (&m.key, x))) + { + let key_size = check_write_key_size(key)?; + total_payload_size += check_value_size(value)? + key_size; + total_key_size += key_size; } for enqueue in &enqueues { |