model/model.go

@@ -1,6 +1,7 @@
package model

import (
    "fmt"
    "gorm.io/gorm"
    "time"
)
@@ -30,6 +31,7 @@
    CreateTime  string `json:"createTime"  gorm:"-"`
    UpdateTime  string `json:"updateTime"  gorm:"-"`
    CreatorName string `json:"creatorName" gorm:"-"`
    Number      string `json:"number"`
}

func (slf *CrmModel) AfterFind(tx *gorm.DB) (err error) {
@@ -38,3 +40,9 @@
    slf.UpdateTime = slf.UpdatedAt.Format("2006-01-02 15:04")
    return nil
}

func (slf *CrmModel) SetNumber(prefix string) {
    if slf.Number == "" {
        slf.Number = fmt.Sprintf("%v%v", prefix, slf.ID)
    }
}

 model/receipt.go

@@ -28,7 +28,7 @@
        BankAccount   BankAccount                `gorm:"foreignKey:BankAccountId" json:"bankAccount"`
        Remark        string                     `gorm:"column:remark;type:varchar(255);not null;default '';comment:备注" json:"remark"` // 备注
        FileId        int                        `gorm:"column:file_id;type:int;not null;default 0;comment:附件id" json:"fileId"`        // 附件id
        gorm.Model    `json:"-"`
        CrmModel
    }

    // ReceiptSearch 收款单搜索条件
@@ -43,10 +43,19 @@
    }
)

func (Receipt) TableName() string {
func (slf *Receipt) TableName() string {
    return "receipt"
}

func (slf *Receipt) AfterFind(db *gorm.DB) (err error) {
    err = slf.CrmModel.AfterFind(db)
    if slf.CrmModel.ID == 0 {
        slf.CrmModel.ID = uint(slf.Id)
    }
    slf.CrmModel.SetNumber(constvar.NumberPrefixOfReceipt)
    return nil
}

func NewReceiptSearch() *ReceiptSearch {
    return &ReceiptSearch{
        Orm: mysqlx.GetDB(),