More functions support Context

ssdp/ssdp.go

@@ -1,6 +1,7 @@
 package ssdp
 
 import (
+	"context"
 	"errors"
 	"log"
 	"net/http"
@@ -34,14 +35,15 @@ type HTTPUClient interface {
 	) ([]*http.Response, error)
 }
 
-// SSDPRawSearch performs a fairly raw SSDP search request, and returns the
+// SSDPRawSearchCtx performs a fairly raw SSDP search request, and returns the
 // unique response(s) that it receives. Each response has the requested
 // searchTarget, a USN, and a valid location. maxWaitSeconds states how long to
 // wait for responses in seconds, and must be a minimum of 1 (the
 // implementation waits an additional 100ms for responses to arrive), 2 is a
 // reasonable value for this. numSends is the number of requests to send - 3 is
 // a reasonable value for this.
-func SSDPRawSearch(
+func SSDPRawSearchCtx(
+	ctx context.Context,
 	httpu HTTPUClient,
 	searchTarget string,
 	maxWaitSeconds int,
@@ -51,7 +53,7 @@ func SSDPRawSearch(
 		return nil, errors.New("ssdp: maxWaitSeconds must be >= 1")
 	}
 
-	req := http.Request{
+	req := (&http.Request{
 		Method: methodSearch,
 		// TODO: Support both IPv4 and IPv6.
 		Host: ssdpUDP4Addr,
@@ -64,8 +66,8 @@ func SSDPRawSearch(
 			"MAN":  []string{ssdpDiscover},
 			"ST":   []string{searchTarget},
 		},
-	}
-	allResponses, err := httpu.Do(&req, time.Duration(maxWaitSeconds)*time.Second+100*time.Millisecond, numSends)
+	}).WithContext(ctx)
+	allResponses, err := httpu.Do(req, time.Duration(maxWaitSeconds)*time.Second+100*time.Millisecond, numSends)
 	if err != nil {
 		return nil, err
 	}
@@ -97,3 +99,9 @@ func SSDPRawSearch(
 
 	return responses, nil
 }
+
+// SSDPRawSearch is the legacy version of SSDPRawSearchCtx, but uses
+// context.Background() as the context.
+func SSDPRawSearch(httpu HTTPUClient, searchTarget string, maxWaitSeconds int, numSends int) ([]*http.Response, error) {
+	return SSDPRawSearchCtx(context.Background(), httpu, searchTarget, maxWaitSeconds, numSends)
+}