OLD | NEW |
1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
3 // that can be found in the LICENSE file. | 3 // that can be 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/data/rand/mathrand" |
14 "github.com/luci/luci-go/common/errors" | 15 "github.com/luci/luci-go/common/errors" |
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).getNamespace() | 22 ns, _ := curGID(ic).getNamespace() |
23 var tqd memContextObj | 23 var tqd memContextObj |
24 | 24 |
25 if !wantTxn { | 25 if !wantTxn { |
(...skipping 226 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 |