go_gameshield/internal/repository/gameshieldbackend.go

package repository

import (
	"context"
	v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
	"github.com/go-nunu/nunu-layout-advanced/internal/model"
)

type GameShieldBackendRepository interface {
	GetGameShieldBackendById(ctx context.Context, id int64) (*model.GameShieldBackend, error)
	AddGameShieldBackend(ctx context.Context, req *model.GameShieldBackend) error
	EditGameShieldBackend(ctx context.Context, req *v1.GameShieldBackendRequest) error
	DeleteGameShieldBackend(ctx context.Context, id int64) error
	GetGameShieldBackendByHostId(ctx context.Context, hostId int) (*[]model.GameShieldBackend, error)
}

func NewGameShieldBackendRepository(
	repository *Repository,
) GameShieldBackendRepository {
	return &gameShieldBackendRepository{
		Repository: repository,
	}
}

type gameShieldBackendRepository struct {
	*Repository
}

func (r *gameShieldBackendRepository) GetGameShieldBackendById(ctx context.Context, id int64) (*model.GameShieldBackend, error) {
	var gameShieldBackend model.GameShieldBackend
	if err := r.DB(ctx).Where("id = ?", id).First(&gameShieldBackend).Error; err != nil {
		return nil, err
	}
	return &gameShieldBackend, nil
}

func (r *gameShieldBackendRepository) AddGameShieldBackend(ctx context.Context, req *model.GameShieldBackend) error {
	if err := r.DB(ctx).Create(&req).Error; err != nil {
		return err
	}
	return nil
}

func (r *gameShieldBackendRepository) EditGameShieldBackend(ctx context.Context, req *v1.GameShieldBackendRequest) error {
	if err := r.DB(ctx).Model(&model.GameShieldBackend{}).Where("id = ?", req.Id).Updates(req).Error; err != nil {
		return err
	}
	return nil
}

func (r *gameShieldBackendRepository) DeleteGameShieldBackend(ctx context.Context, id int64) error {
	if err := r.DB(ctx).Delete(&model.GameShieldBackend{}, id).Where("id = ?", id).Error; err != nil {
		return err
	}
	return nil
}

func (r *gameShieldBackendRepository) GetGameShieldBackendByHostId(ctx context.Context, hostId int) (*[]model.GameShieldBackend, error) {
	var gameShieldBackend []model.GameShieldBackend
	if err := r.DB(ctx).Where("host_id = ?", hostId).Find(&gameShieldBackend).Error; err != nil {
		return nil, err
	}
	return &gameShieldBackend, nil
}