common/repository/answer/answer.go

package answer

import (
	"context"
	"database/sql"
	"encoding/json"
	"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
	AnswerMinio *StorerAnswer
}

type AnswerRepository struct {
	queries     *sqlcgen.Queries
	pool        *sql.DB
	answerMinio *StorerAnswer
}

func NewAnswerRepository(deps Deps) *AnswerRepository {
	return &AnswerRepository{
		queries:     deps.Queries,
		pool:        deps.Pool,
		answerMinio: deps.AnswerMinio,
	}
}

// test +
func (r *AnswerRepository) CreateAnswers(ctx context.Context, answers []model.Answer, session, fp string, quizID uint64) ([]model.Answer, []error) {
	var (
		createdAnswers []model.Answer
		errs           []error
	)

	tx, err := r.pool.BeginTx(ctx, nil)
	if err != nil {
		return nil, []error{err}
	}

	for _, ans := range answers {
		if ans.Utm == nil {
			ans.Utm = make(model.UTMSavingMap)
		}
		utmJSON, err := json.Marshal(ans.Utm)
		if err != nil {
			return nil, []error{err}
		}

		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,
			Device:      ans.Device,
			DeviceType:  ans.DeviceType,
			Ip:          ans.IP,
			Browser:     ans.Browser,
			Os:          ans.OS,
			Start:       ans.Start,
			Utm:         utmJSON,
		}

		row, err := r.queries.InsertAnswers(ctx, params)
		createdAnswer := model.Answer{
			Id:          uint64(row.ID),
			Content:     row.Content.String,
			QuizId:      uint64(row.QuizID),
			QuestionId:  uint64(row.QuestionID),
			Fingerprint: row.Fingerprint.String,
			Session:     row.Session.String,
			Result:      row.Result.Bool,
			New:         row.New.Bool,
			Email:       row.Email,
			DeviceType:  row.DeviceType,
			Device:      row.Device,
			OS:          row.Os,
			Browser:     row.Browser,
			IP:          row.Ip,
			Start:       row.Start,
		}

		if err != nil {
			errs = append(errs, err)
		} else {
			createdAnswers = append(createdAnswers, createdAnswer)
		}
	}

	err = tx.Commit()
	if err != nil {
		errs = append(errs, err)
		return nil, errs
	}

	return createdAnswers, errs
}

// 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 {
		//todo тут забыл добавить проверку на то что minio !=nil
		/*if row.Questiontype == model.TypeFile {
			fmt.Println("GALL", row.Qid, row.QuestionID, row.Content)
			fileURL, err := r.answerMinio.GetAnswerURL(ctx, row.Qid.UUID.String(), row.QuestionID, row.Content.String)
			if err != nil {
				fmt.Println("GetAnswerURL dal answer minio answer", err)
				return nil, err
			}
			row.Content = sql.NullString{String: fmt.Sprintf("%s|%s", fileURL, row.Content.String), Valid: true}
		}*/

		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
}