mirror of
https://github.com/minio/minio.git
synced 2024-12-26 23:25:54 -05:00
266 lines
6.6 KiB
Go
266 lines
6.6 KiB
Go
|
package oss
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"hash/crc64"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/exec"
|
|||
|
|
"runtime"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Get User Agent
|
|||
|
|
// Go sdk相关信息，包括sdk版本，操作系统类型，GO版本
|
|||
|
|
var userAgent = func() string {
|
|||
|
|
sys := getSysInfo()
|
|||
|
|
return fmt.Sprintf("aliyun-sdk-go/%s (%s/%s/%s;%s)", Version, sys.name,
|
|||
|
|
sys.release, sys.machine, runtime.Version())
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
type sysInfo struct {
|
|||
|
|
name string // 操作系统名称windows/Linux
|
|||
|
|
release string // 操作系统版本 2.6.32-220.23.2.ali1089.el5.x86_64等
|
|||
|
|
machine string // 机器类型amd64/x86_64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Get system info
|
|||
|
|
// 获取操作系统信息、机器类型
|
|||
|
|
func getSysInfo() sysInfo {
|
|||
|
|
name := runtime.GOOS
|
|||
|
|
release := "-"
|
|||
|
|
machine := runtime.GOARCH
|
|||
|
|
if out, err := exec.Command("uname", "-s").CombinedOutput(); err == nil {
|
|||
|
|
name = string(bytes.TrimSpace(out))
|
|||
|
|
}
|
|||
|
|
if out, err := exec.Command("uname", "-r").CombinedOutput(); err == nil {
|
|||
|
|
release = string(bytes.TrimSpace(out))
|
|||
|
|
}
|
|||
|
|
if out, err := exec.Command("uname", "-m").CombinedOutput(); err == nil {
|
|||
|
|
machine = string(bytes.TrimSpace(out))
|
|||
|
|
}
|
|||
|
|
return sysInfo{name: name, release: release, machine: machine}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// unpackedRange
|
|||
|
|
type unpackedRange struct {
|
|||
|
|
hasStart bool // 是否指定了起点
|
|||
|
|
hasEnd bool // 是否指定了终点
|
|||
|
|
start int64 // 起点
|
|||
|
|
end int64 // 终点
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// invalid Range Error
|
|||
|
|
func invalidRangeError(r string) error {
|
|||
|
|
return fmt.Errorf("InvalidRange %s", r)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// parseRange parse various styles of range such as bytes=M-N
|
|||
|
|
func parseRange(normalizedRange string) (*unpackedRange, error) {
|
|||
|
|
var err error
|
|||
|
|
hasStart := false
|
|||
|
|
hasEnd := false
|
|||
|
|
var start int64
|
|||
|
|
var end int64
|
|||
|
|
|
|||
|
|
// bytes==M-N or ranges=M-N
|
|||
|
|
nrSlice := strings.Split(normalizedRange, "=")
|
|||
|
|
if len(nrSlice) != 2 || nrSlice[0] != "bytes" {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// bytes=M-N,X-Y
|
|||
|
|
rSlice := strings.Split(nrSlice[1], ",")
|
|||
|
|
rStr := rSlice[0]
|
|||
|
|
|
|||
|
|
if strings.HasSuffix(rStr, "-") { // M-
|
|||
|
|
startStr := rStr[:len(rStr)-1]
|
|||
|
|
start, err = strconv.ParseInt(startStr, 10, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
hasStart = true
|
|||
|
|
} else if strings.HasPrefix(rStr, "-") { // -N
|
|||
|
|
len := rStr[1:]
|
|||
|
|
end, err = strconv.ParseInt(len, 10, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
if end == 0 { // -0
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
hasEnd = true
|
|||
|
|
} else { // M-N
|
|||
|
|
valSlice := strings.Split(rStr, "-")
|
|||
|
|
if len(valSlice) != 2 {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
start, err = strconv.ParseInt(valSlice[0], 10, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
hasStart = true
|
|||
|
|
end, err = strconv.ParseInt(valSlice[1], 10, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, invalidRangeError(normalizedRange)
|
|||
|
|
}
|
|||
|
|
hasEnd = true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &unpackedRange{hasStart, hasEnd, start, end}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// adjustRange return adjusted range, adjust the range according to the length of the file
|
|||
|
|
func adjustRange(ur *unpackedRange, size int64) (start, end int64) {
|
|||
|
|
if ur == nil {
|
|||
|
|
return 0, size
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if ur.hasStart && ur.hasEnd {
|
|||
|
|
start = ur.start
|
|||
|
|
end = ur.end + 1
|
|||
|
|
if ur.start < 0 || ur.start >= size || ur.end > size || ur.start > ur.end {
|
|||
|
|
start = 0
|
|||
|
|
end = size
|
|||
|
|
}
|
|||
|
|
} else if ur.hasStart {
|
|||
|
|
start = ur.start
|
|||
|
|
end = size
|
|||
|
|
if ur.start < 0 || ur.start >= size {
|
|||
|
|
start = 0
|
|||
|
|
}
|
|||
|
|
} else if ur.hasEnd {
|
|||
|
|
start = size - ur.end
|
|||
|
|
end = size
|
|||
|
|
if ur.end < 0 || ur.end > size {
|
|||
|
|
start = 0
|
|||
|
|
end = size
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetNowSec returns Unix time, the number of seconds elapsed since January 1, 1970 UTC.
|
|||
|
|
// 获取当前时间，从UTC开始的秒数。
|
|||
|
|
func GetNowSec() int64 {
|
|||
|
|
return time.Now().Unix()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetNowNanoSec returns t as a Unix time, the number of nanoseconds elapsed
|
|||
|
|
// since January 1, 1970 UTC. The result is undefined if the Unix time
|
|||
|
|
// in nanoseconds cannot be represented by an int64. Note that this
|
|||
|
|
// means the result of calling UnixNano on the zero Time is undefined.
|
|||
|
|
// 获取当前时间，从UTC开始的纳秒。
|
|||
|
|
func GetNowNanoSec() int64 {
|
|||
|
|
return time.Now().UnixNano()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetNowGMT 获取当前时间，格式形如"Mon, 02 Jan 2006 15:04:05 GMT"，HTTP中使用的时间格式
|
|||
|
|
func GetNowGMT() string {
|
|||
|
|
return time.Now().UTC().Format(http.TimeFormat)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// FileChunk 文件片定义
|
|||
|
|
type FileChunk struct {
|
|||
|
|
Number int // 块序号
|
|||
|
|
Offset int64 // 块在文件中的偏移量
|
|||
|
|
Size int64 // 块大小
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SplitFileByPartNum Split big file to part by the num of part
|
|||
|
|
// 按指定的块数分割文件。返回值FileChunk为分割结果，error为nil时有效。
|
|||
|
|
func SplitFileByPartNum(fileName string, chunkNum int) ([]FileChunk, error) {
|
|||
|
|
if chunkNum <= 0 || chunkNum > 10000 {
|
|||
|
|
return nil, errors.New("chunkNum invalid")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
file, err := os.Open(fileName)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
stat, err := file.Stat()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if int64(chunkNum) > stat.Size() {
|
|||
|
|
return nil, errors.New("oss: chunkNum invalid")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var chunks []FileChunk
|
|||
|
|
var chunk = FileChunk{}
|
|||
|
|
var chunkN = (int64)(chunkNum)
|
|||
|
|
for i := int64(0); i < chunkN; i++ {
|
|||
|
|
chunk.Number = int(i + 1)
|
|||
|
|
chunk.Offset = i * (stat.Size() / chunkN)
|
|||
|
|
if i == chunkN-1 {
|
|||
|
|
chunk.Size = stat.Size()/chunkN + stat.Size()%chunkN
|
|||
|
|
} else {
|
|||
|
|
chunk.Size = stat.Size() / chunkN
|
|||
|
|
}
|
|||
|
|
chunks = append(chunks, chunk)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return chunks, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SplitFileByPartSize Split big file to part by the size of part
|
|||
|
|
// 按块大小分割文件。返回值FileChunk为分割结果，error为nil时有效。
|
|||
|
|
func SplitFileByPartSize(fileName string, chunkSize int64) ([]FileChunk, error) {
|
|||
|
|
if chunkSize <= 0 {
|
|||
|
|
return nil, errors.New("chunkSize invalid")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
file, err := os.Open(fileName)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
stat, err := file.Stat()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
var chunkN = stat.Size() / chunkSize
|
|||
|
|
if chunkN >= 10000 {
|
|||
|
|
return nil, errors.New("Too many parts, please increase part size.")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var chunks []FileChunk
|
|||
|
|
var chunk = FileChunk{}
|
|||
|
|
for i := int64(0); i < chunkN; i++ {
|
|||
|
|
chunk.Number = int(i + 1)
|
|||
|
|
chunk.Offset = i * chunkSize
|
|||
|
|
chunk.Size = chunkSize
|
|||
|
|
chunks = append(chunks, chunk)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if stat.Size()%chunkSize > 0 {
|
|||
|
|
chunk.Number = len(chunks) + 1
|
|||
|
|
chunk.Offset = int64(len(chunks)) * chunkSize
|
|||
|
|
chunk.Size = stat.Size() % chunkSize
|
|||
|
|
chunks = append(chunks, chunk)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return chunks, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetPartEnd 计算结束位置
|
|||
|
|
func GetPartEnd(begin int64, total int64, per int64) int64 {
|
|||
|
|
if begin+per > total {
|
|||
|
|
return total - 1
|
|||
|
|
}
|
|||
|
|
return begin + per - 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// crcTable returns the Table constructed from the specified polynomial
|
|||
|
|
var crcTable = func() *crc64.Table {
|
|||
|
|
return crc64.MakeTable(crc64.ECMA)
|
|||
|
|
}
|