zhangqian
2023-10-31 5f0170df9787c6f3ff17cc168b3f2e3f511453f6
model/client.go
@@ -3,6 +3,7 @@
import (
   "aps_crm/constvar"
   "aps_crm/pkg/mysqlx"
   "fmt"
   "gorm.io/gorm"
   "time"
)
@@ -70,6 +71,9 @@
   if slf.Name != "" {
      db.Where("name = ?", slf.Name)
   }
   if slf.Number != "" {
      db.Where("number = ?", slf.Number)
   }
   if len(slf.SearchMap) > 0 {
      for key, value := range slf.SearchMap {
@@ -124,6 +128,10 @@
               } else {
                  db = db.Where("clients.member_id = ?", constvar.UserTypeSuper)
               }
            }
         case []int:
            if key == "member_ids" {
               db = db.Where("clients.member_id in ?", v)
            }
         }
@@ -196,6 +204,30 @@
   return db.Updates(data).Error
}
func (slf *ClientSearch) Count() (int64, error) {
   var db = slf.build()
   var total int64
   err := db.Count(&total).Error
   return total, err
}
func (slf *ClientSearch) MaxAutoIncr() (int, error) {
   type Result struct {
      Max int
   }
   var (
      result Result
      db     = slf.build()
   )
   err := db.Select("MAX(id) as max").Scan(&result).Error
   if err != nil {
      return result.Max, fmt.Errorf("max err: %v", err)
   }
   return result.Max, nil
}
func (slf *ClientSearch) SetPage(page, size int) *ClientSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf