younggee_s_api/db/s_t_cooperate.go

package db

import (
	"context"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"youngee_b_api/model/gorm_model"
)

// CreateSTCooperateInfo 创建服务商-达人合作关系
func CreateSTCooperateInfo(ctx context.Context, cooperateInfo *gorm_model.SupplierTalentCooperate) error {
	db := GetWriteDB(ctx)
	err := db.Create(&cooperateInfo).Error
	if err != nil {
		return err
	}
	return nil
}

// CountCooperateInfoBySupplierAndPlatform 查找服务商-商家是否建立合作关系
func CountCooperateInfoBySupplierAndPlatform(ctx context.Context, supplierId int, platformUserId int, taskType int) (int64, error) {
	db := GetReadDB(ctx)
	whereCondition := gorm_model.SupplierTalentCooperate{
		PlatformUserId: platformUserId,
		SupplierId:     supplierId,
		TaskType:       taskType,
	}
	db = db.Debug().Model(gorm_model.SupplierTalentCooperate{}).Where(whereCondition)
	var total int64
	if err := db.Count(&total).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetCooperateInfoByIds] error query mysql total, err:%+v", err)
		return 0, err
	}
	return total, nil
}

// UpdateSTCooperateInfo 更新服务商-达人合作关系
func UpdateSTCooperateInfo(ctx context.Context, supplierId int, platformUserId int, taskType int) error {
	db := GetWriteDB(ctx)
	whereCondition := gorm_model.SupplierTalentCooperate{SupplierId: supplierId, PlatformUserId: platformUserId, TaskType: taskType}
	err := db.Model(&gorm_model.SupplierTalentCooperate{}).
		Where(whereCondition).
		Update("cooperate_num", gorm.Expr("cooperate_num + 1")).
		Error
	if err != nil {
		return err
	}
	return nil
}

// GetSTCooperateInfo 查找合作列表
func GetSTCooperateInfo(ctx context.Context, supplierId int, taskType int, platform int, nickname string, pageSize, pageNum int64) ([]*gorm_model.SupplierTalentCooperate, int64, error) {
	db := GetReadDB(ctx)

	// 1. 按条件过滤 - 必填条件
	db = db.Debug().Model(gorm_model.SupplierTalentCooperate{}).
		Where("supplier_id = ?", supplierId).
		Where("task_type = ?", taskType)

	// 可选条件
	if platform != 0 {
		db = db.Where("platform = ?", platform)
	}
	// if nickname != "" {
	// 	db = db.Where("nickname LIKE ?", "%"+nickname+"%") // 假设是模糊查询
	// }

	// 2. 返回当前页数据并统计总数
	var total int64
	var cooperateInfo []*gorm_model.SupplierTalentCooperate

	// 先统计总数
	if err := db.Count(&total).Error; err != nil {
		logrus.WithContext(ctx).Errorf("[GetCooperateInfoByIds] error query mysql total, err:%+v", err)
		return nil, 0, err
	}

	// 再查询分页数据
	limit := pageSize
	offset := pageSize * pageNum // assert pageNum start with 0
	err := db.Order("supplier_id desc").
		Limit(int(limit)).
		Offset(int(offset)).
		Find(&cooperateInfo).Error

	if err != nil {
		logrus.WithContext(ctx).Errorf("[GetCooperateInfoByIds] error query mysql total, err:%+v", err)
		return nil, 0, err
	}

	return cooperateInfo, total, nil
}