| 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 |