diff --git a/internal/dao/questionnaire_survey.go b/internal/dao/questionnaire_survey.go index 7f37a28..93ce90a 100644 --- a/internal/dao/questionnaire_survey.go +++ b/internal/dao/questionnaire_survey.go @@ -59,6 +59,10 @@ func SendQuestionnaireSurvey(req *bundle.SendQuestionnaireSurveyRequest) (resp * Order("created_at desc"). First(&orderRecord).Error if err != nil { + if errors.Is(err, gorm.ErrRecordNotFound) { + resp.Status = 1 + return resp, nil + } return resp, err } @@ -69,8 +73,13 @@ func SendQuestionnaireSurvey(req *bundle.SendQuestionnaireSurveyRequest) (resp * Where("month = ?", month). First(&bundleBalance).Error if err != nil { + if errors.Is(err, gorm.ErrRecordNotFound) { + resp.Status = 1 + return resp, nil + } return resp, err } + bundleInfo := &model.BundleInfo{ BundleName: orderRecord.BundleName, StartAt: bundleBalance.StartAt,