fix using routing in bulk operations (#66)

Trendyol · Apr 22, 2024 · f592cce · f592cce
1 parent 15f70c7
commit f592cce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/elasticsearch/bulk/bulk.go b/elasticsearch/bulk/bulk.go
@@ -144,7 +144,7 @@ func (b *Bulk) AddActions(
 			b.typeName,
 		)
 
-		key := getActionKey(action)
+		key := getActionKey(actions[i])
 		if batchIndex, ok := b.batchKeys[key]; ok {
 			b.batchByteSize += len(value) - len(b.batch[batchIndex].Bytes)
 			b.batch[batchIndex] = BatchItem{
@@ -386,6 +386,9 @@ func (b *Bulk) executeSinkResponseHandler(batchActions []*document.ESActionDocum
 }
 
 func getActionKey(action document.ESActionDocument) string {
+	if action.Routing != nil {
+		return fmt.Sprintf("%s:%s:%s", action.ID, action.IndexName, *action.Routing)
+	}
 	return fmt.Sprintf("%s:%s", action.ID, action.IndexName)
 }