kpt-pasture/module/backend/goods.go

package backend

import (
	"context"
	"fmt"
	"kpt-pasture/model"
	"kpt-pasture/util"
	"net/http"
	"time"

	"gorm.io/gorm"

	"gitee.com/xuyiping_admin/pkg/xerr"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
)

func (s *StoreEntry) DrugsList(ctx context.Context, req *pasturePb.SearchDrugsRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchDrugsResponse, error) {
	drugsList := make([]*model.Drugs, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.Drugs))
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if req.Usage > 0 {
		pref.Where("usage_method = ?", req.Usage)
	}

	if req.CategoryId > 0 {
		pref.Where("category_id = ?", req.CategoryId)
	}

	if err := pref.Order("id desc").Count(&count).Limit(int(pagination.PageSize)).Offset(int(pagination.PageOffset)).
		Find(&drugsList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.SearchDrugsResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchDrugsData{
			List:     model.DrugsSlice(drugsList).ToPB(),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) DrugsCreateOrUpdate(ctx context.Context, req *pasturePb.SearchDrugsList) error {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("登录人信息失效")
	}
	req.CategoryName = s.DrugCategoryMaps()[req.CategoryId]
	req.UnitName = s.UnitMap()[req.Unit]
	req.UnitName = s.DrugUsageMaps()[req.Usage]

	newDrugs := model.NewDrugs(req, currentUser)
	if req.Id <= 0 {
		if err = s.DB.Create(newDrugs).Error; err != nil {
			return xerr.WithStack(err)
		}
	} else {
		if err = s.DB.Where("id = ?", req.Id).Updates(newDrugs).Error; err != nil {
			return xerr.WithStack(err)
		}
	}

	return nil
}

func (s *StoreEntry) MedicalEquipmentList(ctx context.Context, req *pasturePb.SearchMedicalEquipmentRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchMedicalEquipmentResponse, error) {
	medicalEquipmentList := make([]*model.MedicalEquipment, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.MedicalEquipment))
	if req.Name != "" {
		pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
	}

	if err := pref.Order("id desc").Count(&count).Limit(int(pagination.PageSize)).Offset(int(pagination.PageOffset)).
		Find(&medicalEquipmentList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	unitMap := s.UnitMap()
	return &pasturePb.SearchMedicalEquipmentResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchMedicalEquipmentData{
			List:     model.MedicalEquipmentSlice(medicalEquipmentList).ToPB(unitMap),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) MedicalEquipmentCreateOrUpdate(ctx context.Context, req *pasturePb.SearchMedicalEquipmentList) error {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("登录人信息失效")
	}
	newDrugs := model.NewMedicalEquipment(req, currentUser)
	if req.Id <= 0 {
		if err = s.DB.Create(newDrugs).Error; err != nil {
			return xerr.WithStack(err)
		}
	} else {
		if err = s.DB.Where("id = ?", req.Id).Updates(newDrugs).Error; err != nil {
			return xerr.WithStack(err)
		}
	}
	return nil
}

