| 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 gae | 5 package wrapper |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "golang.org/x/net/context" | |
| 9 "math/rand" | 8 "math/rand" |
| 10 | 9 |
| 11 » "github.com/luci/luci-go/common/clock" | 10 » "golang.org/x/net/context" |
| 11 » "infra/libs/clock" |
| 12 ) | 12 ) |
| 13 | 13 |
| 14 // MathRandFactory is the function signature for factory methods compatible with | 14 // MathRandFactory is the function signature for factory methods compatible with |
| 15 // SetMathRandFactory. | 15 // SetMathRandFactory. |
| 16 type MathRandFactory func(context.Context) *rand.Rand | 16 type MathRandFactory func(context.Context) *rand.Rand |
| 17 | 17 |
| 18 // GetMathRand gets a *"math/rand".Rand from the context. If one hasn't been | 18 // GetMathRand gets a *"math/rand".Rand from the context. If one hasn't been |
| 19 // set, this creates a new Rand object with a Source initialized from the | 19 // set, this creates a new Rand object with a Source initialized from the |
| 20 // current time clock.Now(c).UnixNano(). | 20 // current time clock.Now(c).UnixNano(). |
| 21 func GetMathRand(c context.Context) *rand.Rand { | 21 func GetMathRand(c context.Context) *rand.Rand { |
| (...skipping 11 matching lines...) Expand all Loading... |
| 33 | 33 |
| 34 // SetMathRand sets the current *"math/rand".Rand object in the context. Useful | 34 // SetMathRand sets the current *"math/rand".Rand object in the context. Useful |
| 35 // for testing with a quick mock. This is just a shorthand SetMathRandFactory | 35 // for testing with a quick mock. This is just a shorthand SetMathRandFactory |
| 36 // invocation to set a factory which always returns the same object. | 36 // invocation to set a factory which always returns the same object. |
| 37 func SetMathRand(c context.Context, r *rand.Rand) context.Context { | 37 func SetMathRand(c context.Context, r *rand.Rand) context.Context { |
| 38 if r == nil { | 38 if r == nil { |
| 39 return SetMathRandFactory(c, nil) | 39 return SetMathRandFactory(c, nil) |
| 40 } | 40 } |
| 41 return SetMathRandFactory(c, func(context.Context) *rand.Rand { return r
}) | 41 return SetMathRandFactory(c, func(context.Context) *rand.Rand { return r
}) |
| 42 } | 42 } |
| OLD | NEW |