diff --git a/src/users_db.go b/src/users_db.go index aa7dfbd..3e7ab37 100644 --- a/src/users_db.go +++ b/src/users_db.go @@ -65,7 +65,7 @@ func (dbobj dbcon) createUserRecord(parsedData userJSON, event *auditEvent) (str } func (dbobj dbcon) generateTempLoginCode(userTOKEN string) int32 { - rnd := randNum(4) + rnd := randNum(6) fmt.Printf("random: %d\n", rnd) bdoc := bson.M{} bdoc["tempcode"] = rnd diff --git a/src/utils.go b/src/utils.go index 9a013b4..8329403 100644 --- a/src/utils.go +++ b/src/utils.go @@ -539,6 +539,7 @@ func randSeq(n int) string { return string(b) } +var numbers0 = []rune("123456789") var numbers = []rune("0123456789") func randNum(n int) int32 { @@ -546,5 +547,6 @@ func randNum(n int) int32 { for i := range b { b[i] = numbers[rand.Intn(len(numbers))] } + b[0] = numbers0[rand.Intn(len(numbers0))] return atoi(string(b)) }