package models
|
|
import (
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/logger.git"
|
"strconv"
|
"strings"
|
)
|
|
type FileStack struct {
|
Id string `gorm:"column:id;primary_key" json:"id"`
|
Name string `gorm:"column:name" json:"name"`
|
Type int `gorm:"column:type" json:"type"`
|
Enable bool `gorm:"column:enable" json:"enable"`
|
IsAutoDelFile bool `gorm:"column:isAutoDelFile" json:"isAutoDelFile"`
|
Status int `gorm:"column:status;default:1" json:"status"`
|
CreateTime string `gorm:"column:createTime" json:"createTime"`
|
UpdateTime string `gorm:"column:updateTime" json:"updateTime"`
|
Sort int `gorm:"column:sort" json:"sort"`
|
ResolutionWidth int `gorm:"column:resolution_width;default:0" json:"resolution_width"`//分辨率宽
|
ResolutionHeight int `gorm:"column:resolution_height;default:0" json:"resolution_height"`//分辨率高
|
}
|
|
const (
|
Id_Stack_Pre = "stack_"
|
Stack_Status_NoRule = 0 //未配规则
|
Stack_Status_Wait = 1 //等待处理
|
Stack_Status_Doing = 2 //处理中
|
Stack_Status_Done = 9 //处理完成
|
)
|
|
func (FileStack) TableName() string {
|
return "file_stack"
|
}
|
|
func (fs *FileStack) Total(name string, typ int, filterCIds *string) int {
|
var total int
|
sql := "select count(1) from file_stack where 1=1"
|
if name !="" {
|
sql += " and name like '%"+name+"%'"
|
}
|
if typ > 0 {
|
if typ == 1 { //1.查处理完成
|
sql += " and status="+strconv.Itoa(bhomeclient.Stack_Status_Done)
|
} else if typ == 2 { //2.查处理中
|
sql += " and status="+strconv.Itoa(bhomeclient.Stack_Status_Doing)
|
} else if typ == 3 { //3.查未配规则
|
if *filterCIds != "" {
|
sql += " and id not in ("+*filterCIds+")"
|
}
|
} else if typ == 4 { //4.查未开启
|
sql += " and enable=0"
|
}
|
}
|
err := db.Raw(sql).Count(&total).Error
|
if err !=nil {
|
total = 0
|
}
|
return total
|
}
|
|
func (fs *FileStack) FindAllByPage(typ int,name string, page int, size int, filterCIds *string) (list []FileStack, err error) {
|
sql := "select * from file_stack where 1=1"
|
if name != "" {
|
sql += " and name like '%"+name+"%'"
|
}
|
if typ > 0 {
|
if typ == 1 { //1.查处理完成
|
sql += " and status="+strconv.Itoa(bhomeclient.Stack_Status_Done)
|
} else if typ == 2 { //2.查处理中
|
sql += " and status="+strconv.Itoa(bhomeclient.Stack_Status_Doing)
|
} else if typ == 3 { //3.查未配规则
|
if *filterCIds != "" {
|
sql += " and id not in ("+*filterCIds+")"
|
}
|
} else if typ == 4 { //4.查未开启
|
sql += " and enable=0"
|
}
|
}
|
offset := (page-1)*size
|
err = db.Raw(sql).Offset(offset).Limit(size).Find(&list).Error
|
return list,nil
|
}
|
|
func (fs *FileStack) FindAll() (list []FileStack, err error) {
|
err = db.Raw("select * from file_stack").Find(&list).Error
|
return
|
}
|
|
func (fs *FileStack) SelectById(id string) (rows int64, err error) {
|
result := db.Table(fs.TableName()).Where("id = ?", id).First(&fs)
|
if result.Error != nil || result.RowsAffected == 0 {
|
return 0, err
|
}
|
return result.RowsAffected, nil
|
}
|
|
func (fs *FileStack) DeleteById(id string) bool {
|
result := db.Exec("delete from file_stack where id='"+id+"'")
|
if result.Error != nil {
|
return false
|
}
|
if result.RowsAffected > 0 {
|
db.Exec("delete from file_analysis where stack_id='"+id+"'")
|
return true
|
}
|
return false
|
}
|
|
func (fs *FileStack) UpdateEnable(id string, enable bool) bool {
|
logger.Debug("models.FileStack UpdateEnable id:",id,"enable:",enable)
|
str := "0"
|
if enable {
|
str = "1"
|
}
|
sql := "update file_stack set enable ="+str+" where id='"+id+"'"
|
result := db.Exec(sql)
|
if result.Error != nil {
|
return false
|
}
|
if result.RowsAffected > 0 {
|
return true
|
} else {
|
return false
|
}
|
}
|
|
func (fs *FileStack) Add() bool {
|
result := db.Table(fs.TableName()).Create(&fs)
|
if result.Error ==nil && result.RowsAffected >0 {
|
return true
|
}
|
return false
|
}
|
|
func (fs *FileStack) Update() bool {
|
result := db.Table(fs.TableName()).Where("id=?",fs.Id).Update(&fs)
|
if result.Error ==nil && result.RowsAffected >0 {
|
return true
|
}
|
return false
|
}
|
|
//ids为正在做的数据栈列表
|
func (fs *FileStack) UpdateStatus(ids []string, status int) bool {
|
idStr := ""
|
for _,id := range ids {
|
idStr += "'"+id+"',"
|
}
|
idStr = strings.TrimRight(idStr, ",")
|
if idStr == "" {
|
return false
|
}
|
result := db.Exec("update file_stack set status=? where id in ("+idStr+")", status)
|
if result.Error ==nil && result.RowsAffected > 0 {
|
return true
|
}
|
return false
|
}
|
|
func (fs *FileStack) HavingWorkFiles(stackId string) bool {
|
var count int
|
err := db.Raw("select count(1) as count from file_analysis where stack_id=? and (status="+strconv.Itoa(bhomeclient.File_Status_Doing)+" or status="+strconv.Itoa(bhomeclient.File_Status_Wait)+")", stackId).Count(&count).Error
|
if err !=nil {
|
return false
|
}
|
return count > 0
|
}
|
