From 2e8f0cb3f25ba904a484305a66db095e9dac088d Mon Sep 17 00:00:00 2001
From: lzh <18320341470>
Date: Fri, 20 Jun 2025 14:49:45 +0800
Subject: [PATCH] sms_tool

---
 sms_tool/sms_client.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/sms_tool/sms_client.go b/sms_tool/sms_client.go
index d0f815b..00c6a95 100644
--- a/sms_tool/sms_client.go
+++ b/sms_tool/sms_client.go
@@ -70,7 +70,15 @@ func (c *SmsClient) GetCode(ctx context.Context, key string) (code string, err e
 	if err != nil {
 		return "", err
 	}
-	return value.(string), nil
+	if value == nil {
+		return "", errors.New("验证码不存在，请重新发送")
+	}
+	switch value.(type) {
+	case string:
+		return value.(string), nil
+	default:
+		return "", errors.New("验证码类型错误，请联系管理员")
+	}
 }
 
 // SaveCode 保存验证码