mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
Merge pull request #6173 from gyuho/ccc
pkg/httputil: simplify RequestCanceler args
This commit is contained in:
commit
96e018634a
@ -13,7 +13,7 @@ import (
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func RequestCanceler(rt http.RoundTripper, req *http.Request) func() {
|
||||
func RequestCanceler(req *http.Request) func() {
|
||||
ch := make(chan struct{})
|
||||
req.Cancel = ch
|
||||
|
||||
|
@ -110,7 +110,7 @@ func (p *reverseProxy) ServeHTTP(rw http.ResponseWriter, clientreq *http.Request
|
||||
var requestClosed int32
|
||||
completeCh := make(chan bool, 1)
|
||||
closeNotifier, ok := rw.(http.CloseNotifier)
|
||||
cancel := httputil.RequestCanceler(p.transport, proxyreq)
|
||||
cancel := httputil.RequestCanceler(proxyreq)
|
||||
if ok {
|
||||
closeCh := closeNotifier.CloseNotify()
|
||||
go func() {
|
||||
|
@ -117,7 +117,7 @@ func (p *pipeline) post(data []byte) (err error) {
|
||||
req := createPostRequest(u, RaftPrefix, bytes.NewBuffer(data), "application/protobuf", p.tr.URLs, p.tr.ID, p.tr.ClusterID)
|
||||
|
||||
done := make(chan struct{}, 1)
|
||||
cancel := httputil.RequestCanceler(p.tr.pipelineRt, req)
|
||||
cancel := httputil.RequestCanceler(req)
|
||||
go func() {
|
||||
select {
|
||||
case <-done:
|
||||
|
@ -103,7 +103,7 @@ func (s *snapshotSender) send(merged snap.Message) {
|
||||
// post posts the given request.
|
||||
// It returns nil when request is sent out and processed successfully.
|
||||
func (s *snapshotSender) post(req *http.Request) (err error) {
|
||||
cancel := httputil.RequestCanceler(s.tr.pipelineRt, req)
|
||||
cancel := httputil.RequestCanceler(req)
|
||||
|
||||
type responseAndError struct {
|
||||
resp *http.Response
|
||||
|
@ -413,7 +413,7 @@ func (cr *streamReader) dial(t streamType) (io.ReadCloser, error) {
|
||||
return nil, fmt.Errorf("stream reader is stopped")
|
||||
default:
|
||||
}
|
||||
cr.cancel = httputil.RequestCanceler(cr.tr.streamRt, req)
|
||||
cr.cancel = httputil.RequestCanceler(req)
|
||||
cr.mu.Unlock()
|
||||
|
||||
resp, err := cr.tr.streamRt.RoundTrip(req)
|
||||
|
Loading…
x
Reference in New Issue
Block a user