learnsteam-quiz-api/internal/repositories/quiz.go
2023-11-17 01:38:01 +09:00

85 lines
2.0 KiB
Go

package repositories
import (
"fmt"
"learnsteam/learsteam-quiz-api/internal/models"
"gorm.io/gorm"
)
type quizRepository struct {
DB *gorm.DB
}
func NewQuizRepository(db *gorm.DB) QuizRepository {
return &quizRepository{
DB: db,
}
}
type QuizRepository interface {
List(string, int, int) (*[]models.Quiz, error)
Total(string) (int64, error)
Find(string) (*models.Quiz, error)
Create(*models.Quiz) (*models.Quiz, error)
Update(*models.Quiz) (*models.Quiz, error)
Delete(string) error
}
func (r *quizRepository) List(program_id string, page int, limit int) (*[]models.Quiz, error) {
var quizzes *[]models.Quiz
var err error
offset := limit * (page - 1)
if program_id != "" {
fmt.Println("program_id", program_id)
err = r.DB.Offset(offset).Limit(limit).Order("sequence ASC").Where("program_id = ?", program_id).Find(&quizzes).Error
} else {
fmt.Println("program_id", "none")
err = r.DB.Offset(offset).Limit(limit).Find(&quizzes).Error
}
return quizzes, err
}
func (r *quizRepository) Total(program_id string) (int64, error) {
var total int64
var err error
if program_id != "" {
err = r.DB.Model(&models.Quiz{}).Where("program_id = ?", program_id).Count(&total).Error
} else {
err = r.DB.Model(&models.Quiz{}).Count(&total).Error
}
return total, err
}
func (r *quizRepository) Find(id string) (*models.Quiz, error) {
var quiz *models.Quiz
err := r.DB.Where("id = ?", id).First(&quiz).Error
return quiz, err
}
func (r *quizRepository) Create(quiz *models.Quiz) (*models.Quiz, error) {
err := r.DB.Create(&quiz).Error
return quiz, err
}
func (r *quizRepository) Update(quiz *models.Quiz) (*models.Quiz, error) {
var row *models.Quiz
if err := r.DB.Where("id=?", quiz.ID).First(&row).Error; err != nil {
return nil, err
}
err := r.DB.Model(&row).Select("*").Updates(&quiz).Error
return row, err
}
func (r *quizRepository) Delete(id string) error {
var quiz *models.Quiz
if err := r.DB.Where("id=?", id).First(&quiz).Error; err != nil {
return err
}
err := r.DB.Delete(&quiz).Error
return err
}