// Copyright 2019 github.com. All rights reserved. // Use of this source code is governed by github.com. package model import ( "fmt" "github.com/jinzhu/gorm" "smart-supplier-management/consts" "time" ) type SlagcarDevice struct { ID int64 `gorm:"column:ID;PRIMARY_KEY" json:"id"` Code string `gorm:"column:Code" json:"code"` Name string `gorm:"column:name" json:"name"` DeviceNum string `gorm:"column:deviceNum" json:"devicenum"` DeviceModel string `gorm:"column:deviceModel" json:"devicemodel"` DeviceName string `gorm:"column:deviceName" json:"devicename"` Manufacturer string `gorm:"column:manufacturer" json:"manufacturer"` Batch string `gorm:"column:batch" json:"batch"` SN string `gorm:"column:SN" json:"sn"` SIM string `gorm:"column:SIM" json:"sim"` State int64 `gorm:"column:state" json:"state"` Unit string `gorm:"column:unit" json:"unit"` Person string `gorm:"column:contactName" json:"person"` Phone string `gorm:"column:contactPerson" json:"phone"` Remark string `gorm:"column:remark" json:"remark"` Longitude string `gorm:"column:longitude" json:"longitude"` Latitude string `gorm:"column:latitude" json:"latitude"` VerifyTime time.Time `gorm:"column:VerifyTime"` VerifyStatus int `gorm:"column:VerifyStatus"` Key string `gorm:"column:Key"` ProjectId int64 `gorm:"column:ProjectId"` ProviderId int64 `gorm:"column:ProviderId"` CreatedAt time.Time `gorm:"column:CreatedAt"` } func (SlagcarDevice) TableName() string { return "db_smart_v2.SlagcarDevice" } func (p *SlagcarDevice) Insert(db *gorm.DB) error { return db.Table(p.TableName()).Create(p).Error } func (p *SlagcarDevice) Del(db *gorm.DB, where map[string]interface{}) error { cond, val, err := whereBuild(where) if err != nil { return err } return db.Table(p.TableName()).Where(cond, val...).Delete(p).Error } func (p *SlagcarDevice) Find(db *gorm.DB, where map[string]interface{}) error { cond, val, err := whereBuild(where) if err != nil { return err } return db.Table(p.TableName()).Where(cond, val...).First(p).Error } func (p *SlagcarDevice) Update(db *gorm.DB, where map[string]interface{}, values map[string]interface{}) error { cond, val, err := whereBuild(where) if err != nil { return err } return db.Table(p.TableName()).Where(cond, val...).Updates(values).Error } func (p *SlagcarDevice) Save(db *gorm.DB) error { return db.Save(p).Error } func (p *SlagcarDevice) Count(db *gorm.DB, where map[string]interface{}) (int64, error) { if len(where) > 0 { cond, val, err := whereBuild(where) if err != nil { return 0, err } ret := int64(0) err = db.Table(p.TableName()).Where(cond, val...).Count(&ret).Error return ret, err } ret := int64(0) err := db.Table(p.TableName()).Count(&ret).Error return ret, err } func (p *SlagcarDevice) List(db *gorm.DB, where map[string]interface{}, page int) (list []SlagcarDevice, err error) { if len(where) > 0 { cond, val, err := whereBuild(where) if err != nil { return list, err } result := db.Table(p.TableName()).Where(cond, val...).Limit(PageSize).Offset(page).Find(&list) return list, result.Error } result := db.Table(p.TableName()).Limit(10).Offset(page).Find(&list) return list, result.Error } type SlagDeviceItem struct { Id int64 Sn string TypeCode int32 TypeName string ProjectName string ProjectId int64 SafetyRecordNo string CreatedTime time.Time Status int32 State int32 ProjectApproveTime time.Time Key string ProviderName string SocialCode string Name string Ip string Port int MediaTransport string ChannelCount int } func slagDeviceListSql(req SlagDeviceListRequest)(string, string, []interface{}) { sql := fmt.Sprintf("select t1.ID as id, t1.Key as 'key', t1.deviceName as name, t1.SN as sn, t1.VerifyTime, "+ "t1.CreatedAt as created_time, t1.state as state, t1.VerifyStatus as status, t2.SafetyNo as safety_record_no, t2.Name as project_name, t3.Name as provider_name, t3.SocialCode as social_code from DustDeviceInfo as t1 left join ProjectInfo as t2 on t1.ProjectId=t2.ID left join Provider as t3 on t1.ProviderId=t3.ID") countSql := fmt.Sprintf("select count(1) as count from DustDeviceInfo as t1 left join ProjectInfo as t2 on t1.ProjectID=t2.ID ") args := []interface{}{} whereArray := []string{} if req.ProviderId > 0 { whereArray = append(whereArray, fmt.Sprintf("t1.ProviderId=%d", req.ProviderId)) } if req.Filter != "" { whereArray = append(whereArray, fmt.Sprintf("(t1.SN like '%%%s%%' or t2.SafetyNo like '%%%s%%' or t2.Name like '%%%s%%')", req.Filter, req.Filter, req.Filter)) } if req.CanDel { req.IsAll = true req.StatusFilter = []int32{consts.DeviceStatusAddAuditted} whereArray = append(whereArray, fmt.Sprintf("(select count(1) from DeviceDelJob where DeviceId=t1.ID and (Status = 0 or Status = 1)) = 0")) } if len(req.StatusFilter) > 0 { args = append(args, req.StatusFilter) whereArray = append(whereArray, fmt.Sprintf("t1.VerifyStatus in(?)")) } where := "" for _, v := range whereArray { if where == "" { where = fmt.Sprintf(" where %s", v) continue } where = fmt.Sprintf("%s and %s", where, v) } offset := (req.Page - 1) *int32(PageSize) if req.IsAll { sql = fmt.Sprintf("%s %s order by t1.CreatedAt desc", sql, where) countSql = fmt.Sprintf("%s %s", countSql, where) } else { sql = fmt.Sprintf("%s %s order by t1.CreatedAt desc limit %d offset %d", sql, where, PageSize, offset) countSql = fmt.Sprintf("%s %s", countSql, where) } return sql, countSql, args } type SlagDeviceListRequest struct { ProviderId int64 Filter string StatusFilter []int32 IsAll bool CanDel bool Page int32 ProjectId int64 } func (p *SlagcarDevice) SlagDeviceList(db *gorm.DB, req SlagDeviceListRequest) ([]SlagDeviceItem, int64, error) { type ResultCount struct { Count int64 } array := []ResultCount{} ret := []SlagDeviceItem{} var err error sql, countSql, args := slagDeviceListSql(req) err = db.Raw(countSql, args...).Scan(&array).Error if err != nil { return nil, 0, err } if len(array) == 0 { return nil, 0, nil } if array[0].Count == 0 { return nil, 0, nil } err = db.Raw(sql, args...).Scan(&ret).Error if err != nil { return nil, array[0].Count, err } return ret, array[0].Count, nil }