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 保存验证码