package dao

import (
	"youngee_b_api/app/entity"
)

type LocalTaskLinkStatisticDao struct{}

// 获取指定服务商下的作品数据
func (d LocalTaskLinkStatisticDao) GetSupplierData(supplierId int64) ([]entity.LocalTaskLinkStatistic, error) {
	localTaskLinkStatistics := []entity.LocalTaskLinkStatistic{}
	err1 := Db.Model(&entity.LocalTaskLinkStatistic{}).Where("supplier_id = ?", supplierId).Find(&localTaskLinkStatistics).Error
	if err1 != nil {
		return localTaskLinkStatistics, err1
	}
	return localTaskLinkStatistics, nil
}

// 获取指定服务商下的作品数据
func (d LocalTaskLinkStatisticDao) GetLocalList(supplierId int64, page int, pageSize int) ([]entity.LocalTaskLinkStatistic, int64, error) {
	var localTaskLinkStatistics []entity.LocalTaskLinkStatistic
	var total int64
	query := Db.Model(&entity.LocalTaskLinkStatistic{}).Where("supplier_id = ?", supplierId)
	query.Count(&total)
	offset := (page - 1) * pageSize
	if err := query.Order("create_time desc").Offset(offset).Limit(pageSize).Find(&localTaskLinkStatistics).Error; err != nil {
		return nil, 0, err
	}

	return localTaskLinkStatistics, total, nil
}