feat: escape quotes on key. (#55)

* escaping key

elasticsearch/bulk/bulk.go

@@ -174,7 +174,7 @@ func getEsActionJSON(docID []byte, action document.EsAction, indexName string, r
 	}
 	meta = append(meta, helper.Byte(indexName)...)
 	meta = append(meta, idPrefix...)
-	meta = append(meta, docID...)
+	meta = append(meta, helper.EscapePredefinedBytes(docID)...)
 	if routing != nil {
 		meta = append(meta, routingPrefix...)
 		meta = append(meta, helper.Byte(*routing)...)

helper/escape.go

@@ -0,0 +1,21 @@
+package helper
+
+var (
+	EscapeBytes = []byte{
+		34, // quote
+	}
+	BackSlash byte = 92
+)
+
+func EscapePredefinedBytes(docID []byte) []byte {
+	newByteArr := make([]byte, 0, len(docID))
+	for _, byt := range docID {
+		for _, escapeByte := range EscapeBytes {
+			if escapeByte == byt {
+				newByteArr = append(newByteArr, BackSlash)
+			}
+		}
+		newByteArr = append(newByteArr, byt)
+	}
+	return newByteArr
+}

helper/escape_test.go

@@ -0,0 +1,29 @@
+package helper
+
+import (
+	"testing"
+)
+
+func TestEscapeQuote(t *testing.T) {
+	input := "12345-999\""
+	byteArr := []byte(input)
+	result := EscapePredefinedBytes(byteArr)
+	if len(result) == 10 || result[len(result)-1] != 34 {
+		t.Error("Expected backslash byte")
+	}
+}
+
+func TestDoNotEscapeQuote(t *testing.T) {
+	input := "12345-999"
+	byteArr := []byte(input)
+
+	result := EscapePredefinedBytes(byteArr)
+	if len(result) != 9 {
+		t.Error("Not expected any change related with length")
+	}
+	for _, b := range result {
+		if b == 34 {
+			t.Error("Not expected any backslash")
+		}
+	}
+}