Snapshot all Indices Additional Query Params

.gitignore

@@ -14,3 +14,5 @@
 tmp/*
 .vscode/settings.json
 .idea
+
+oryxBuildBinary

es.go

@@ -1174,6 +1174,34 @@ func (c *Client) SnapshotAllIndices(repository string, snapshot string) error {
 	return err
 }
 
+func (c *Client) SnapshotAllIndicesWithBodyParams(repository string, snapshot string, bodyParams map[string]interface{}) error {
+	if repository == "" {
+		return errors.New("empty string for repository is not allowed")
+	}
+
+	if snapshot == "" {
+		return errors.New("empty string for snapshot is not allowed")
+	}
+
+	if bodyParams == nil {
+		return errors.New("no body params provided, please use SnapshotAllIndices Function instead")
+	}
+
+	parsedJSON, parsingErr := json.Marshal(bodyParams)
+
+	if parsingErr != nil {
+		return parsingErr
+	}
+
+	agent := c.buildPutRequest(fmt.Sprintf("_snapshot/%s/%s", repository, snapshot)).
+		Set("Content-Type", "application/json").
+		Send(string(parsedJSON))
+
+	_, err := handleErrWithBytes(agent)
+
+	return err
+}
+
 // Restore an index or indices on the cluster
 //
 // Use case: You want to restore a particular index or indices onto your cluster with a new name.

es_test.go

@@ -1381,6 +1381,74 @@ func TestSnapshotAllIndices(t *testing.T) {
 	}
 }
 
+func TestSnapshotAllIndicesWithAdditionalParameters(t *testing.T) {
+	testSetup := &ServerSetup{
+		Method:   "PUT",
+		Path:     "/_snapshot/backup-repo/snapshot1",
+		Body:     `{"metadata":{"taken_because":"backup before upgrading","taken_by":"user123"}}`,
+		Response: `{"acknowledged": true }`,
+	}
+
+	host, port, ts := setupTestServers(t, []*ServerSetup{testSetup})
+	defer ts.Close()
+	client := NewClient(host, port)
+
+	bodyParams := map[string]interface{}{
+		"metadata": map[string]interface{}{
+			"taken_by":      "user123",
+			"taken_because": "backup before upgrading",
+		},
+	}
+
+	err := client.SnapshotAllIndicesWithBodyParams("backup-repo", "snapshot1", bodyParams)
+
+	if err != nil {
+		t.Fatalf("Got error taking snapshot: %s", err)
+	}
+}
+
+func TestSnapshotAllIndicesWithAdditionalParametersIncludeGlobalState(t *testing.T) {
+	testSetup := &ServerSetup{
+		Method:   "PUT",
+		Path:     "/_snapshot/backup-repo/snapshot1",
+		Body:     `{"include_global_state":true}`,
+		Response: `{"acknowledged": true }`,
+	}
+
+	host, port, ts := setupTestServers(t, []*ServerSetup{testSetup})
+	defer ts.Close()
+	client := NewClient(host, port)
+
+	bodyParams := map[string]interface{}{
+		"include_global_state": true,
+	}
+
+	err := client.SnapshotAllIndicesWithBodyParams("backup-repo", "snapshot1", bodyParams)
+
+	if err != nil {
+		t.Fatalf("Got error taking snapshot: %s", err)
+	}
+}
+
+func TestSnapshotAllIndicesWithAdditionalParametersErr(t *testing.T) {
+	testSetup := &ServerSetup{
+		Method:   "PUT",
+		Path:     "/_snapshot/backup-repo/snapshot1",
+		Body:     `{"metadata":{"taken_because":"backup before upgrading","taken_by":"user123"}}`,
+		Response: `{"acknowledged": true }`,
+	}
+
+	host, port, ts := setupTestServers(t, []*ServerSetup{testSetup})
+	defer ts.Close()
+	client := NewClient(host, port)
+
+	err := client.SnapshotAllIndicesWithBodyParams("backup-repo", "snapshot1", nil)
+
+	if err == nil {
+		t.Fatalf("should have thrown an error")
+	}
+}
+
 func TestRestoreSnapshotIndices_ErrorConditions(t *testing.T) {
 	tt := []struct {
 		Name        string