From 1fab8f0f684d28ac8b996b345402b45e116eaed9 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 30 四月 2024 17:30:51 +0800
Subject: [PATCH] 薪资报表查询与修改

---
 models/salary_report_form.go |   29 +++++++++++++++++++++++++++++
 1 files changed, 29 insertions(+), 0 deletions(-)

diff --git a/models/salary_report_form.go b/models/salary_report_form.go
index a03738b..4c2c08e 100644
--- a/models/salary_report_form.go
+++ b/models/salary_report_form.go
@@ -59,6 +59,16 @@
 	return slf
 }
 
+func (slf *SalaryReportFormSearch) SetMonth(month string) *SalaryReportFormSearch {
+	slf.Month = month
+	return slf
+}
+
+func (slf *SalaryReportFormSearch) SetWorkerId(workerId string) *SalaryReportFormSearch {
+	slf.WorkerId = workerId
+	return slf
+}
+
 func (slf *SalaryReportFormSearch) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -67,6 +77,12 @@
 	}
 	if slf.Keyword != "" {
 		db = db.Where("worker_name like ? or worker_id like ?", "%"+slf.Keyword+"%", "%"+slf.Keyword+"%")
+	}
+	if slf.Month != "" {
+		db = db.Where("month = ?", slf.Month)
+	}
+	if slf.WorkerId != "" {
+		db = db.Where("worker_id = ?", slf.WorkerId)
 	}
 
 	return db
@@ -159,6 +175,19 @@
 	return records, nil
 }
 
+func (slf *SalaryReportFormSearch) First() (*SalaryReportForm, error) {
+	var (
+		record = new(SalaryReportForm)
+		db     = slf.build()
+	)
+
+	if err := db.First(record).Error; err != nil {
+		return record, err
+	}
+
+	return record, nil
+}
+
 func (slf *SalaryReportFormSearch) Count() (int64, error) {
 	var (
 		total int64

--
Gitblit v1.8.0