330 lines
9.4 KiB
Go
330 lines
9.4 KiB
Go
|
package bus_service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/service"
|
|||
|
|
"go-admin/app/admin/models/bus_models"
|
|||
|
|
"go-admin/common/global"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"math/rand"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type CooperativeService struct {
|
|||
|
|
service.Service
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetCooperativeList 获取合作商列表
|
|||
|
|
func (e *CooperativeService) GetCooperativeList(ctx context.Context, req bus_models.CooperativeListReq) (
|
|||
|
|
bus_models.CooperativeListResp, error) {
|
|||
|
|
var cooperativeList []bus_models.BusCooperative
|
|||
|
|
var resp bus_models.CooperativeListResp
|
|||
|
|
page := req.Page - 1
|
|||
|
|
if page < 0 {
|
|||
|
|
page = 0
|
|||
|
|
}
|
|||
|
|
if req.PageSize == 0 {
|
|||
|
|
req.PageSize = 10
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建查询条件，假设我们支持根据 product_code 查询
|
|||
|
|
db := e.Orm.WithContext(ctx)
|
|||
|
|
|
|||
|
|
// 如果需要支持分页，可以根据请求条件调整查询
|
|||
|
|
if req.CooperativeNumber != "" {
|
|||
|
|
db = db.Where("cooperative_number = ?", req.CooperativeNumber)
|
|||
|
|
}
|
|||
|
|
if req.CooperativeName != "" {
|
|||
|
|
db = db.Where("cooperative_name LIKE ?", "%"+req.CooperativeName+"%")
|
|||
|
|
}
|
|||
|
|
if req.Status != 0 {
|
|||
|
|
db = db.Where("status = ?", req.Status)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 查询数据库
|
|||
|
|
if err := db.Offset(page * req.Page).Limit(req.PageSize).Find(&cooperativeList).Error; err != nil {
|
|||
|
|
return resp, errors.New("查询合作商列表失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp.List = cooperativeList
|
|||
|
|
resp.Total = len(cooperativeList)
|
|||
|
|
resp.Page = page + 1
|
|||
|
|
resp.PageSize = req.PageSize
|
|||
|
|
|
|||
|
|
return resp, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CreateCooperative 新建合作商（使用事务）
|
|||
|
|
func (e *CooperativeService) CreateCooperative(req bus_models.CreateCooperativeReq) (bus_models.CreateCooperativeResp, error) {
|
|||
|
|
var resp bus_models.CreateCooperativeResp
|
|||
|
|
|
|||
|
|
// 开启事务
|
|||
|
|
tx := e.Orm.Begin()
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// 创建合作商
|
|||
|
|
cooperative := bus_models.BusCooperative{
|
|||
|
|
CooperativeName: req.CooperativeName,
|
|||
|
|
Contact: req.Contact,
|
|||
|
|
Tel: req.Tel,
|
|||
|
|
Status: global.CooperativeStatusOnUse,
|
|||
|
|
Account: req.Account,
|
|||
|
|
Password: req.Password,
|
|||
|
|
Balance: req.Balance,
|
|||
|
|
Free: req.Free,
|
|||
|
|
Bond: req.Bond,
|
|||
|
|
CardHolder: req.CardHolder,
|
|||
|
|
Bank: req.Bank,
|
|||
|
|
CardID: req.CardID,
|
|||
|
|
TaxID: req.TaxID,
|
|||
|
|
}
|
|||
|
|
var err error
|
|||
|
|
cooperative.CooperativeNumber, err = GenerateCooperativeNumber(e.Orm)
|
|||
|
|
if err != nil {
|
|||
|
|
return resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := tx.Create(&cooperative).Error; err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
return resp, errors.New("创建合作商失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 插入合作商产品信息
|
|||
|
|
var cooperativeProducts []bus_models.BusCooperativeProduct
|
|||
|
|
for _, product := range req.Products {
|
|||
|
|
cooperativeProducts = append(cooperativeProducts, bus_models.BusCooperativeProduct{
|
|||
|
|
CooperativeID: cooperative.ID,
|
|||
|
|
ProductID: product.ProductID,
|
|||
|
|
Discount: product.Discount,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
if len(cooperativeProducts) > 0 {
|
|||
|
|
if err := tx.Create(&cooperativeProducts).Error; err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
return resp, errors.New("创建合作商产品失败")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 提交事务
|
|||
|
|
if err := tx.Commit().Error; err != nil {
|
|||
|
|
return resp, errors.New("提交事务失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp.CooperativeNumber = cooperative.CooperativeNumber
|
|||
|
|
return resp, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e *CooperativeService) EditCooperative(req bus_models.EditCooperativeReq) error {
|
|||
|
|
// 开启事务
|
|||
|
|
tx := e.Orm.Begin()
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// 查找合作商
|
|||
|
|
var cooperative bus_models.BusCooperative
|
|||
|
|
if err := tx.Where("cooperative_number = ?", req.CooperativeNumber).First(&cooperative).Error; err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
return errors.New("合作商不存在")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 更新合作商信息
|
|||
|
|
updateData := map[string]interface{}{
|
|||
|
|
"cooperative_name": req.CooperativeName,
|
|||
|
|
"contact": req.Contact,
|
|||
|
|
"tel": req.Tel,
|
|||
|
|
"status": req.Status,
|
|||
|
|
"account": req.Account,
|
|||
|
|
"balance": req.Balance,
|
|||
|
|
"free": req.Free,
|
|||
|
|
"bond": req.Bond,
|
|||
|
|
"card_holder": req.CardHolder,
|
|||
|
|
"bank": req.Bank,
|
|||
|
|
"card_id": req.CardID,
|
|||
|
|
"tax_id": req.TaxID,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if req.Password != "" {
|
|||
|
|
updateData["password"] = req.Password
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := tx.Model(&bus_models.BusCooperative{}).Where("cooperative_number = ?", req.CooperativeNumber).
|
|||
|
|
Updates(updateData).Error; err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
return errors.New("更新合作商信息失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取数据库中的旧产品数据
|
|||
|
|
var existingProducts []bus_models.BusCooperativeProduct
|
|||
|
|
tx.Where("cooperative_id = ?", cooperative.ID).Find(&existingProducts)
|
|||
|
|
|
|||
|
|
// 旧数据转换为 map (ProductID => Discount)
|
|||
|
|
existingMap := make(map[uint64]float64)
|
|||
|
|
for _, p := range existingProducts {
|
|||
|
|
existingMap[p.ProductID] = p.Discount
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 计算需要更新、删除、插入的数据
|
|||
|
|
newProductMap := make(map[uint64]float64) // (ProductID => Discount)
|
|||
|
|
var newProducts []bus_models.BusCooperativeProduct
|
|||
|
|
for _, product := range req.Products {
|
|||
|
|
newProductMap[product.ProductID] = product.Discount
|
|||
|
|
|
|||
|
|
// 如果是新产品，或者折扣变化，则需要插入/更新
|
|||
|
|
if discount, exists := existingMap[product.ProductID]; !exists || discount != product.Discount {
|
|||
|
|
newProducts = append(newProducts, bus_models.BusCooperativeProduct{
|
|||
|
|
CooperativeID: cooperative.ID,
|
|||
|
|
ProductID: product.ProductID,
|
|||
|
|
Discount: product.Discount,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 删除不存在于新数据中的旧产品
|
|||
|
|
var productIDsToDelete []uint64
|
|||
|
|
for oldProductID := range existingMap {
|
|||
|
|
if _, exists := newProductMap[oldProductID]; !exists {
|
|||
|
|
productIDsToDelete = append(productIDsToDelete, oldProductID)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(productIDsToDelete) > 0 {
|
|||
|
|
tx.Where("cooperative_id = ? AND product_id IN ?", cooperative.ID, productIDsToDelete).Delete(&bus_models.BusCooperativeProduct{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 插入新的或有折扣变化的产品
|
|||
|
|
if len(newProducts) > 0 {
|
|||
|
|
if err := tx.Create(&newProducts).Error; err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
return errors.New("添加/更新合作商产品信息失败")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 提交事务
|
|||
|
|
if err := tx.Commit().Error; err != nil {
|
|||
|
|
return errors.New("提交事务失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DeleteCooperative 删除合作商
|
|||
|
|
func (e *CooperativeService) DeleteCooperative(req bus_models.DeleteCooperativeReq) error {
|
|||
|
|
// 执行删除操作
|
|||
|
|
if err := e.Orm.Where("cooperative_number = ?", req.CooperativeNumber).
|
|||
|
|
Delete(&bus_models.BusCooperative{}).Error; err != nil {
|
|||
|
|
return errors.New("删除产品失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetCooperativeDetail 获取合作商详细信息，包括产品信息
|
|||
|
|
func (e *CooperativeService) GetCooperativeDetail(ctx context.Context, req bus_models.CooperativeDetailReq) (
|
|||
|
|
bus_models.CooperativeDetailResp, error) {
|
|||
|
|
var resp bus_models.CooperativeDetailResp
|
|||
|
|
var cooperative bus_models.BusCooperative
|
|||
|
|
|
|||
|
|
// 查询合作商基本信息
|
|||
|
|
db := e.Orm.WithContext(ctx).Where("id = ?", req.CooperativeID)
|
|||
|
|
|
|||
|
|
if err := db.First(&cooperative).Error; err != nil {
|
|||
|
|
return resp, errors.New("查询合作商详情失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 查询该合作商的产品信息
|
|||
|
|
var cooperativeProducts []bus_models.BusCooperativeProduct
|
|||
|
|
err := e.Orm.WithContext(ctx).
|
|||
|
|
Where("cooperative_id = ?", req.CooperativeID).
|
|||
|
|
Find(&cooperativeProducts).Error
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return resp, errors.New("查询合作商产品信息失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 根据产品ID查询产品名称和编码
|
|||
|
|
var productIDs []uint64
|
|||
|
|
for _, cp := range cooperativeProducts {
|
|||
|
|
productIDs = append(productIDs, cp.ProductID)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 查询产品详细信息
|
|||
|
|
var products []bus_models.BusProduct
|
|||
|
|
if len(productIDs) > 0 {
|
|||
|
|
err = e.Orm.WithContext(ctx).
|
|||
|
|
Where("id IN (?)", productIDs).
|
|||
|
|
Find(&products).Error
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return resp, errors.New("查询产品信息失败")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将产品信息填充到响应中
|
|||
|
|
var productDetails []bus_models.ProductDetail
|
|||
|
|
for _, cp := range cooperativeProducts {
|
|||
|
|
// 查找对应产品
|
|||
|
|
for _, p := range products {
|
|||
|
|
if p.ID == cp.ProductID { // 确保ID匹配
|
|||
|
|
productDetails = append(productDetails, bus_models.ProductDetail{
|
|||
|
|
ProductCode: p.ProductCode,
|
|||
|
|
ProductName: p.ProductName,
|
|||
|
|
Discount: cp.Discount,
|
|||
|
|
})
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 填充合作商信息
|
|||
|
|
resp.CooperativeNumber = cooperative.CooperativeNumber
|
|||
|
|
resp.CooperativeName = cooperative.CooperativeName
|
|||
|
|
resp.Contact = cooperative.Contact
|
|||
|
|
resp.Tel = cooperative.Tel
|
|||
|
|
resp.Status = cooperative.Status
|
|||
|
|
resp.Account = cooperative.Account
|
|||
|
|
resp.Balance = cooperative.Balance
|
|||
|
|
resp.Free = cooperative.Free
|
|||
|
|
resp.Bond = cooperative.Bond
|
|||
|
|
resp.Products = productDetails
|
|||
|
|
|
|||
|
|
return resp, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateCooperativeNumber 生成唯一的合作商编号，日期 + 10位随机数
|
|||
|
|
func GenerateCooperativeNumber(db *gorm.DB) (string, error) {
|
|||
|
|
// 获取当前日期（年月日）
|
|||
|
|
dateStr := time.Now().Format("20060102") // 格式为：YYYYMMDD
|
|||
|
|
|
|||
|
|
// 生成一个10位随机数
|
|||
|
|
rand.Seed(time.Now().UnixNano()) // 设置随机数种子
|
|||
|
|
randomNum := rand.Int63n(10000000000) // 生成一个0到9999999999之间的随机数
|
|||
|
|
|
|||
|
|
// 将随机数转为固定长度的字符串，确保10位
|
|||
|
|
randomNumStr := fmt.Sprintf("%010d", randomNum) // 确保为10位，不足补零
|
|||
|
|
|
|||
|
|
// 拼接日期和随机数
|
|||
|
|
cooperativeNumber := fmt.Sprintf("%s%s", dateStr, randomNumStr)
|
|||
|
|
|
|||
|
|
// 检查生成的合作商编号是否已存在
|
|||
|
|
var count int64
|
|||
|
|
err := db.Model(&bus_models.BusCooperative{}).Where("cooperative_number = ?", cooperativeNumber).Count(&count).Error
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果已存在，重新生成
|
|||
|
|
if count > 0 {
|
|||
|
|
return GenerateCooperativeNumber(db) // 递归重新生成
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return cooperativeNumber, nil
|
|||
|
|
}
|