diff --git a/repository/quiz/quiz.go b/repository/quiz/quiz.go index 078f89d..752c886 100644 --- a/repository/quiz/quiz.go +++ b/repository/quiz/quiz.go @@ -702,13 +702,29 @@ func (r *QuizRepository) GetQuizAudience(ctx context.Context, quizID int64) ([]m } func (r *QuizRepository) DeleteQuizAudience(ctx context.Context, quizID int64, audienceID int64) error { - err := r.queries.DeleteQuizAudience(ctx, quizID) + err := r.queries.DeleteQuizAudience(ctx, sqlcgen.DeleteQuizAudienceParams{ + Quizid: quizID, + ID: audienceID, + }) if err != nil { return err } return nil } +func (r *QuizRepository) CheckIsOwnerAudience(ctx context.Context, quizID int64, audienceID int64, accountID string) (bool, error) { + ok, err := r.queries.CheckIsOwnerAudience(ctx, sqlcgen.CheckIsOwnerAudienceParams{ + Quizid: quizID, + ID: audienceID, + Accountid: accountID, + }) + if err != nil { + return false, err + } + + return ok, nil +} + func (r *QuizRepository) CheckQuizOwner(ctx context.Context, accountID string, quizID uint64) (bool, error) { id, err := r.queries.CheckQuizOwner(ctx, sqlcgen.CheckQuizOwnerParams{ Accountid: accountID,