vendor/golang.org/x/net/http2/go17.go

   1 // Copyright 2016 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.
   4
   5 // +build go1.7
   6
   7 package http2
   8
   9 import (
  10         "context"
  11         "net"
  12         "net/http"
  13         "net/http/httptrace"
  14         "time"
  15 )
  16
  17 type contextContext interface {
  18         context.Context
  19 }
  20
  21 func serverConnBaseContext(c net.Conn, opts *ServeConnOpts) (ctx contextContext, cancel func()) {
  22         ctx, cancel = context.WithCancel(context.Background())
  23         ctx = context.WithValue(ctx, http.LocalAddrContextKey, c.LocalAddr())
  24         if hs := opts.baseConfig(); hs != nil {
  25                 ctx = context.WithValue(ctx, http.ServerContextKey, hs)
  26         }
  27         return
  28 }
  29
  30 func contextWithCancel(ctx contextContext) (_ contextContext, cancel func()) {
  31         return context.WithCancel(ctx)
  32 }
  33
  34 func requestWithContext(req *http.Request, ctx contextContext) *http.Request {
  35         return req.WithContext(ctx)
  36 }
  37
  38 type clientTrace httptrace.ClientTrace
  39
  40 func reqContext(r *http.Request) context.Context { return r.Context() }
  41
  42 func (t *Transport) idleConnTimeout() time.Duration {
  43         if t.t1 != nil {
  44                 return t.t1.IdleConnTimeout
  45         }
  46         return 0
  47 }
  48
  49 func setResponseUncompressed(res *http.Response) { res.Uncompressed = true }
  50
  51 func traceGotConn(req *http.Request, cc *ClientConn) {
  52         trace := httptrace.ContextClientTrace(req.Context())
  53         if trace == nil || trace.GotConn == nil {
  54                 return
  55         }
  56         ci := httptrace.GotConnInfo{Conn: cc.tconn}
  57         cc.mu.Lock()
  58         ci.Reused = cc.nextStreamID > 1
  59         ci.WasIdle = len(cc.streams) == 0 && ci.Reused
  60         if ci.WasIdle && !cc.lastActive.IsZero() {
  61                 ci.IdleTime = time.Now().Sub(cc.lastActive)
  62         }
  63         cc.mu.Unlock()
  64
  65         trace.GotConn(ci)
  66 }
  67
  68 func traceWroteHeaders(trace *clientTrace) {
  69         if trace != nil && trace.WroteHeaders != nil {
  70                 trace.WroteHeaders()
  71         }
  72 }
  73
  74 func traceGot100Continue(trace *clientTrace) {
  75         if trace != nil && trace.Got100Continue != nil {
  76                 trace.Got100Continue()
  77         }
  78 }
  79
  80 func traceWait100Continue(trace *clientTrace) {
  81         if trace != nil && trace.Wait100Continue != nil {
  82                 trace.Wait100Continue()
  83         }
  84 }
  85
  86 func traceWroteRequest(trace *clientTrace, err error) {
  87         if trace != nil && trace.WroteRequest != nil {
  88                 trace.WroteRequest(httptrace.WroteRequestInfo{Err: err})
  89         }
  90 }
  91
  92 func traceFirstResponseByte(trace *clientTrace) {
  93         if trace != nil && trace.GotFirstResponseByte != nil {
  94                 trace.GotFirstResponseByte()
  95         }
  96 }
  97
  98 func requestTrace(req *http.Request) *clientTrace {
  99         trace := httptrace.ContextClientTrace(req.Context())
 100         return (*clientTrace)(trace)
 101 }
 102
 103 // Ping sends a PING frame to the server and waits for the ack.
 104 func (cc *ClientConn) Ping(ctx context.Context) error {
 105         return cc.ping(ctx)
 106 }