grpcrand.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. /*
  2. *
  3. * Copyright 2018 gRPC authors.
  4. *
  5. * Licensed under the Apache License, Version 2.0 (the "License");
  6. * you may not use this file except in compliance with the License.
  7. * You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS,
  13. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. *
  17. */
  18. // Package grpcrand implements math/rand functions in a concurrent-safe way
  19. // with a global random source, independent of math/rand's global source.
  20. package grpcrand
  21. import (
  22. "math/rand"
  23. "sync"
  24. "time"
  25. )
  26. var (
  27. r = rand.New(rand.NewSource(time.Now().UnixNano()))
  28. mu sync.Mutex
  29. )
  30. // Int63n implements rand.Int63n on the grpcrand global source.
  31. func Int63n(n int64) int64 {
  32. mu.Lock()
  33. res := r.Int63n(n)
  34. mu.Unlock()
  35. return res
  36. }
  37. // Intn implements rand.Intn on the grpcrand global source.
  38. func Intn(n int) int {
  39. mu.Lock()
  40. res := r.Intn(n)
  41. mu.Unlock()
  42. return res
  43. }
  44. // Float64 implements rand.Float64 on the grpcrand global source.
  45. func Float64() float64 {
  46. mu.Lock()
  47. res := r.Float64()
  48. mu.Unlock()
  49. return res
  50. }