ducky/devices
2015-11-15
Parent:b6494e1a499e
ducky/devices/vendor/golang.org/x/net/context/ctxhttp/ctxhttp.go
Implement ListDevicesByOwner. Write a unit test for listing Devices by their Owner property, as per our Storer interface. Basically, if we insert Devices into the datastore, we need to be able to retrieve the Devices that someone owns. Simple enough. This meant that I needed to implement the Memstore version, past the barebones stub, so the tests would continue to pass.
1 // Copyright 2015 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
5 // Package ctxhttp provides helper functions for performing context-aware HTTP requests.
6 package ctxhttp
8 import (
9 "io"
10 "net/http"
11 "net/url"
12 "strings"
14 "golang.org/x/net/context"
15 )
17 // Do sends an HTTP request with the provided http.Client and returns an HTTP response.
18 // If the client is nil, http.DefaultClient is used.
19 // If the context is canceled or times out, ctx.Err() will be returned.
20 func Do(ctx context.Context, client *http.Client, req *http.Request) (*http.Response, error) {
21 if client == nil {
22 client = http.DefaultClient
23 }
25 // Request cancelation changed in Go 1.5, see cancelreq.go and cancelreq_go14.go.
26 cancel := canceler(client, req)
28 type responseAndError struct {
29 resp *http.Response
30 err error
31 }
32 result := make(chan responseAndError, 1)
34 go func() {
35 resp, err := client.Do(req)
36 result <- responseAndError{resp, err}
37 }()
39 select {
40 case <-ctx.Done():
41 cancel()
42 return nil, ctx.Err()
43 case r := <-result:
44 return r.resp, r.err
45 }
46 }
48 // Get issues a GET request via the Do function.
49 func Get(ctx context.Context, client *http.Client, url string) (*http.Response, error) {
50 req, err := http.NewRequest("GET", url, nil)
51 if err != nil {
52 return nil, err
53 }
54 return Do(ctx, client, req)
55 }
57 // Head issues a HEAD request via the Do function.
58 func Head(ctx context.Context, client *http.Client, url string) (*http.Response, error) {
59 req, err := http.NewRequest("HEAD", url, nil)
60 if err != nil {
61 return nil, err
62 }
63 return Do(ctx, client, req)
64 }
66 // Post issues a POST request via the Do function.
67 func Post(ctx context.Context, client *http.Client, url string, bodyType string, body io.Reader) (*http.Response, error) {
68 req, err := http.NewRequest("POST", url, body)
69 if err != nil {
70 return nil, err
71 }
72 req.Header.Set("Content-Type", bodyType)
73 return Do(ctx, client, req)
74 }
76 // PostForm issues a POST request via the Do function.
77 func PostForm(ctx context.Context, client *http.Client, url string, data url.Values) (*http.Response, error) {
78 return Post(ctx, client, url, "application/x-www-form-urlencoded", strings.NewReader(data.Encode()))
79 }