Skip to content

Commit

Permalink
UD-7239: Turns "append" and "finish" (header BD mode values) into con…
Browse files Browse the repository at this point in the history
…stants.
  • Loading branch information
christianprando committed Apr 13, 2021
1 parent 52f66d0 commit e2294fd
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions hubclient/rapid-scans-client.go
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand All @@ -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 {
Expand All @@ -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)
Expand Down

0 comments on commit e2294fd

Please sign in to comment.