Use Request.WithContext

pull/19/head
Yasuhiro Matsumoto 2017-04-17 12:25:20 +09:00
parent 5e84b57bf3
commit 6fe43e545a
1 changed files with 12 additions and 29 deletions

View File

@ -24,23 +24,6 @@ type Client struct {
config *Config
}
func httpDo(ctx context.Context, req *http.Request, f func(*http.Response, error) error) error {
tr := &http.Transport{}
client := &http.Client{Transport: tr}
c := make(chan error, 1)
go func() {
c <- f(client.Do(req))
}()
select {
case <-ctx.Done():
tr.CancelRequest(req)
<-c
return ctx.Err()
case err := <-c:
return err
}
}
func (c *Client) doAPI(ctx context.Context, method string, uri string, params url.Values, res interface{}) error {
u, err := url.Parse(c.config.Server)
if err != nil {
@ -52,12 +35,13 @@ func (c *Client) doAPI(ctx context.Context, method string, uri string, params ur
if err != nil {
return err
}
req.WithContext(ctx)
req.Header.Set("Authorization", "Bearer "+c.config.AccessToken)
if params != nil {
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
}
return httpDo(ctx, req, func(resp *http.Response, err error) error {
resp, err := c.Do(req)
if err != nil {
return err
}
@ -69,7 +53,6 @@ func (c *Client) doAPI(ctx context.Context, method string, uri string, params ur
return nil
}
return json.NewDecoder(resp.Body).Decode(&res)
})
}
// NewClient return new mastodon API client.