lc_finance/server/dao/reimbursement.go

package dao

import (
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
	"server/global"
)

type Reimbursement struct {
	global.GVA_MODEL
	ProjectCode string          `json:"projectCode" form:"projectCode" gorm:"comment:项目编号"`
	Name        string          `json:"name" form:"name" gorm:"comment:项目名称"`
	Applicant   string          `json:"applicant" form:"applicant" gorm:"comment:报销申请人"`
	Approval    string          `json:"approval" form:"approval" gorm:"comment:报销审批人"`
	Price       decimal.Decimal `json:"price" form:"price" gorm:"comment:项目金额"`
	Genre       string          `json:"genre" form:"genre" gorm:"comment:项目类型"`
	FeeDetails  []FeeDetails    `json:"feeDetails" form:"feeDetails" gorm:"-"`
}

func (Reimbursement) TableName() string {
	return "reimbursement"
}

// TODO:报销查询

// QueryReimbursementList 查询报销
func QueryReimbursementList(limit, offset int, code, name string) (reimbursement []Reimbursement, total int64, err error) {
	// 创建db
	db := global.GVA_DB.Model(&Project{}).Where("project_code = ?", code)
	// 如果有条件搜索 下方会自动创建搜索语句
	if name != "" {
		db = db.Where("name LIKE ?", "%"+name+"%")
	}
	err = db.Count(&total).Error
	if err != nil {
		return
	}
	err = db.Order("id desc").Limit(limit).Offset(offset).Find(&reimbursement).Error
	return reimbursement, total, err
}

func QueryReimbursement(code string) (reimbursement []Reimbursement, err error) {
	err = global.GVA_DB.Model(&Reimbursement{}).Where("project_code = ?", code).Find(&reimbursement).Error
	return reimbursement, err
}

// TODO:报销新增

// CreateReimbursement 新增报销单
func (r Reimbursement) CreateReimbursement() error {
	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		txErr := tx.Create(&r).Error
		if txErr != nil {
			return txErr
		}
		for _, feeDetail := range r.FeeDetails {
			txErr := tx.Create(&feeDetail).Error
			if txErr != nil {
				return txErr
			}
		}
		return txErr
	})
}

// TODO:报销修改

// UpdateReimbursement 修改报销单
func (r Reimbursement) UpdateReimbursement() error {
	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		txErr := tx.Unscoped().Where("reimbursement_id = ?", r.ID).Delete(&FeeDetails{}).Error
		if txErr != nil {
			return txErr
		}
		txErr = tx.Model(&r).Updates(&r).Error
		if txErr != nil {
			return txErr
		}
		for _, feeDetail := range r.FeeDetails {
			txErr := tx.Create(&feeDetail).Error
			if txErr != nil {
				return txErr
			}
		}
		return txErr
	})
}

// TODO:报销删除

// DeleteReimbursement 删除报销单
func (r Reimbursement) DeleteReimbursement() error {
	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		txErr := tx.Unscoped().Where("reimbursement_id = ?", r.ID).Delete(&FeeDetails{}).Error
		if txErr != nil {
			return txErr
		}
		txErr = tx.Unscoped().Delete(&r).Error
		if txErr != nil {
			return txErr
		}
		return txErr
	})
}