sms_tool

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
 	}
+	if value == nil {
+		return "", errors.New("验证码不存在，请重新发送")
+	}
+	switch value.(type) {
+	case string:
 		return value.(string), nil
+	default:
+		return "", errors.New("验证码类型错误，请联系管理员")
+	}
 }
 
 // SaveCode 保存验证码