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

impl/prod/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 prod
 
 import (
         "fmt"
         "reflect"
 
         tq "github.com/luci/gae/service/taskqueue"
         "golang.org/x/net/context"
         "google.golang.org/appengine"
         "google.golang.org/appengine/taskqueue"
 )
 
 // useTQ adds a gae.TaskQueue implementation to context, accessible
 // by gae.GetTQ(c)
 func useTQ(c context.Context) context.Context {
-        return tq.SetRawFactory(c, func(ci context.Context) tq.RawInterface {
-                return tqImpl{AEContext(ci)}
+        return tq.SetRawFactory(c, func(ci context.Context, wantTxn bool) tq.RawInterface {
+                if wantTxn {
+                        return tqImpl{AEContext(ci)}
+                }
+                return tqImpl{AEContextNoTxn(ci)}
         })
 }
 
 type tqImpl struct {
         aeCtx context.Context
 }
 
 func init() {
         const taskExpectedFields = 10
         // Runtime-assert that the number of fields in the Task structs match, to
@@ skipping 96 matching lines @@
         }
         for _, s := range stats {
                 cb((*tq.Statistics)(&s), nil)
         }
         return nil
 }
 
 func (t tqImpl) Testable() tq.Testable {
         return nil
 }