You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
691 B
41 lines
691 B
package api
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
// BytesFromRequest 读取http.Request.Body的内容到slice中
|
|
func BytesFromRequest(r *http.Request) (b []byte, err error) {
|
|
if r.ContentLength == 0 {
|
|
return
|
|
}
|
|
if r.ContentLength > 0 {
|
|
b = make([]byte, int(r.ContentLength))
|
|
_, err = io.ReadFull(r.Body, b)
|
|
return
|
|
}
|
|
return ioutil.ReadAll(r.Body)
|
|
}
|
|
|
|
func SeekerLen(s io.Seeker) (int64, error) {
|
|
|
|
curOffset, err := s.Seek(0, io.SeekCurrent)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
endOffset, err := s.Seek(0, io.SeekEnd)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
_, err = s.Seek(curOffset, io.SeekStart)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return endOffset - curOffset, nil
|
|
}
|