opms_backend/opms_parent/app/service/proj/business.go

package proj

import (
	"context"
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/util/gconv"

	projDao "dashoo.cn/micro/app/dao/proj"
	model "dashoo.cn/micro/app/model/proj"
	"dashoo.cn/micro/app/service"
)

type businessService struct {
	*service.ContextService
	Dao *projDao.ProjBusinessDao
}

func NewBusinessService(ctx context.Context) (svc *businessService, err error) {
	svc = new(businessService)
	if svc.ContextService, err = svc.Init(ctx); err != nil {
		return nil, err
	}
	svc.Dao = projDao.NewProjBusinessDao(svc.Tenant)
	return svc, nil
}

func (p *businessService) GetList(req *model.ProjBusinessSearchReq) (total int, businessList []*model.ProjBusiness, err error) {
	db := p.Dao.M
	if req.NboName != "" {
		db = db.WhereLike(p.Dao.Columns.NboName, "%"+req.NboName+"%")
	}
	if req.CustName != "" {
		db = db.WhereLike(p.Dao.Columns.CustName, "%"+req.CustName+"%")
	}
	if req.SaleName != "" {
		db = db.WhereLike(p.Dao.Columns.SaleName, "%"+req.SaleName+"%")
	}
	if req.NboType != "" {
		db = db.Where(p.Dao.Columns.NboType, req.NboType)
	}
	total, err = db.Count()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取总行数失败")
		return
	}

	err = db.Page(req.PageNum, req.PageSize).Order("id asc").Scan(&businessList)
	return
}

func (p *businessService) GetEntityById(id int64) (business *model.ProjBusiness, err error) {
	err = p.Dao.Where(projDao.ProjBusiness.Columns.Id, id).Scan(&business)
	return
}

func (p *businessService) GetBusinessProduct(id int64) (productList []*model.ProjBusinessProduct, err error) {
	productDao := projDao.NewProjBusinessProductDao(p.Tenant)
	err = productDao.Where(productDao.ProjBusinessProductDao.Columns.BusId, id).Scan(&productList)
	return
}

func (p *businessService) Create(req *model.AddProjBusinessReq) (err error) {
	businessData := new(model.ProjBusiness)
	if err = gconv.Struct(req, businessData); err != nil {
		return
	}
	products := make([]*model.ProjBusinessProduct, len(req.Products))
	if err = gconv.Structs(req.Products, &products); err != nil {
		return
	}

	service.SetCreatedInfo(businessData, p.GetCxtUserId(), p.GetCxtUserName())
	err = p.Dao.Transaction(context.TODO(), func(ctx context.Context, tx *gdb.TX) error {
		res, err := p.Dao.TX(tx).Insert(businessData)
		if err != nil {
			return err
		}
		lastId, _ := res.LastInsertId()
		for _, v := range products {
			v.BusId = int(lastId)
			v.TotalPrice = v.SellingPrice * v.Quantity
			service.SetCreatedInfo(v, p.GetCxtUserId(), p.GetCxtUserName())
		}
		productDao := projDao.NewProjBusinessProductDao(p.Tenant)
		_, err = productDao.TX(tx).Insert(products)
		return err
	})

	return
}

func (p *businessService) UpdateById(req *model.UpdateProjBusinessReq) (err error) {
	record, err := p.Dao.Count("Id", req.Id)
	if err != nil || record == 0 {
		err = gerror.New("该数据不存在")
		return err
	}
	businessData := new(model.ProjBusiness)
	if err = gconv.Struct(req, businessData); err != nil {
		return
	}
	service.SetUpdatedInfo(businessData, p.GetCxtUserId(), p.GetCxtUserName())
	_, err = p.Dao.FieldsEx(service.UpdateFieldEx).WherePri(projDao.ProjBusiness.Columns.Id, req.Id).Update(businessData)
	return
}

func (p *businessService) DeleteByIds(ids []int64) (err error) {
	_, err = p.Dao.WhereIn(projDao.ProjBusiness.Columns.Id, ids).Delete()
	return
}