|
Packit |
63bb0d |
package azblob
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
import (
|
|
Packit |
63bb0d |
"errors"
|
|
Packit |
63bb0d |
"fmt"
|
|
Packit |
63bb0d |
"io"
|
|
Packit |
63bb0d |
"strconv"
|
|
Packit |
63bb0d |
)
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
// httpRange defines a range of bytes within an HTTP resource, starting at offset and
|
|
Packit |
63bb0d |
// ending at offset+count. A zero-value httpRange indicates the entire resource. An httpRange
|
|
Packit |
63bb0d |
// which has an offset but na zero value count indicates from the offset to the resource's end.
|
|
Packit |
63bb0d |
type httpRange struct {
|
|
Packit |
63bb0d |
offset int64
|
|
Packit |
63bb0d |
count int64
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
func (r httpRange) pointers() *string {
|
|
Packit |
63bb0d |
if r.offset == 0 && r.count == CountToEnd { // Do common case first for performance
|
|
Packit |
63bb0d |
return nil // No specified range
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
endOffset := "" // if count == CountToEnd (0)
|
|
Packit |
63bb0d |
if r.count > 0 {
|
|
Packit |
63bb0d |
endOffset = strconv.FormatInt((r.offset+r.count)-1, 10)
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
dataRange := fmt.Sprintf("bytes=%v-%s", r.offset, endOffset)
|
|
Packit |
63bb0d |
return &dataRange
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
func validateSeekableStreamAt0AndGetCount(body io.ReadSeeker) (int64, error) {
|
|
Packit |
63bb0d |
if body == nil { // nil body's are "logically" seekable to 0 and are 0 bytes long
|
|
Packit |
63bb0d |
return 0, nil
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
err := validateSeekableStreamAt0(body)
|
|
Packit |
63bb0d |
if err != nil {
|
|
Packit |
63bb0d |
return 0, err
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
count, err := body.Seek(0, io.SeekEnd)
|
|
Packit |
63bb0d |
if err != nil {
|
|
Packit |
63bb0d |
return 0, errors.New("body stream must be seekable")
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
body.Seek(0, io.SeekStart)
|
|
Packit |
63bb0d |
return count, nil
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
|
|
Packit |
63bb0d |
// return an error if body is not a valid seekable stream at 0
|
|
Packit |
63bb0d |
func validateSeekableStreamAt0(body io.ReadSeeker) error {
|
|
Packit |
63bb0d |
if body == nil { // nil body's are "logically" seekable to 0
|
|
Packit |
63bb0d |
return nil
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
if pos, err := body.Seek(0, io.SeekCurrent); pos != 0 || err != nil {
|
|
Packit |
63bb0d |
// Help detect programmer error
|
|
Packit |
63bb0d |
if err != nil {
|
|
Packit |
63bb0d |
return errors.New("body stream must be seekable")
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
return errors.New("body stream must be set to position 0")
|
|
Packit |
63bb0d |
}
|
|
Packit |
63bb0d |
return nil
|
|
Packit |
63bb0d |
}
|