From 7438f17c0b3a9f79ee5890c554bedc64c362859b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 13 十月 2023 14:04:59 +0800 Subject: [PATCH] 销售计划简单数据权限 --- model/plan.go | 46 ++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 38 insertions(+), 8 deletions(-) diff --git a/model/plan.go b/model/plan.go index d6d0ae5..0ba2f15 100644 --- a/model/plan.go +++ b/model/plan.go @@ -3,7 +3,6 @@ import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" - "time" ) type ( @@ -16,8 +15,8 @@ SubOrder SubOrder `json:"subOrder" gorm:"foreignKey:SubOrderId"` SalesDetailsId int `json:"salesDetailsId" gorm:"column:sales_details_id;type:int;comment:閿�鍞槑缁唅d"` SalesDetails SalesDetails `json:"salesDetails" gorm:"foreignKey:SalesDetailsId"` - StartTime time.Time `json:"startTime" gorm:"column:start_time;type:datetime;comment:寮�濮嬫椂闂�"` - EndTime time.Time `json:"endTime" gorm:"column:end_time;type:datetime;comment:缁撴潫鏃堕棿"` + StartTime *CustomTime `json:"startTime" gorm:"column:start_time;type:datetime;comment:寮�濮嬫椂闂�"` + EndTime *CustomTime `json:"endTime" gorm:"column:end_time;type:datetime;comment:缁撴潫鏃堕棿"` Content string `json:"content" gorm:"column:content;type:varchar(255);comment:璁″垝鍐呭"` File string `json:"file" gorm:"column:file;type:varchar(255);comment:闄勪欢"` gorm.Model `json:"-"` @@ -25,7 +24,12 @@ PlanSearch struct { Plan - Orm *gorm.DB + + Orm *gorm.DB + Keyword string + OrderBy string + PageNum int + PageSize int } ) @@ -41,6 +45,9 @@ func (slf *PlanSearch) build() *gorm.DB { var db = slf.Orm.Model(&Plan{}) + if slf.Keyword != "" { + db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") + } if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } @@ -70,14 +77,37 @@ return record, err } -func (slf *PlanSearch) FindAll() ([]*Plan, error) { +func (slf *PlanSearch) FindAll() ([]*Plan, int64, error) { var db = slf.build() - var record = make([]*Plan, 0) - err := db.Find(&record).Error - return record, err + var records = make([]*Plan, 0) + var total int64 + if err := db.Count(&total).Error; err != nil { + return records, total, err + } + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } + + err := db.Order("id desc").Find(&records).Error + return records, total, err } func (slf *PlanSearch) SetId(id int) *PlanSearch { slf.Id = id return slf } + +func (slf *PlanSearch) SetKeyword(keyword string) *PlanSearch { + slf.Keyword = keyword + return slf +} + +func (slf *PlanSearch) SetPage(page, size int) *PlanSearch { + slf.PageNum, slf.PageSize = page, size + return slf +} + +func (slf *PlanSearch) SetOrder(order string) *PlanSearch { + slf.OrderBy = order + return slf +} -- Gitblit v1.8.0