mirror of
https://github.com/eiblog/eiblog.git
synced 2026-02-17 03:42:26 +08:00
update vendor
This commit is contained in:
deepzz0
107
vendor/github.com/qiniu/x/rpc.v7/gob/gobrpc_client.go
generated
vendored
Normal file
107
vendor/github.com/qiniu/x/rpc.v7/gob/gobrpc_client.go
generated
vendored
Normal file
@@ -0,0 +1,107 @@
|
||||
package gob
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/gob"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
"qiniupkg.com/x/rpc.v7"
|
||||
|
||||
. "golang.org/x/net/context"
|
||||
)
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
func Register(value interface{}) {
|
||||
|
||||
gob.Register(value)
|
||||
}
|
||||
|
||||
func RegisterName(name string, value interface{}) {
|
||||
|
||||
gob.RegisterName(name, value)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
func ResponseError(resp *http.Response) (err error) {
|
||||
|
||||
e := &rpc.ErrorInfo{
|
||||
Reqid: resp.Header.Get("X-Reqid"),
|
||||
Code: resp.StatusCode,
|
||||
}
|
||||
if resp.StatusCode > 299 {
|
||||
e.Err = resp.Header.Get("X-Err")
|
||||
if errno := resp.Header.Get("X-Errno"); errno != "" {
|
||||
v, err2 := strconv.ParseInt(errno, 10, 32)
|
||||
if err2 != nil {
|
||||
e.Err = err2.Error()
|
||||
}
|
||||
e.Errno = int(v)
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func CallRet(ctx Context, ret interface{}, resp *http.Response) (err error) {
|
||||
|
||||
defer func() {
|
||||
io.Copy(ioutil.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
}()
|
||||
|
||||
if resp.StatusCode/100 == 2 {
|
||||
if ret != nil && resp.ContentLength != 0 {
|
||||
err = gob.NewDecoder(resp.Body).Decode(ret)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
if resp.StatusCode == 200 {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return ResponseError(resp)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
type Client struct {
|
||||
rpc.Client
|
||||
}
|
||||
|
||||
var (
|
||||
DefaultClient = Client{rpc.DefaultClient}
|
||||
)
|
||||
|
||||
func (r Client) Call(
|
||||
ctx Context, ret interface{}, method, url1 string) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWith(ctx, method, url1, "application/gob", nil, 0)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
func (r Client) CallWithGob(
|
||||
ctx Context, ret interface{}, method, url1 string, params interface{}) (err error) {
|
||||
|
||||
var b bytes.Buffer
|
||||
err = gob.NewEncoder(&b).Encode(params)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
resp, err := r.DoRequestWith(ctx, method, url1, "application/gob", &b, b.Len())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
344
vendor/github.com/qiniu/x/rpc.v7/rpc_client.go
generated
vendored
Normal file
344
vendor/github.com/qiniu/x/rpc.v7/rpc_client.go
generated
vendored
Normal file
@@ -0,0 +1,344 @@
|
||||
package rpc
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
|
||||
"qiniupkg.com/x/reqid.v7"
|
||||
|
||||
. "golang.org/x/net/context"
|
||||
)
|
||||
|
||||
var (
|
||||
UserAgent = "Golang qiniu/rpc package"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrInvalidRequestURL = errors.New("invalid request url")
|
||||
)
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
type Client struct {
|
||||
*http.Client
|
||||
}
|
||||
|
||||
var (
|
||||
DefaultClient = Client{&http.Client{Transport: http.DefaultTransport}}
|
||||
)
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
func newRequest(method, url1 string, body io.Reader) (req *http.Request, err error) {
|
||||
|
||||
var host string
|
||||
|
||||
// url1 = "-H <Host> http://<ip>[:<port>]/<path>"
|
||||
//
|
||||
if strings.HasPrefix(url1, "-H") {
|
||||
url2 := strings.TrimLeft(url1[2:], " \t")
|
||||
pos := strings.Index(url2, " ")
|
||||
if pos <= 0 {
|
||||
return nil, ErrInvalidRequestURL
|
||||
}
|
||||
host = url2[:pos]
|
||||
url1 = strings.TrimLeft(url2[pos+1:], " \t")
|
||||
}
|
||||
|
||||
req, err = http.NewRequest(method, url1, body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if host != "" {
|
||||
req.Host = host
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (r Client) DoRequest(ctx Context, method, url string) (resp *http.Response, err error) {
|
||||
|
||||
req, err := newRequest(method, url, nil)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return r.Do(ctx, req)
|
||||
}
|
||||
|
||||
func (r Client) DoRequestWith(
|
||||
ctx Context, method, url1 string,
|
||||
bodyType string, body io.Reader, bodyLength int) (resp *http.Response, err error) {
|
||||
|
||||
req, err := newRequest(method, url1, body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("Content-Type", bodyType)
|
||||
req.ContentLength = int64(bodyLength)
|
||||
return r.Do(ctx, req)
|
||||
}
|
||||
|
||||
func (r Client) DoRequestWith64(
|
||||
ctx Context, method, url1 string,
|
||||
bodyType string, body io.Reader, bodyLength int64) (resp *http.Response, err error) {
|
||||
|
||||
req, err := newRequest(method, url1, body)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
req.Header.Set("Content-Type", bodyType)
|
||||
req.ContentLength = bodyLength
|
||||
return r.Do(ctx, req)
|
||||
}
|
||||
|
||||
func (r Client) DoRequestWithForm(
|
||||
ctx Context, method, url1 string, data map[string][]string) (resp *http.Response, err error) {
|
||||
|
||||
msg := url.Values(data).Encode()
|
||||
if method == "GET" || method == "HEAD" || method == "DELETE" {
|
||||
if strings.ContainsRune(url1, '?') {
|
||||
url1 += "&"
|
||||
} else {
|
||||
url1 += "?"
|
||||
}
|
||||
return r.DoRequest(ctx, method, url1 + msg)
|
||||
}
|
||||
return r.DoRequestWith(
|
||||
ctx, method, url1, "application/x-www-form-urlencoded", strings.NewReader(msg), len(msg))
|
||||
}
|
||||
|
||||
func (r Client) DoRequestWithJson(
|
||||
ctx Context, method, url1 string, data interface{}) (resp *http.Response, err error) {
|
||||
|
||||
msg, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return r.DoRequestWith(
|
||||
ctx, method, url1, "application/json", bytes.NewReader(msg), len(msg))
|
||||
}
|
||||
|
||||
func (r Client) Do(ctx Context, req *http.Request) (resp *http.Response, err error) {
|
||||
|
||||
if ctx == nil {
|
||||
ctx = Background()
|
||||
}
|
||||
|
||||
if reqid, ok := reqid.FromContext(ctx); ok {
|
||||
req.Header.Set("X-Reqid", reqid)
|
||||
}
|
||||
|
||||
if _, ok := req.Header["User-Agent"]; !ok {
|
||||
req.Header.Set("User-Agent", UserAgent)
|
||||
}
|
||||
|
||||
transport := r.Transport // don't change r.Transport
|
||||
if transport == nil {
|
||||
transport = http.DefaultTransport
|
||||
}
|
||||
|
||||
// avoid cancel() is called before Do(req), but isn't accurate
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
err = ctx.Err()
|
||||
return
|
||||
default:
|
||||
}
|
||||
|
||||
if tr, ok := getRequestCanceler(transport); ok { // support CancelRequest
|
||||
reqC := make(chan bool, 1)
|
||||
go func() {
|
||||
resp, err = r.Client.Do(req)
|
||||
reqC <- true
|
||||
}()
|
||||
select {
|
||||
case <-reqC:
|
||||
case <-ctx.Done():
|
||||
tr.CancelRequest(req)
|
||||
<-reqC
|
||||
err = ctx.Err()
|
||||
}
|
||||
} else {
|
||||
resp, err = r.Client.Do(req)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
type ErrorInfo struct {
|
||||
Err string `json:"error,omitempty"`
|
||||
Key string `json:"key,omitempty"`
|
||||
Reqid string `json:"reqid,omitempty"`
|
||||
Errno int `json:"errno,omitempty"`
|
||||
Code int `json:"code"`
|
||||
}
|
||||
|
||||
func (r *ErrorInfo) ErrorDetail() string {
|
||||
|
||||
msg, _ := json.Marshal(r)
|
||||
return string(msg)
|
||||
}
|
||||
|
||||
func (r *ErrorInfo) Error() string {
|
||||
|
||||
return r.Err
|
||||
}
|
||||
|
||||
func (r *ErrorInfo) RpcError() (code, errno int, key, err string) {
|
||||
|
||||
return r.Code, r.Errno, r.Key, r.Err
|
||||
}
|
||||
|
||||
func (r *ErrorInfo) HttpCode() int {
|
||||
|
||||
return r.Code
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
func parseError(e *ErrorInfo, r io.Reader) {
|
||||
|
||||
body, err1 := ioutil.ReadAll(r)
|
||||
if err1 != nil {
|
||||
e.Err = err1.Error()
|
||||
return
|
||||
}
|
||||
|
||||
var ret struct {
|
||||
Err string `json:"error"`
|
||||
Key string `json:"key"`
|
||||
Errno int `json:"errno"`
|
||||
}
|
||||
if json.Unmarshal(body, &ret) == nil && ret.Err != "" {
|
||||
// qiniu error msg style returns here
|
||||
e.Err, e.Key, e.Errno = ret.Err, ret.Key, ret.Errno
|
||||
return
|
||||
}
|
||||
e.Err = string(body)
|
||||
}
|
||||
|
||||
func ResponseError(resp *http.Response) (err error) {
|
||||
|
||||
e := &ErrorInfo{
|
||||
Reqid: resp.Header.Get("X-Reqid"),
|
||||
Code: resp.StatusCode,
|
||||
}
|
||||
if resp.StatusCode > 299 {
|
||||
if resp.ContentLength != 0 {
|
||||
ct, ok := resp.Header["Content-Type"]
|
||||
if ok && strings.HasPrefix(ct[0], "application/json") {
|
||||
parseError(e, resp.Body)
|
||||
}
|
||||
}
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func CallRet(ctx Context, ret interface{}, resp *http.Response) (err error) {
|
||||
|
||||
defer func() {
|
||||
io.Copy(ioutil.Discard, resp.Body)
|
||||
resp.Body.Close()
|
||||
}()
|
||||
|
||||
if resp.StatusCode/100 == 2 {
|
||||
if ret != nil && resp.ContentLength != 0 {
|
||||
err = json.NewDecoder(resp.Body).Decode(ret)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
if resp.StatusCode == 200 {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return ResponseError(resp)
|
||||
}
|
||||
|
||||
func (r Client) CallWithForm(
|
||||
ctx Context, ret interface{}, method, url1 string, param map[string][]string) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWithForm(ctx, method, url1, param)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
func (r Client) CallWithJson(
|
||||
ctx Context, ret interface{}, method, url1 string, param interface{}) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWithJson(ctx, method, url1, param)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
func (r Client) CallWith(
|
||||
ctx Context, ret interface{}, method, url1, bodyType string, body io.Reader, bodyLength int) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWith(ctx, method, url1, bodyType, body, bodyLength)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
func (r Client) CallWith64(
|
||||
ctx Context, ret interface{}, method, url1, bodyType string, body io.Reader, bodyLength int64) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWith64(ctx, method, url1, bodyType, body, bodyLength)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
func (r Client) Call(
|
||||
ctx Context, ret interface{}, method, url1 string) (err error) {
|
||||
|
||||
resp, err := r.DoRequestWith(ctx, method, url1, "application/x-www-form-urlencoded", nil, 0)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return CallRet(ctx, ret, resp)
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
type requestCanceler interface {
|
||||
CancelRequest(req *http.Request)
|
||||
}
|
||||
|
||||
type nestedObjectGetter interface {
|
||||
NestedObject() interface{}
|
||||
}
|
||||
|
||||
func getRequestCanceler(tp http.RoundTripper) (rc requestCanceler, ok bool) {
|
||||
|
||||
if rc, ok = tp.(requestCanceler); ok {
|
||||
return
|
||||
}
|
||||
|
||||
p := interface{}(tp)
|
||||
for {
|
||||
getter, ok1 := p.(nestedObjectGetter)
|
||||
if !ok1 {
|
||||
return
|
||||
}
|
||||
p = getter.NestedObject()
|
||||
if rc, ok = p.(requestCanceler); ok {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
66
vendor/github.com/qiniu/x/rpc.v7/rpc_client_test.go
generated
vendored
Normal file
66
vendor/github.com/qiniu/x/rpc.v7/rpc_client_test.go
generated
vendored
Normal file
@@ -0,0 +1,66 @@
|
||||
package rpc
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
func TestNewRequest(t *testing.T) {
|
||||
|
||||
req, err := http.NewRequest("GET", "-H\t abc.com \thttp://127.0.0.1/foo/bar", nil)
|
||||
if err != nil {
|
||||
t.Fatal("http.NewRequest failed")
|
||||
}
|
||||
if req.Host != "" {
|
||||
t.Fatal(`http.NewRequest: req.Host != ""`)
|
||||
}
|
||||
|
||||
req, err = newRequest("GET", "-H\t abc.com \thttp://127.0.0.1/foo/bar", nil)
|
||||
if err != nil {
|
||||
t.Fatal("newRequest failed:", err)
|
||||
}
|
||||
|
||||
fmt.Println("Host:", req.Host, "path:", req.URL.Path, "url.host:", req.URL.Host)
|
||||
|
||||
if req.Host != "abc.com" || req.URL.Path != "/foo/bar" || req.URL.Host != "127.0.0.1" {
|
||||
t.Fatal(`req.Host != "abc.com" || req.URL.Path != "/foo/bar" || req.URL.Host != "127.0.0.1"`)
|
||||
}
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
type transport struct {
|
||||
a http.RoundTripper
|
||||
}
|
||||
|
||||
func (p *transport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
|
||||
return p.a.RoundTrip(req)
|
||||
}
|
||||
|
||||
func (p *transport) NestedObject() interface{} {
|
||||
return p.a
|
||||
}
|
||||
|
||||
func Test_getRequestCanceler(t *testing.T) {
|
||||
|
||||
p := &transport{a: http.DefaultTransport}
|
||||
if _, ok := getRequestCanceler(p); !ok {
|
||||
t.Fatal("getRequestCanceler failed")
|
||||
}
|
||||
|
||||
p2 := &transport{a: p}
|
||||
if _, ok := getRequestCanceler(p2); !ok {
|
||||
t.Fatal("getRequestCanceler(p2) failed")
|
||||
}
|
||||
|
||||
p3 := &transport{}
|
||||
if _, ok := getRequestCanceler(p3); ok {
|
||||
t.Fatal("getRequestCanceler(p3)?")
|
||||
}
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user