| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 package prod | 5 package prod |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "net/http" | 8 "net/http" |
| 9 | 9 |
| 10 uf "github.com/luci/gae/service/urlfetch" | 10 uf "github.com/luci/gae/service/urlfetch" |
| 11 "golang.org/x/net/context" | 11 "golang.org/x/net/context" |
| 12 "google.golang.org/appengine/urlfetch" | 12 "google.golang.org/appengine/urlfetch" |
| 13 ) | 13 ) |
| 14 | 14 |
| 15 // useURLFetch adds a http.RoundTripper implementation to the context. | 15 // useURLFetch adds a http.RoundTripper implementation to the context. |
| 16 func useURLFetch(c context.Context) context.Context { | 16 func useURLFetch(c context.Context) context.Context { |
| 17 return uf.SetFactory(c, func(ci context.Context) http.RoundTripper { | 17 return uf.SetFactory(c, func(ci context.Context) http.RoundTripper { |
| 18 » » return &urlfetch.Transport{Context: ci} | 18 » » return &urlfetch.Transport{Context: AEContext(ci)} |
| 19 }) | 19 }) |
| 20 } | 20 } |
| OLD | NEW |