go_gameshield/internal/repository/admin/waflog.go

package admin

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

type WafLogRepository interface {
	GetWafLog(ctx context.Context, id int64) (*model.WafLog, error)
	GetWafLogList(ctx context.Context) ([]model.WafLog, error)
	AddWafLog(ctx context.Context, log *model.WafLog) error
	BatchAddWafLog(ctx context.Context, logs []*model.WafLog) error
}

func NewWafLogRepository(
	repository *repository.Repository,
) WafLogRepository {
	return &wafLogRepository{
		Repository: repository,
	}
}

type wafLogRepository struct {
	*repository.Repository
}

func (r *wafLogRepository) GetWafLog(ctx context.Context, id int64) (*model.WafLog, error) {
	var res model.WafLog
	return &res, r.DBWithName(ctx,"admin").Where("id = ?", id).First(&res).Error
}

func (r *wafLogRepository) GetWafLogList(ctx context.Context) ([]model.WafLog, error) {
	var res []model.WafLog
	return res, r.DBWithName(ctx,"admin").Find(&res).Error
}

func (r *wafLogRepository) AddWafLog(ctx context.Context, log *model.WafLog) error {
	return r.DBWithName(ctx,"admin").Create(log).Error
}

func (r *wafLogRepository) BatchAddWafLog(ctx context.Context, logs []*model.WafLog) error {
	if len(logs) == 0 {
		return nil
	}
	return r.DBWithName(ctx, "admin").CreateInBatches(logs, len(logs)).Error
}