59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
|
package http
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"storj.io/drpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ drpc.Transport = &httpTransport{}
|
||
|
|
|
||
|
|
type httpTransport struct {
|
||
|
|
c *http.Client
|
||
|
|
method string
|
||
|
|
endpoint string
|
||
|
|
ct string
|
||
|
|
closed bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTransport(c *http.Client, endpoint, method, ct string) *httpTransport {
|
||
|
|
if ct == "" {
|
||
|
|
ct = "application/drpc"
|
||
|
|
}
|
||
|
|
return &httpTransport{c: c, method: method, endpoint: endpoint, ct: ct}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *httpTransport) Read(buf []byte) (int, error) {
|
||
|
|
req, err := http.NewRequest(t.method, t.endpoint, nil)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
req.Header.Add("Content-Type", t.ct)
|
||
|
|
rsp, err := t.c.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
buf, err = ioutil.ReadAll(rsp.Body)
|
||
|
|
return len(buf), err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *httpTransport) Write(buf []byte) (int, error) {
|
||
|
|
req, err := http.NewRequest(t.method, t.endpoint, bytes.NewReader(buf))
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
req.Header.Add("Content-Type", t.ct)
|
||
|
|
_, err = t.c.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
return len(buf), err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *httpTransport) Close() error {
|
||
|
|
t.closed = true
|
||
|
|
return nil
|
||
|
|
}
|