-
Notifications
You must be signed in to change notification settings - Fork 21
/
range_put.go
130 lines (112 loc) · 4.65 KB
/
range_put.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
package files
import (
"context"
"fmt"
"net/http"
"strings"
"github.com/Azure/go-autorest/autorest"
"github.com/Azure/go-autorest/autorest/azure"
"github.com/Azure/go-autorest/autorest/validation"
"github.com/tombuildsstuff/giovanni/storage/internal/endpoints"
)
type PutByteRangeInput struct {
StartBytes int64
EndBytes int64
// Content is the File Contents for the specified range
// which can be at most 4MB
Content []byte
}
// PutByteRange puts the specified Byte Range in the specified File.
func (client Client) PutByteRange(ctx context.Context, accountName, shareName, path, fileName string, input PutByteRangeInput) (result autorest.Response, err error) {
if accountName == "" {
return result, validation.NewError("files.Client", "PutByteRange", "`accountName` cannot be an empty string.")
}
if shareName == "" {
return result, validation.NewError("files.Client", "PutByteRange", "`shareName` cannot be an empty string.")
}
if strings.ToLower(shareName) != shareName {
return result, validation.NewError("files.Client", "PutByteRange", "`shareName` must be a lower-cased string.")
}
if fileName == "" {
return result, validation.NewError("files.Client", "PutByteRange", "`fileName` cannot be an empty string.")
}
if input.StartBytes < 0 {
return result, validation.NewError("files.Client", "PutByteRange", "`input.StartBytes` must be greater or equal to 0.")
}
if input.EndBytes <= 0 {
return result, validation.NewError("files.Client", "PutByteRange", "`input.EndBytes` must be greater than 0.")
}
expectedBytes := input.EndBytes - input.StartBytes
actualBytes := len(input.Content)
if expectedBytes != int64(actualBytes) {
return result, validation.NewError("files.Client", "PutByteRange", fmt.Sprintf("The specified byte-range (%d) didn't match the content size (%d).", expectedBytes, actualBytes))
}
if expectedBytes < (4 * 1024) {
return result, validation.NewError("files.Client", "PutByteRange", "Specified Byte Range must be at least 4KB.")
}
if expectedBytes > (4 * 1024 * 1024) {
return result, validation.NewError("files.Client", "PutByteRange", "Specified Byte Range must be at most 4MB.")
}
req, err := client.PutByteRangePreparer(ctx, accountName, shareName, path, fileName, input)
if err != nil {
err = autorest.NewErrorWithError(err, "files.Client", "PutByteRange", nil, "Failure preparing request")
return
}
resp, err := client.PutByteRangeSender(req)
if err != nil {
result = autorest.Response{Response: resp}
err = autorest.NewErrorWithError(err, "files.Client", "PutByteRange", resp, "Failure sending request")
return
}
result, err = client.PutByteRangeResponder(resp)
if err != nil {
err = autorest.NewErrorWithError(err, "files.Client", "PutByteRange", resp, "Failure responding to request")
return
}
return
}
// PutByteRangePreparer prepares the PutByteRange request.
func (client Client) PutByteRangePreparer(ctx context.Context, accountName, shareName, path, fileName string, input PutByteRangeInput) (*http.Request, error) {
if path != "" {
path = fmt.Sprintf("%s/", path)
}
pathParameters := map[string]interface{}{
"shareName": autorest.Encode("path", shareName),
"directory": autorest.Encode("path", path),
"fileName": autorest.Encode("path", fileName),
}
queryParameters := map[string]interface{}{
"comp": autorest.Encode("query", "range"),
}
headers := map[string]interface{}{
"x-ms-version": APIVersion,
"x-ms-write": "update",
"x-ms-range": fmt.Sprintf("bytes=%d-%d", input.StartBytes, input.EndBytes-1),
"Content-Length": int(len(input.Content)),
}
preparer := autorest.CreatePreparer(
autorest.AsPut(),
autorest.WithBaseURL(endpoints.GetFileEndpoint(client.BaseURI, accountName)),
autorest.WithPathParameters("/{shareName}/{directory}{fileName}", pathParameters),
autorest.WithHeaders(headers),
autorest.WithQueryParameters(queryParameters),
autorest.WithBytes(&input.Content))
return preparer.Prepare((&http.Request{}).WithContext(ctx))
}
// PutByteRangeSender sends the PutByteRange request. The method will close the
// http.Response Body if it receives an error.
func (client Client) PutByteRangeSender(req *http.Request) (*http.Response, error) {
return autorest.SendWithSender(client, req,
azure.DoRetryWithRegistration(client.Client))
}
// PutByteRangeResponder handles the response to the PutByteRange request. The method always
// closes the http.Response Body.
func (client Client) PutByteRangeResponder(resp *http.Response) (result autorest.Response, err error) {
err = autorest.Respond(
resp,
client.ByInspecting(),
azure.WithErrorUnlessStatusCode(http.StatusCreated),
autorest.ByClosing())
result = autorest.Response{Response: resp}
return
}