Make impl/memory have an AppID.

impl/memory/context.go

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package memory
 
 import (
         "errors"
         "sync"
 
         ds "github.com/luci/gae/service/datastore"
         "golang.org/x/net/context"
 )
 
 var serializationDeterministic = false
 
 type memContextObj interface {
         sync.Locker
         canApplyTxn(m memContextObj) bool
         applyTxn(c context.Context, m memContextObj)
 
         endTxn()
         mkTxn(*ds.TransactionOptions) memContextObj
 }
 
 type memContext []memContextObj
 
 var _ = memContextObj((memContext)(nil))
 
-func newMemContext() memContext {
+func newMemContext(aid string) memContext {
         return memContext{
                 newTaskQueueData(),
-                newDataStoreData(),
+                newDataStoreData(aid),
         }
 }
 
 type memContextIdx int
 
 const (
         memContextTQIdx memContextIdx = iota
         memContextDSIdx
 )
 
@@ skipping 37 matching lines @@
         return true
 }
 
 func (m memContext) applyTxn(c context.Context, txnCtxObj memContextObj) {
         txnCtx := txnCtxObj.(memContext)
         for i := range m {
                 m[i].applyTxn(c, txnCtx[i])
         }
 }
 
-// Use adds implementations for the following gae interfaces to the
+// Use calls UseWithAppID with the appid of "dev~app"
+func Use(c context.Context) context.Context {
+        return UseWithAppID(c, "dev~app")
+}
+
+// UseWithAppID adds implementations for the following gae interfaces to the
 // context:
 //   * gae.Datastore
 //   * gae.TaskQueue
 //   * gae.Memcache
 //   * gae.GlobalInfo
 //
+// The application id wil be set to 'aid', and will not be modifiable in this
+// context.
+//
 // These can be retrieved with the gae.Get functions.
 //
 // The implementations are all backed by an in-memory implementation, and start
 // with an empty state.
 //
 // Using this more than once per context.Context will cause a panic.
-func Use(c context.Context) context.Context {
+func UseWithAppID(c context.Context, aid string) context.Context {
         if c.Value(memContextKey) != nil {
                 panic(errors.New("memory.Use: called twice on the same Context"))
         }
         c = context.WithValue(
-                context.WithValue(c, memContextKey, newMemContext()),
-                giContextKey, &globalInfoData{})
-        return useTQ(useRDS(useMC(useGI(c))))
+                context.WithValue(c, memContextKey, newMemContext(aid)),
+                giContextKey, &globalInfoData{appid: aid})
+        return useTQ(useRDS(useMC(useGI(c, aid))))
 }
 
 func cur(c context.Context) (p memContext) {
         p, _ = c.Value(memContextKey).(memContext)
         return
 }
 
 type memContextKeyType int
 
 var memContextKey memContextKeyType
@@ skipping 50 matching lines @@
         if o != nil && o.Attempts != 0 {
                 attempts = o.Attempts
         }
         for attempt := 0; attempt < attempts; attempt++ {
                 if err := loopBody(attempt >= d.data.txnFakeRetry); err != ds.ErrConcurrentTransaction {
                         return err
                 }
         }
         return ds.ErrConcurrentTransaction
 }