liulei/jx-callback
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tuiguangm

Browse Source
This commit is contained in:
苏尹岚
2020-11-16 11:45:06 +08:00
parent eaaa634e5a
commit 0c140fa4c1
2 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
business/jxstore/cms/user2.go
View File

@@ -227,7 +227,7 @@ func CreateUser(user *model.User, creatorName string) (err error) {
user.Status = model.UserStatusNormal user.Status = model.UserStatusNormal
// user.DividePercentage = 1 // user.DividePercentage = 1
//推广码 //推广码
user.PopCode = jxutils.GenValidateCode(6) user.PopCode = jxutils.GenRandomString(6)
return dao.CreateEntity(nil, user) return dao.CreateEntity(nil, user)
} }

19
business/jxutils/jxutils.go
View File

@@ -894,16 +894,13 @@ func polarTriangleArea(tan1, lng1, tan2, lng2 float64) (s float64) {
return 2 * math.Atan2(t*math.Sin(deltaLng), 1+t*math.Cos(deltaLng)) return 2 * math.Atan2(t*math.Sin(deltaLng), 1+t*math.Cos(deltaLng))
} }
func GenValidateCode(width int) string { func GenRandomString(l int) string {
numeric := [36]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, str := "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', bytes := []byte(str)
'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'} result := []byte{}
r := len(numeric) r := rand.New(rand.NewSource(time.Now().UnixNano()))
rand.Seed(time.Now().UnixNano()) for i := 0; i < l; i++ {
result = append(result, bytes[r.Intn(len(bytes))])
var sb strings.Builder
for i := 0; i < width; i++ {
fmt.Fprintf(&sb, "%d", numeric[rand.Intn(r)])
} }
return sb.String() return string(result)
} }