mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-06 01:50:26 +08:00
156 lines
3.0 KiB
Go
156 lines
3.0 KiB
Go
package nodes
|
||
|
||
import (
|
||
"crypto/rand"
|
||
"fmt"
|
||
teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
|
||
"github.com/TeaOSLab/EdgeNode/internal/utils"
|
||
"io"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"sync/atomic"
|
||
)
|
||
|
||
// 分解Range
|
||
func httpRequestParseContentRange(rangeValue string) (result [][]int64, ok bool) {
|
||
// 参考RFC：https://tools.ietf.org/html/rfc7233
|
||
index := strings.Index(rangeValue, "=")
|
||
if index == -1 {
|
||
return
|
||
}
|
||
unit := rangeValue[:index]
|
||
if unit != "bytes" {
|
||
return
|
||
}
|
||
|
||
rangeSetString := rangeValue[index+1:]
|
||
if len(rangeSetString) == 0 {
|
||
ok = true
|
||
return
|
||
}
|
||
|
||
pieces := strings.Split(rangeSetString, ", ")
|
||
for _, piece := range pieces {
|
||
index := strings.Index(piece, "-")
|
||
if index == -1 {
|
||
return
|
||
}
|
||
first := piece[:index]
|
||
firstInt := int64(-1)
|
||
|
||
var err error
|
||
last := piece[index+1:]
|
||
var lastInt = int64(-1)
|
||
|
||
if len(first) > 0 {
|
||
firstInt, err = strconv.ParseInt(first, 10, 64)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
if len(last) > 0 {
|
||
lastInt, err = strconv.ParseInt(last, 10, 64)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if lastInt < firstInt {
|
||
return
|
||
}
|
||
}
|
||
} else {
|
||
if len(last) == 0 {
|
||
return
|
||
}
|
||
|
||
lastInt, err = strconv.ParseInt(last, 10, 64)
|
||
if err != nil {
|
||
return
|
||
}
|
||
lastInt = -lastInt
|
||
}
|
||
|
||
result = append(result, []int64{firstInt, lastInt})
|
||
}
|
||
|
||
ok = true
|
||
return
|
||
}
|
||
|
||
// 读取内容Range
|
||
func httpRequestReadRange(reader io.Reader, buf []byte, start int64, end int64, callback func(buf []byte, n int) error) (ok bool, err error) {
|
||
if start < 0 || end < 0 {
|
||
return
|
||
}
|
||
seeker, ok := reader.(io.Seeker)
|
||
if !ok {
|
||
return
|
||
}
|
||
_, err = seeker.Seek(start, io.SeekStart)
|
||
if err != nil {
|
||
return false, nil
|
||
}
|
||
|
||
offset := start
|
||
for {
|
||
n, err := reader.Read(buf)
|
||
if n > 0 {
|
||
offset += int64(n)
|
||
if end < offset {
|
||
err = callback(buf, n-int(offset-end-1))
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
return true, nil
|
||
} else {
|
||
err = callback(buf, n)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
}
|
||
}
|
||
|
||
if err != nil {
|
||
if err == io.EOF {
|
||
return true, nil
|
||
}
|
||
return false, err
|
||
}
|
||
}
|
||
}
|
||
|
||
// 生成boundary
|
||
// 仿照Golang自带的函数（multipart包）
|
||
func httpRequestGenBoundary() string {
|
||
var buf [30]byte
|
||
_, err := io.ReadFull(rand.Reader, buf[:])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return fmt.Sprintf("%x", buf[:])
|
||
}
|
||
|
||
// 判断状态是否为跳转
|
||
func httpStatusIsRedirect(statusCode int) bool {
|
||
return statusCode == http.StatusPermanentRedirect ||
|
||
statusCode == http.StatusTemporaryRedirect ||
|
||
statusCode == http.StatusMovedPermanently ||
|
||
statusCode == http.StatusSeeOther ||
|
||
statusCode == http.StatusFound
|
||
}
|
||
|
||
// 生成请求ID
|
||
var httpRequestTimestamp int64
|
||
var httpRequestId int32 = 1_000_000
|
||
|
||
func httpRequestNextId() string {
|
||
var unixTime = utils.UnixTimeMilli()
|
||
if unixTime > httpRequestTimestamp {
|
||
atomic.StoreInt32(&httpRequestId, 1_000_000)
|
||
httpRequestTimestamp = unixTime
|
||
}
|
||
|
||
// timestamp + requestId + nodeId
|
||
return strconv.FormatInt(unixTime, 10) + teaconst.NodeIdString + strconv.Itoa(int(atomic.AddInt32(&httpRequestId, 1)))
|
||
}
|