impl/prod: Embed AppEngine SDK into Context.

impl/prod/info.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package prod
 
 import (
         "time"
 
         "github.com/luci/gae/service/info"
         "golang.org/x/net/context"
         "google.golang.org/appengine"
         "google.golang.org/appengine/datastore"
 )
 
 // useGI adds a gae.GlobalInfo implementation to context, accessible
 // by gae.GetGI(c)
 func useGI(usrCtx context.Context) context.Context {
         probeCache := getProbeCache(usrCtx)
         if probeCache == nil {
-                usrCtx = withProbeCache(usrCtx, probe(AEContext(usrCtx)))
+                usrCtx = withProbeCache(usrCtx, probe(getAEContext(usrCtx)))
         }
 
         return info.SetFactory(usrCtx, func(ci context.Context) info.RawInterface {
-                return giImpl{ci, AEContext(ci)}
+                return giImpl{ci, getAEContext(ci)}
         })
 }
 
 type giImpl struct {
         usrCtx context.Context
         aeCtx  context.Context
 }
 
 func (g giImpl) AccessToken(scopes ...string) (token string, expiry time.Time, err error) {
         return appengine.AccessToken(g.aeCtx, scopes...)
@@ skipping 113 matching lines @@
 func getProbeCache(c context.Context) *infoProbeCache {
         if pc, ok := c.Value(&probeCacheKey).(*infoProbeCache); ok {
                 return pc
         }
         return nil
 }
 
 func withProbeCache(c context.Context, pc *infoProbeCache) context.Context {
         return context.WithValue(c, &probeCacheKey, pc)
 }