diff options
author | Arthur Rand <arand@ucsc.edu> | 2018-02-27 12:48:31 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-02-27 12:48:31 -0800 |
commit | ab589175d1f0fd9d8f54cf3067e8eeef44a1bea8 (patch) | |
tree | 26f02bff0297a392eabf2887be6dac682b3d627e | |
parent | 558d963486a773d4d4e2e834f78626cb7d289b07 (diff) | |
parent | 6a2894282e29771c1a8fe8d44f66d1a2ec7380ea (diff) | |
download | driver-core-ab589175d1f0fd9d8f54cf3067e8eeef44a1bea8.tar.gz driver-core-ab589175d1f0fd9d8f54cf3067e8eeef44a1bea8.tar.bz2 driver-core-ab589175d1f0fd9d8f54cf3067e8eeef44a1bea8.zip |
Merge pull request #126 from drivergroup/ar-add-number-token-generator
add nextNumericToken
-rw-r--r-- | src/main/scala/xyz/driver/core/generators.scala | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/main/scala/xyz/driver/core/generators.scala b/src/main/scala/xyz/driver/core/generators.scala index e2a4d5a..e842dd2 100644 --- a/src/main/scala/xyz/driver/core/generators.scala +++ b/src/main/scala/xyz/driver/core/generators.scala @@ -20,8 +20,14 @@ object generators { private val DefaultMaxLength = 10 private val StringLetters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ ".toSet private val NonAmbigiousCharacters = "abcdefghijkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ23456789".toSet + private val Numbers = "0123456789".toSet - def nextToken(length: Int): String = List.fill(length)(oneOf(NonAmbigiousCharacters)).mkString + private def nextTokenString(length: Int, charSet: Set[Char]): String = + List.fill(length)(oneOf(charSet)).mkString + + def nextToken(length: Int): String = nextTokenString(length, NonAmbigiousCharacters) + + def nextNumericToken(length: Int): String = nextTokenString(length, Numbers) def nextInt(maxValue: Int, minValue: Int = 0): Int = random.nextInt(maxValue - minValue) + minValue |