func (s *StoreEntry) NeckRingLogCreateOrUpdate(ctx context.Context, req *pasturePb.NeckRingCreateRequest) error {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("登录人信息失效")
	}

	if req.Items == nil || len(req.Items) == 0 {
		return xerr.Custom("请选择要脖环数据")
	}

	newNeckRingLogList := model.NewNeckRingLogList(req.Items, currentUser)
	cowIds := make([]int64, 0)
	for _, v := range newNeckRingLogList {
		cowIds = append(cowIds, v.CowId)
	}

	if err = s.DB.Transaction(func(tx *gorm.DB) error {
		// 解绑脖环号
		if err = tx.Model(new(model.NeckRingLog)).
			Where("cowId IN ?", cowIds).
			Updates(map[string]interface{}{
				"unbind_at":      time.Now().Unix(),
				"status":         pasturePb.NeckRingStatus_Unbind,
				"operation_id":   currentUser.Id,
				"operation_name": currentUser.Name,
			}).Error; err != nil {
			return xerr.WithStack(err)
		}
		// 绑定新脖环
		if err = tx.Create(newNeckRingLogList).Error; err != nil {
			return xerr.WithStack(err)
		}

		for _, v := range req.Items {
			if v.CowId > 0 {
				if err = tx.Model(new(model.Cow)).
					Where("id = ?", v.CowId).
					Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
					Updates(map[string]interface{}{
						"neck_ring_number": v.Number,
					}).Error; err != nil {
					return xerr.WithStack(err)
				}
			}
		}

		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) NeckRingLogList(ctx context.Context, req *pasturePb.SearchNeckRingRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchNeckRingResponse, error) {
	neckRingLogList := make([]*model.NeckRingLog, 0)
	var count int64 = 0

	pref := s.DB.Model(new(model.NeckRingLog)).Where("status > ?", pasturePb.NeckRingStatus_Unbind)
	if req.Status > 0 {
		pref.Where("status = ?", req.Status)
	}

	if req.CowId > 0 {
		pref.Where("cow_id = ?", req.CowId)
	}

	if req.Number != "" {
		pref.Where("number like ?", fmt.Sprintf("%s%s%s", "%", req.Number, "%"))
	}

	if err := pref.Order("id desc").
		Count(&count).
		Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&neckRingLogList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	neckRingStatusMap := s.NeckRingStatusMap()
	return &pasturePb.SearchNeckRingResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchNeckRingData{
			List:     model.NeckRingLogSlice(neckRingLogList).ToPB(neckRingStatusMap),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) OutboundApply(ctx context.Context, req *pasturePb.OutboundApplyItem) error {
	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("登录人信息失效")
	}

	if len(req.Goods) <= 0 {
		return xerr.Custom("请选择要出库商品")
	}
	goodsItems := make([]*pasturePb.OutboundApplyGoodsItem, 0)
	switch req.OutType {
	case pasturePb.OutType_Drugs:
		for _, v := range req.Goods {
			if v.Quantity <= 0 {
				return xerr.Custom("请填写商品数量")
			}
			if v.GoodsId <= 0 {
				return xerr.Custom("请选择要出库商品")
			}

			newDrugs := &model.Drugs{}
			if err = s.DB.Model(new(model.Drugs)).
				Where("id = ?", v.GoodsId).
				Where("quantity >= ?", v.Quantity).
				First(newDrugs).Error; err != nil {
				return xerr.WithStack(err)
			}
			goodsItems = append(goodsItems, &pasturePb.OutboundApplyGoodsItem{
				GoodsId:  v.GoodsId,
				Quantity: v.Quantity,
				Unit:     v.Unit,
			})
		}
	case pasturePb.OutType_Medical_Equipment:
		for _, v := range req.Goods {
			if v.Quantity <= 0 {
				return xerr.Custom("请填写商品数量")
			}
			if v.GoodsId <= 0 {
				return xerr.Custom("请选择要出库商品")
			}

			newMedicalEquipment := &model.MedicalEquipment{}
			if err = s.DB.Model(new(model.Drugs)).
				Where("id = ?", v.GoodsId).
				Where("quantity >= ?", v.Quantity).
				First(newMedicalEquipment).Error; err != nil {
				return xerr.WithStack(err)
			}
			goodsItems = append(goodsItems, &pasturePb.OutboundApplyGoodsItem{
				GoodsId:  v.GoodsId,
				Quantity: v.Quantity,
				Unit:     v.Unit,
			})
		}
	default:
		return xerr.Custom("未知的出库类型")
	}
	unitMap := s.UnitMap()
	outbound := model.NewOutbound(req, currentUser)
	outboundLog := model.NewOutboundLogList(goodsItems, unitMap)
	if err = s.DB.Transaction(func(tx *gorm.DB) error {

		// 创建出库申请
		if err = tx.Create(outbound).Error; err != nil {
			return xerr.WithStack(err)
		}

		if err = tx.Create(outboundLog).Error; err != nil {
			return xerr.WithStack(err)
		}
		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}
	return nil
}

func (s *StoreEntry) OutboundList(ctx context.Context, req *pasturePb.SearchOutboundApplyRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchOutboundApplyResponse, error) {
	startUnix := util.TimeParseLocalUnix(req.StartDayTime)
	endUnix := util.TimeParseLocalEndUnix(req.EndDayTime)

	var count int64 = 0
	outboundList := make([]*model.Outbound, 0)
	pref := s.DB.Model(new(model.Outbound))
	if req.OutType > 0 {
		pref.Where("out_type = ?", req.OutType)
	}

	if startUnix > 0 && endUnix > 0 && startUnix <= endUnix {
		pref.Where("applicant_at BETWEEN ? AND ?", startUnix, endUnix)
	}

	if req.Number != "" {
		pref.Where("number like ?", fmt.Sprintf("%s%s%s", "%", req.Number, "%"))
	}

	if req.AuditStatus > 0 {
		pref.Where("audit_status = ?", req.AuditStatus)
	}

	if req.ApplicantId > 0 {
		pref.Where("applicant_id = ?", req.ApplicantId)
	}

	if req.ExamineId > 0 {
		pref.Where("examine_id = ?", req.ExamineId)
	}

	if err := pref.Order("id desc").
		Count(&count).
		Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&outboundList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	outTypeMap := s.OutTypeMap()
	auditStatusMap := s.AuditStatusMap()

	return &pasturePb.SearchOutboundApplyResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchOutboundApplyData{
			List:     model.OutboundSlice(outboundList).ToPB(outTypeMap, auditStatusMap),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) OutboundAudit(ctx context.Context, req *pasturePb.OutboundApplyAuditRequest) error {
	outbound, err := s.GetOutboundById(ctx, int64(req.Id))
	if err != nil {
		return xerr.WithStack(err)
	}

	if outbound == nil {
		return xerr.Custom("出库单不存在")
	}

	if req.AuditStatus != pasturePb.AuditStatus_Pass && req.AuditStatus != pasturePb.AuditStatus_Reject && req.AuditStatus != pasturePb.AuditStatus_Cancel {
		return xerr.Custom("审核状态异常")
	}

	if outbound.AuditStatus != pasturePb.AuditStatus_Pending {
		return xerr.Custom("异常出库单")
	}

	currentUser, err := s.GetCurrentSystemUser(ctx)
	if err != nil {
		return xerr.Custom("登录人信息失效")
	}

	outboundLogs, err := s.GetOutboundLogsByOutboundId(ctx, int64(outbound.Id))
	if err != nil {
		return xerr.WithStack(err)
	}

	if len(outboundLogs) <= 0 {
		return xerr.Custom("出库单商品不存在")
	}

	if err = s.DB.Transaction(func(tx *gorm.DB) error {
		if err = tx.Model(outbound).
			Where("id = ?", outbound.Id).
			Updates(map[string]interface{}{
				"audit_status":    req.AuditStatus,
				"examine_id":      currentUser.Id,
				"examine_name":    currentUser.Name,
				"examine_remarks": req.ExamineRemarks,
				"examine_at":      time.Now().Unix(),
			}).Error; err != nil {
			return xerr.WithStack(err)
		}

		if req.AuditStatus != pasturePb.AuditStatus_Pass {
			return nil
		}

		tableName := ""
		switch outbound.OutType {
		case pasturePb.OutType_Drugs:
			tableName = new(model.Drugs).TableName()
		case pasturePb.OutType_Medical_Equipment:
			tableName = new(model.MedicalEquipment).TableName()
		default:
			return nil
		}

		for _, v := range outboundLogs {
			if err = tx.Table(tableName).
				Where("id = ?", v.GoodsId).
				Updates(map[string]interface{}{
					"quantity": gorm.Expr("quantity - ?", v.Quantity),
				}).Error; err != nil {
				return xerr.WithStack(err)
			}
		}
		return nil
	}); err != nil {
		return xerr.WithStack(err)
	}

	return nil
}

func (s *StoreEntry) OutboundDetail(ctx context.Context, id int64) (*pasturePb.OutboundDetailResponse, error) {
	outbound, err := s.GetOutboundById(ctx, id)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	outboundLogs, err := s.GetOutboundLogsByOutboundId(ctx, id)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	outTypeMap := s.OutTypeMap()
	auditStatusMap := s.AuditStatusMap()
	applicantAtFormat, examineAtFormat := "", ""
	if outbound.ApplicantAt > 0 {
		applicantAtFormat = time.Unix(outbound.ApplicantAt, 0).Format(model.LayoutTime)
	}
	if outbound.ExamineAt > 0 {
		examineAtFormat = time.Unix(outbound.ExamineAt, 0).Format(model.LayoutTime)
	}
	return &pasturePb.OutboundDetailResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.OutboundApplyDetail{
			Id:                outbound.Id,
			Number:            outbound.Number,
			OutType:           outbound.OutType,
			OutTypeName:       outTypeMap[outbound.OutType],
			AuditStatus:       outbound.AuditStatus,
			AuditStatusName:   auditStatusMap[outbound.AuditStatus],
			ApplicantName:     outbound.ApplicantName,
			ApplicantRemarks:  outbound.ApplicantRemarks,
			ExamineName:       outbound.ExamineName,
			ExamineRemarks:    outbound.ExamineRemarks,
			ApplicantAtFormat: applicantAtFormat,
			ExamineAtFormat:   examineAtFormat,
			GoodsItem: &pasturePb.OutboundApplyItem{
				OutType:          outbound.OutType,
				Goods:            model.OutboundLogSlice(outboundLogs).ToPB(),
				ApplicantRemarks: outbound.ApplicantRemarks,
			},
		},
	}, nil
}

func (s *StoreEntry) FrozenSemenList(ctx context.Context, req *pasturePb.FrozenSemenRequest, pagination *pasturePb.PaginationModel) (*pasturePb.FrozenSemenResponse, error) {
	frozenSemenList := make([]*model.FrozenSemen, 0)
	var count int64 = 0
	pref := s.DB.Table(new(model.FrozenSemen).TableName())
	if req.BullId != "" {
		pref.Where("bull_id = ?", req.BullId)
	}

	if req.Producer != "" {
		pref.Where("producer = ?", req.Producer)
	}

	if err := pref.Order("id desc").
		Count(&count).Limit(int(pagination.PageSize)).
		Offset(int(pagination.PageOffset)).
		Find(&frozenSemenList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	frozenSemenTypeMap := s.FrozenSemenTypeMap()
	unitMap := s.UnitMap()

	return &pasturePb.FrozenSemenResponse{
		Code:    http.StatusOK,
		Message: "ok",
		Data: &pasturePb.SearchFrozenSemenData{
			List:     model.FrozenSemenSlice(frozenSemenList).ToPB(frozenSemenTypeMap, unitMap),
			Total:    int32(count),
			PageSize: pagination.PageSize,
			Page:     pagination.Page,
		},
	}, nil
}

func (s *StoreEntry) FrozenSemenCreate(ctx context.Context, req *pasturePb.SearchFrozenSemenList) error {
	currentUser, _ := s.GetCurrentSystemUser(ctx)
	req.CowKindName = s.CowKindMap()[req.CowKind]
	newFrozenSemen := model.NewFrozenSemen(req, currentUser)
	if err := s.DB.Create(newFrozenSemen).Error; err != nil {
		return xerr.WithStack(err)
	}
	return nil
}