package dao

import (
	"youngee_b_api/app/entity"
)

type ProjectTaskLinkStatisticDao struct{}

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

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

	return projectTaskLinkStatistics, total, nil
}