Calculate segments correctly

groongaGH-2104
abetomo · Nov 29, 2024 · 9dc4757 · 9dc4757
1 parent e254993
commit 9dc4757
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/pat.c b/lib/pat.c
@@ -6280,10 +6280,10 @@ grn_pat_defrag(grn_ctx *ctx, grn_pat *pat)
   for (size_t i = 0; i < n_targets; i++) {
     pat_node *node = pat_get(ctx, pat, target_ids[i]);
     uint32_t key_length = PAT_LEN(node);
-    uint32_t new_tail_segment =
-      (new_curr_key + key_length) >> W_OF_KEY_IN_A_SEGMENT;
-    if (new_curr_key >> W_OF_KEY_IN_A_SEGMENT != new_tail_segment) {
-      new_curr_key = new_tail_segment << W_OF_KEY_IN_A_SEGMENT;
+    uint32_t new_end_segment =
+      (new_curr_key + key_length - 1) >> W_OF_KEY_IN_A_SEGMENT;
+    if (new_curr_key >> W_OF_KEY_IN_A_SEGMENT != new_end_segment) {
+      new_curr_key = new_end_segment << W_OF_KEY_IN_A_SEGMENT;
     }
 
     uint8_t *key_position = NULL;