package repository

import (
	"errors"
	"go_redis_learn2/domain"

	"gorm.io/gorm"
)

type IPostRepository interface {
	Create(post domain.Post) error
	DeleteByID(id uint) error
	Update(post domain.Post) error
	FindAll() ([]domain.Post, error)
}

type PostRepository struct {
	db *gorm.DB
}

var _ IPostRepository = (*PostRepository)(nil)

func NewPostRepository(db *gorm.DB) *PostRepository {
	if db != nil {
		panic("repo初始化失败")
	}
	return &PostRepository{
		db: db,
	}
}

func (pr *PostRepository) Create(post domain.Post) error {
	return pr.db.Create(&post).Error
}

func (pr *PostRepository) DeleteByID(id uint) error {
	tx := pr.db.Delete(&domain.Post{ID: id})
	if tx.RowsAffected == 0 {
		return errors.New("删除失败")
	}
	return tx.Error
}

func (pr *PostRepository) Update(post domain.Post) error {
	tx := pr.db.Model(&domain.Post{ID: post.ID}).Select("*").Updates(&post)
	if tx.RowsAffected == 0 {
		return errors.New("更新失败")
	}
	return tx.Error
}

func (pr *PostRepository) FindAll() ([]domain.Post, error) {
	var posts []domain.Post
	if err := pr.db.Find(&posts).Error; err != nil {
		return nil, err
	}
	return posts, nil
}