zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/model.go
@@ -1,6 +1,7 @@
package model
import (
   "fmt"
   "gorm.io/gorm"
   "time"
)
@@ -20,19 +21,28 @@
   UpdatedAt time.Time `json:"updated_at"`
   // The date when the item was deleted
   // example: 2023-08-10 15:48:25
   DeletedAt *time.Time `json:"deleted_at,omitempty"`
   DeletedAt *time.Time `json:"-"`
}
type CrmModel struct {
   gorm.Model  `json:"-"`
   CreatorId   int       `json:"-" gorm:"column:creator_id;type:int;comment:创建人id"`
   Creator     User      `json:"-"  gorm:"foreignKey:CreatorId"`
   CreateTime  time.Time `json:"createTime"  gorm:"-"`
   CreatorName string    `json:"creatorName" gorm:"-"`
   CreatorId   int    `json:"-" gorm:"column:creator_id;type:int;comment:创建人id"`
   Creator     User   `json:"-"  gorm:"foreignKey:CreatorId"`
   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) {
   slf.CreatorName = slf.Creator.Username
   slf.CreateTime = slf.CreatedAt
   slf.CreateTime = slf.CreatedAt.Format("2006-01-02 15:04")
   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)
   }
}