add multiple index support for elastic search document (#47)

Signed-off-by: alihan.yalcin <alihan.yalcin@trendyol.com>
Trendyol · Dec 4, 2023 · af04586 · af04586
1 parent 7bf409f
commit af04586
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/elasticsearch/bulk/bulk.go b/elasticsearch/bulk/bulk.go
@@ -127,7 +127,7 @@ func (b *Bulk) AddActions(
 		value := getEsActionJSON(
 			action.ID,
 			action.Type,
-			b.collectionIndexMapping[collectionName],
+			b.getIndexName(collectionName, action.IndexName),
 			action.Routing,
 			action.Source,
 			b.typeName,
@@ -313,3 +313,16 @@ func joinErrors(body map[string]any) error {
 	}
 	return fmt.Errorf(sb.String())
 }
+
+func (b *Bulk) getIndexName(collectionName, actionIndexName string) string {
+	if actionIndexName != "" {
+		return actionIndexName
+	}
+
+	indexName := b.collectionIndexMapping[collectionName]
+	if indexName == "" {
+		panic(fmt.Sprintf("there is no index mapping for collection: %s on your configuration", collectionName))
+	}
+
+	return indexName
+}
diff --git a/elasticsearch/document/document.go b/elasticsearch/document/document.go
@@ -8,10 +8,11 @@ const (
 )
 
 type ESActionDocument struct {
-	Routing *string
-	Type    EsAction
-	Source  []byte
-	ID      []byte
+	Routing   *string
+	Type      EsAction
+	Source    []byte
+	ID        []byte
+	IndexName string
 }
 
 func NewDeleteAction(key []byte, routing *string) ESActionDocument {