Add API to allow you to get the non-transactional datastore or taskqueue.

impl/memory/taskqueue.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 (
         "regexp"
         "sync/atomic"
 
         "golang.org/x/net/context"
 
         tq "github.com/luci/gae/service/taskqueue"
         "github.com/luci/luci-go/common/errors"
         "github.com/luci/luci-go/common/mathrand"
 )
 
 /////////////////////////////// public functions ///////////////////////////////
 
 func useTQ(c context.Context) context.Context {
-        return tq.SetRawFactory(c, func(ic context.Context) tq.RawInterface {
-                tqd := cur(ic).Get(memContextTQIdx)
+        return tq.SetRawFactory(c, func(ic context.Context, wantTxn bool) tq.RawInterface {
+                ns := curGID(ic).namespace
+                var tqd memContextObj
+
+                if !wantTxn {
+                        tqd = curNoTxn(ic).Get(memContextTQIdx)
+                } else {
+                        tqd = cur(ic).Get(memContextTQIdx)
+                }
+
                 if x, ok := tqd.(*taskQueueData); ok {
-                        return &taskqueueImpl{
-                                x,
-                                ic,
-                                curGID(ic).namespace,
-                        }
+                        return &taskqueueImpl{x, ic, ns}

[iannucci, 2015/12/04 03:35:10]

don't need to tweak the context because taskqueue only uses it for clock and
mathrand (it never gets passed back out to the user)

                 }
-                return &taskqueueTxnImpl{
-                        tqd.(*txnTaskQueueData),
-                        ic,
-                        curGID(ic).namespace,
-                }
+                return &taskqueueTxnImpl{tqd.(*txnTaskQueueData), ic, ns}
         })
 }
 
 //////////////////////////////// taskqueueImpl /////////////////////////////////
 
 type taskqueueImpl struct {
         *taskQueueData
 
         ctx context.Context
         ns  string
@@ skipping 207 matching lines @@
 func dupQueue(q tq.QueueData) tq.QueueData {
         r := make(tq.QueueData, len(q))
         for k, q := range q {
                 r[k] = make(map[string]*tq.Task, len(q))
                 for tn, t := range q {
                         r[k][tn] = t.Duplicate()
                 }
         }
         return r
 }