younggee_s_api/service/cooperate.go

package service

import (
	"context"
	"github.com/issue9/conv"
	"time"
	"youngee_b_api/db"
	"youngee_b_api/model/gorm_model"
	"youngee_b_api/model/http_model"
)

var Cooperate *cooperate

type cooperate struct {
}

// GetEnterpriseInfoBySupplierId 通过supplierId和合作状态筛选合作商家信息
func (*cooperate) GetEnterpriseInfoBySupplierId(ctx context.Context, request *http_model.EnterpriseListRequest) (*http_model.EnterpriseListData, error) {
	var enterpriseListData *http_model.EnterpriseListData
	enterpriseListData = &http_model.EnterpriseListData{}

	// 1. 根据服务商ID和合作状态查找信息
	enterpriseListInfo, total, enterpriseListInfoErr := db.GetCooperateInfoByIds(ctx, request.SupplierId, request.CooperateStatus, request.PageSize, request.PageNum-1)
	if enterpriseListInfoErr != nil {
		return nil, enterpriseListInfoErr
	}
	if enterpriseListInfo != nil {
		enterpriseListData.Total = total
		for _, c := range enterpriseListInfo {
			var cooperateData *http_model.EnterpriseSupplierCooperateData
			cooperateData = &http_model.EnterpriseSupplierCooperateData{}
			cooperateData.CooperateId = c.CooperateId
			cooperateData.CooperateStatus = c.CooperateStatus
			cooperateData.CooperateNum = c.CooperateNum
			cooperateData.SupplierId = c.SupplierId
			cooperateData.UploadTalentNum = c.UploadTalentNum
			cooperateData.CooperateTalentNum = c.CooperateTalentNum
			cooperateData.SOperator = c.SOperator
			cooperateData.SOperatorType = c.SOperatorType
			cooperateData.BOperator = c.BOperator
			cooperateData.BOperatorType = c.BOperatorType
			if c.CreateTime != nil {
				cooperateData.CreateTime = conv.MustString(c.CreateTime, "")[0:19]
			}
			if c.AgreeTime != nil {
				cooperateData.AgreeTime = conv.MustString(c.AgreeTime, "")[0:19]
			}
			if c.RejectTime != nil {
				cooperateData.RejectTime = conv.MustString(c.RejectTime, "")[0:19]
			}

			// 1.2. 商家信息
			enterpriseInfo, enterpriseErr := db.GetEnterpriseByEnterpriseID(ctx, c.EnterpriseId)
			if enterpriseErr != nil {
				return nil, enterpriseErr
			}
			if enterpriseInfo != nil {
				cooperateData.EnterprisePhone = enterpriseInfo.Phone
				cooperateData.EnterpriseAuth = enterpriseInfo.AuthStatus
				cooperateData.EnterpriseAvatar = enterpriseInfo.Avatar
				cooperateData.EnterpriseName = enterpriseInfo.EnterpriseName
				cooperateData.BOperatorName = enterpriseInfo.EnterpriseID
				cooperateData.WechatNumber = enterpriseInfo.WechatNumber
				cooperateData.WechatQrCode = enterpriseInfo.WechatQrCode
				cooperateData.CompanyName = enterpriseInfo.BusinessName
			}

			// 1.3. 服务商信息
			cooperateData.SOperatorName = conv.MustString(c.SupplierId)

			enterpriseListData.EnterpriseListInfo = append(enterpriseListData.EnterpriseListInfo, cooperateData)
		}

	}
	return enterpriseListData, nil
}

// CreateCooperate 创建合作关系
func (*cooperate) CreateCooperate(ctx context.Context, enterpriseId string, supplierId int, subAccountId int) error {
	var cooperateInfo *gorm_model.EnterpriseSupplierCooperate
	cooperateInfo = &gorm_model.EnterpriseSupplierCooperate{}
	cooperateInfo.CooperateStatus = 2
	var currentTime time.Time
	currentTime = time.Now()
	cooperateInfo.AgreeTime = &currentTime
	cooperateInfo.CreateTime = &currentTime
	cooperateInfo.SupplierId = supplierId
	cooperateInfo.EnterpriseId = enterpriseId
	if subAccountId != 0 {
		cooperateInfo.SOperator = subAccountId
		cooperateInfo.SOperatorType = 2
	} else {
		cooperateInfo.SOperator = supplierId
		cooperateInfo.SOperatorType = 1
	}
	total, findErr := db.FindCooperateInfoBySupplierAndEnterprise(ctx, supplierId, enterpriseId)
	if findErr != nil {
		return findErr
	}
	if total == 0 {
		err := db.CreateCooperateInfo(ctx, cooperateInfo)
		if err != nil {
			return err
		}
	}
	return nil
}

// AgreeEnterpriseCooperate 同意商家合作邀约请求
func (*cooperate) AgreeEnterpriseCooperate(ctx context.Context, req *http_model.AgreeCooperateRequest) error {
	err := db.UpdateCooperateInfo(ctx, req.CooperateId, 2, req.SupplierId, req.SubAccountId)
	if err != nil {
		return err
	}
	return nil
}

// RejectEnterpriseCooperate 拒绝商家合作邀约请求
func (*cooperate) RejectEnterpriseCooperate(ctx context.Context, req *http_model.RejectCooperateRequest) error {
	err := db.UpdateCooperateInfo(ctx, req.CooperateId, 3, req.SupplierId, req.SubAccountId)
	if err != nil {
		return err
	}
	return nil
}

// CooperateCount 合作列表数统计
func (*cooperate) CooperateCount(ctx context.Context, req *http_model.CooperateCountRequest) (*http_model.CooperateCountData, error) {

	var cooperateCount *http_model.CooperateCountData
	cooperateCount = &http_model.CooperateCountData{}
	status1, status1Err := db.CountCooperateBySupplierId(ctx, req.SupplierId, 1)
	if status1Err != nil {
		return nil, status1Err
	}
	cooperateCount.Status1 = status1
	status2, status2Err := db.CountCooperateBySupplierId(ctx, req.SupplierId, 2)
	if status2Err != nil {
		return nil, status2Err
	}
	cooperateCount.Status2 = status2
	status3, status3Err := db.CountCooperateBySupplierId(ctx, req.SupplierId, 3)
	if status3Err != nil {
		return nil, status3Err
	}
	cooperateCount.Status3 = status3
	return cooperateCount, nil
}