From e2294fd73edffb434458add329152978ea612d7a Mon Sep 17 00:00:00 2001 From: Christian Prando Date: Tue, 13 Apr 2021 08:58:57 -0600 Subject: [PATCH] UD-7239: Turns "append" and "finish" (header BD mode values) into constants. --- hubclient/rapid-scans-client.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/hubclient/rapid-scans-client.go b/hubclient/rapid-scans-client.go index 2cc5bd5..aa43f2c 100644 --- a/hubclient/rapid-scans-client.go +++ b/hubclient/rapid-scans-client.go @@ -24,6 +24,8 @@ const ( apiDeveloperScans = "/api/developer-scans" headerBdMode = "X-BD-MODE" headerBdDocumentCount = "X-BD-DOCUMENT-COUNT" + bdModeAppend = "append" + bdModeFinish = "finish" ) func (c *Client) StartRapidScan(bdioHeaderContent string) (error, string) { @@ -39,7 +41,7 @@ func (c *Client) StartRapidScan(bdioHeaderContent string) (error, string) { } func (c *Client) UploadBdioFiles(bdioUploadEndpoint string, bdioContents []string) error { - c.AddHeaderValue(headerBdMode, "append") + c.AddHeaderValue(headerBdMode, bdModeAppend) c.AddHeaderValue(headerBdDocumentCount, strconv.Itoa(len(bdioContents))) for _, bdioContent := range bdioContents { @@ -50,7 +52,7 @@ func (c *Client) UploadBdioFiles(bdioUploadEndpoint string, bdioContents []strin } } - c.SetHeaderValue(headerBdMode, "finish") + c.SetHeaderValue(headerBdMode, bdModeFinish) err := c.HttpPutJSON(bdioUploadEndpoint, "", hubapi.ContentTypeRapidScan, 202) if err != nil { log.Errorf("Error uploading bdio files.", err)