From 38ce98fada6d0c325b74091ce42f1e1010cf0893 Mon Sep 17 00:00:00 2001
From: wucongxing8150 <815046773@qq.com>
Date: Tue, 10 Dec 2024 14:50:55 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BA=86=E5=88=86=E4=BA=AB?=
 =?UTF-8?q?=E5=98=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/service/QuestionQa.go |  2 +-
 utils/aes.go              | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/api/service/QuestionQa.go b/api/service/QuestionQa.go
index ed31efc..03bb113 100644
--- a/api/service/QuestionQa.go
+++ b/api/service/QuestionQa.go
@@ -205,7 +205,7 @@ func (r *QuestionQaService) AddQuestionQa(req requests.AddQuestionQa) (bool, err
 
 	// 生成分享链接
 	questionQaData := make(map[string]interface{})
-	questionQaData["qa_share_id"] = utils.HashString(fmt.Sprintf("%d", questionQa.QaId))
+	questionQaData["qa_share_id"] = utils.GenerateUniqueCode()
 	err = questionQaDao.EditQuestionQaById(tx, questionQa.QaId, questionQaData)
 	if err != nil {
 		tx.Rollback()
diff --git a/utils/aes.go b/utils/aes.go
index ba70469..88b454f 100644
--- a/utils/aes.go
+++ b/utils/aes.go
@@ -7,7 +7,10 @@ import (
 	"crypto/sha256"
 	"encoding/base64"
 	"encoding/hex"
+	"fmt"
 	"io"
+	mathRand "math/rand"
+	"time"
 )
 
 // Encrypt 加密函数
@@ -88,3 +91,10 @@ func hashKey(key string) []byte {
 	hash := sha256.Sum256([]byte(key))
 	return hash[:]
 }
+
+// GenerateUniqueCode 生成6位数唯一码
+func GenerateUniqueCode() string {
+	mathRand.Seed(time.Now().UnixNano())
+	code := fmt.Sprintf("%06d", mathRand.Intn(999999))
+	return code
+}