package dao import ( "enterprise/common/model" "gorm.io/gorm" "time" ) type ApprovalCheckinDao struct { } func NewApprovalCheckinDao() *ApprovalCheckinDao { return &ApprovalCheckinDao{} } func (d *ApprovalCheckinDao) TableName() string { return "approval_checkin" } func (d *ApprovalCheckinDao) Create(o *model.ApprovalCheckin) (int64, error) { o.CreateTime = time.Now().Unix() res := GetDB().Table(d.TableName()).Create(o) return o.Id, res.Error } func (d *ApprovalCheckinDao) Update(o *model.ApprovalCheckin) error { o.UpdateTime = time.Now().Unix() tx := GetDB().Table(d.TableName()) res := tx.Save(o) return res.Error } func (d *ApprovalCheckinDao) Delete(id int64) error { res := GetDB().Table(d.TableName()).Delete(&model.ApprovalCheckin{}, id) return res.Error } func (d *ApprovalCheckinDao) Get(id int64) (*model.ApprovalCheckin, error) { var u model.ApprovalCheckin tx := GetDB().Table(d.TableName()) tx = tx.Where("id = ?", id) res := tx.First(&u) if res.Error == gorm.ErrRecordNotFound { return nil, nil } if res.Error != nil { return nil, res.Error } return &u, nil } func (d *ApprovalCheckinDao) GetBySpNo(spNo string) (*model.ApprovalCheckin, error) { var u model.ApprovalCheckin tx := GetDB().Table(d.TableName()) tx = tx.Where("sp_no = ?", spNo) res := tx.First(&u) if res.Error == gorm.ErrRecordNotFound { return nil, nil } if res.Error != nil { return nil, res.Error } return &u, nil } func (d *ApprovalCheckinDao) GetByUsernameMonth(username, month string) ([]*model.ApprovalCheckin, error) { var u []*model.ApprovalCheckin tx := GetDB().Table(d.TableName()) tx = tx.Where("month = ?", month) tx = tx.Where("username = ?", username) res := tx.Find(&u) if res.Error != nil { return nil, res.Error } return u, nil } func (d *ApprovalCheckinDao) GetByUsernameDay(username, day string) (*model.ApprovalCheckin, error) { var u model.ApprovalCheckin tx := GetDB().Table(d.TableName()) tx = tx.Where("checkin_date = ?", day) tx = tx.Where("username = ?", username) res := tx.First(&u) if res.Error == gorm.ErrRecordNotFound { return nil, nil } if res.Error != nil { return nil, res.Error } return &u, nil }