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 memory | 5 package memory |
6 | 6 |
7 import ( | 7 import ( |
8 "regexp" | 8 "regexp" |
9 "sync/atomic" | 9 "sync/atomic" |
10 | 10 |
11 "golang.org/x/net/context" | 11 "golang.org/x/net/context" |
12 | 12 |
13 tq "github.com/luci/gae/service/taskqueue" | 13 tq "github.com/luci/gae/service/taskqueue" |
14 "github.com/luci/luci-go/common/errors" | 14 "github.com/luci/luci-go/common/errors" |
15 "github.com/luci/luci-go/common/mathrand" | 15 "github.com/luci/luci-go/common/mathrand" |
16 ) | 16 ) |
17 | 17 |
18 /////////////////////////////// public functions /////////////////////////////// | 18 /////////////////////////////// public functions /////////////////////////////// |
19 | 19 |
20 func useTQ(c context.Context) context.Context { | 20 func useTQ(c context.Context) context.Context { |
21 return tq.SetRawFactory(c, func(ic context.Context, wantTxn bool) tq.Raw
Interface { | 21 return tq.SetRawFactory(c, func(ic context.Context, wantTxn bool) tq.Raw
Interface { |
22 » » ns := curGID(ic).namespace | 22 » » ns, _ := curGID(ic).getNamespace() |
23 var tqd memContextObj | 23 var tqd memContextObj |
24 | 24 |
25 if !wantTxn { | 25 if !wantTxn { |
26 tqd = curNoTxn(ic).Get(memContextTQIdx) | 26 tqd = curNoTxn(ic).Get(memContextTQIdx) |
27 } else { | 27 } else { |
28 tqd = cur(ic).Get(memContextTQIdx) | 28 tqd = cur(ic).Get(memContextTQIdx) |
29 } | 29 } |
30 | 30 |
31 if x, ok := tqd.(*taskQueueData); ok { | 31 if x, ok := tqd.(*taskQueueData); ok { |
32 return &taskqueueImpl{x, ic, ns} | 32 return &taskqueueImpl{x, ic, ns} |
(...skipping 219 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
252 func dupQueue(q tq.QueueData) tq.QueueData { | 252 func dupQueue(q tq.QueueData) tq.QueueData { |
253 r := make(tq.QueueData, len(q)) | 253 r := make(tq.QueueData, len(q)) |
254 for k, q := range q { | 254 for k, q := range q { |
255 r[k] = make(map[string]*tq.Task, len(q)) | 255 r[k] = make(map[string]*tq.Task, len(q)) |
256 for tn, t := range q { | 256 for tn, t := range q { |
257 r[k][tn] = t.Duplicate() | 257 r[k][tn] = t.Duplicate() |
258 } | 258 } |
259 } | 259 } |
260 return r | 260 return r |
261 } | 261 } |
OLD | NEW |