验证码随机生成的业务逻辑，优化版

backend/verifyCode/cmd/verifyCode/main.go

@@ -2,7 +2,9 @@ package main
 
 import (
 	"flag"
+	"math/rand"
 	"os"
+	"time"
 
 	"verifyCode/internal/conf"
 
@@ -78,6 +80,9 @@ func main() {
 	}
 	defer cleanup()
 
+	// set rand seed
+	rand.Seed(time.Now().UnixNano())
+
 	// start and wait for stop signal
 	if err := app.Run(); err != nil {
 		panic(err)

backend/verifyCode/internal/service/verifycode.go

@@ -2,7 +2,7 @@ package service
 
 import (
 	"context"
-
+	"math/rand"
 	pb "verifyCode/api/verifyCode"
 )
 
@@ -15,5 +15,88 @@ func NewVerifyCodeService() *VerifyCodeService {
 }
 
 func (s *VerifyCodeService) GetVerifyCode(ctx context.Context, req *pb.GetVerifyCodeRequest) (*pb.GetVerifyCodeReply, error) {
-	return &pb.GetVerifyCodeReply{}, nil
+	return &pb.GetVerifyCodeReply{
+		Code: RandCode(int(req.Length), req.Type),
+	}, nil
+}
+
+// RandCode 开放的被调用的方法，用于区分类型
+func RandCode(l int, t pb.TYPE) string {
+	switch t {
+	case pb.TYPE_DEFAULT:
+		fallthrough
+	case pb.TYPE_DIGIT:
+		//return randCode("0123456789", l)
+		return randCode("0123456789", l, 4)
+	case pb.TYPE_LETTER:
+		//return randCode("abcdefghijklmnopqrstuvwxyz", l)
+		return randCode("abcdefghijklmnopqrstuvwxyz", l, 5)
+	case pb.TYPE_MIXED:
+		//return randCode("0123456789abcdefghijklmnopqrstuvwxyz", l)
+		return randCode("0123456789abcdefghijklmnopqrstuvwxyz", l, 6)
+	default:
+
+	}
+	return ""
 }
+
+// 随机数的核心方法（优化实现）
+// 一次随机多个随机位，分部分多次使用
+func randCode(chars string, l, idxBits int) string {
+	// 计算有效的二进制数位，基于 chars 的长度
+	// 推荐写死，因为chars固定，对应的位数长度也固定
+	//idxBits = len(fmt.Sprintf("%b", len(chars)))
+
+	// 形成掩码，mask
+	// 例如，使用低六位：00000000000111111
+	idxMask := 1<<idxBits - 1 // 00001000000 - 1 = 00000111111
+	// 63 位可以用多少次
+	idxMax := 63 / idxBits
+
+	// 结果
+	result := make([]byte, l)
+
+	// 生成随机字符
+	// cache 随机位缓存
+	// remain 当前还可以用几次
+	for i, cache, remain := 0, rand.Int63(), idxMax; i < l; {
+		// 如果使用的剩余次数为0，则重新获取随机
+		if 0 == remain {
+			cache, remain = rand.Int63(), idxMax
+		}
+
+		// 利用掩码获取有效部位的随机数位
+		// 00110100 10110100 10110100 10110100 10110100 10110100 10110100 10110100
+		// &
+		// 00000000 00000000 00000000 00000000 00000000 00000000 00000000 00111111
+		// =
+		// 00000000 00000000 00000000 00000000 00000000 00000000 00000000 00110100
+		if randIndex := int(cache & int64(idxMask)); randIndex < len(chars) {
+			result[i] = chars[randIndex]
+			i++
+		}
+
+		// 使用下一组随机位
+		// 00110100 10110100 10110100 10110100 10110100 10110100 10110100 10110100
+		// 000000 00110100 10110100 10110100 10110100 10110100 10110100 10110100 10
+		cache >>= idxBits
+		// 减少一次使用次数
+		remain--
+	}
+
+	return string(result)
+}
+
+// 随机的核心方法(简单的实现)
+//func randCode(chars string, l int) string {
+//	charsLen := len(chars)
+//	// 结果
+//	result := make([]byte, l)
+//	// 根据目标长度，进行循环
+//	for i := 0; i < l; i++ {
+//		// 核心函数 rand.Intn() 生成[0, n)的整型随机数
+//		randIndex := rand.Intn(charsLen)
+//		result[i] = chars[randIndex]
+//	}
+//	return string(result)
+//}