90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package answer
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"penahub.gitlab.yandexcloud.net/backend/quiz/common.git/dal/sqlcgen"
|
|
"penahub.gitlab.yandexcloud.net/backend/quiz/common.git/model"
|
|
)
|
|
|
|
type Deps struct {
|
|
Queries *sqlcgen.Queries
|
|
Pool *sql.DB
|
|
}
|
|
|
|
type AnswerRepository struct {
|
|
queries *sqlcgen.Queries
|
|
pool *sql.DB
|
|
}
|
|
|
|
func NewAnswerRepository(deps Deps) *AnswerRepository {
|
|
return &AnswerRepository{
|
|
queries: deps.Queries,
|
|
pool: deps.Pool,
|
|
}
|
|
}
|
|
|
|
// test +
|
|
func (r *AnswerRepository) CreateAnswers(ctx context.Context, answers []model.Answer, session, fp string, quizID uint64) ([]uint64, []error) {
|
|
var (
|
|
answered []uint64
|
|
errs []error
|
|
)
|
|
|
|
tx, err := r.pool.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return nil, []error{err}
|
|
}
|
|
|
|
for _, ans := range answers {
|
|
params := sqlcgen.InsertAnswersParams{
|
|
Content: sql.NullString{String: ans.Content, Valid: true},
|
|
QuizID: int64(quizID),
|
|
QuestionID: int64(ans.QuestionId),
|
|
Fingerprint: sql.NullString{String: fp, Valid: true},
|
|
Session: sql.NullString{String: session, Valid: true},
|
|
Result: sql.NullBool{Bool: ans.Result, Valid: true},
|
|
Email: ans.Email,
|
|
}
|
|
|
|
err := r.queries.InsertAnswers(ctx, params)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
} else {
|
|
answered = append(answered, ans.QuestionId)
|
|
}
|
|
}
|
|
|
|
err = tx.Commit()
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
return nil, errs
|
|
}
|
|
|
|
return answered, nil
|
|
}
|
|
|
|
// test +
|
|
func (r *AnswerRepository) GetAllAnswersByQuizID(ctx context.Context, session string) ([]model.ResultAnswer, error) {
|
|
var results []model.ResultAnswer
|
|
|
|
rows, err := r.queries.GetAllAnswersByQuizID(ctx, sql.NullString{String: session, Valid: true})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, row := range rows {
|
|
|
|
resultAnswer := model.ResultAnswer{
|
|
Content: row.Content.String,
|
|
CreatedAt: row.CreatedAt.Time,
|
|
QuestionID: uint64(row.QuestionID),
|
|
AnswerID: uint64(row.ID),
|
|
}
|
|
|
|
results = append(results, resultAnswer)
|
|
}
|
|
|
|
return results, nil
|
|
}
|