package models type CameraRuleBase struct { CameraId string `gorm:"column:camera_id" json:"camera_id"` //摄像机id SdkId string `gorm:"column:sdk_id" json:"sdk_id"` //算法id PolygonId string `gorm:"column:polygon_id" json:"polygon_id"` //多边形id SdkSet string `gorm:"column:sdk_set" json:"sdk_set"` //算法id RuleWithPre string `gorm:"column:rule_with_pre" json:"rule_with_pre"` //与上一条记录的逻辑运算规则(&&,||) IsSaveAnyhow bool `gorm:"column:is_save_anyhow;default:0;" json:"is_save_anyhow"` //rule_with_pre是触发时,是否无论被触发的条件满不满足,都保留数据 GroupId string `gorm:"column:group_id" json:"group_id"` //分组id Sort int `gorm:"column:sort" json:"sort"` //排序 } type CameraRuleGroupArg struct { Id string `gorm:"column:id;primary_key;type:varchar(50);unique;not null;" json:"id"` CameraRuleBase } func (CameraRuleGroupArg) TableName() string { return "camera_rule_group_arg" } func (rga *CameraRuleGroupArg) InsertByGroup() bool { tx := db.Table(rga.TableName()).Begin() if tx.Error != nil { return false } if err := tx.Create(&rga).Error; err != nil { tx.Rollback() return false } if tx.Commit().Error == nil { return true } return false } func (rga *CameraRuleGroupArg) FindAll() (args []CameraRuleGroupArg) { if err := db.Table(rga.TableName()).Scan(&args).Error; err != nil { return nil } return args } type DbGroupId struct { GroupId string `json:"group_id"` } func (rga *CameraRuleGroupArg) FindGroupIdByCameraId(cameraId string) (groupIds []DbGroupId, err error) { if err := db.Raw("select distinct group_id from "+rga.TableName()+" where camera_id=?", cameraId).Scan(&groupIds).Error; err != nil { return nil, err } return groupIds, nil } //DeleteByGroup 根据group_id删除规则组设置 func (rga CameraRuleGroupArg) DeleteByGroup(groupId string) bool { var crg CameraRuleGroup var err error tx := db.Begin() defer func() { if err != nil && tx != nil { tx.Rollback() } }() if err = tx.Table(rga.TableName()).Where("group_id=?", groupId).Delete(CameraRuleGroupArg{}).Error; err != nil { return false } if err = tx.Table(crg.TableName()).Where("group_id=?", groupId).Delete(crg).Error; err != nil { return false } if err = tx.Commit().Error; err != nil { return false } else { return true } } func (rga CameraRuleGroupArg) FindByGroupId(groupId string) (rags []CameraRuleGroupArg) { if err := db.Table(rga.TableName()).Where("group_id=?", groupId).Find(&rags).Error; err != nil { return nil } return rags } //判断多边形是否在规则设置中正在使用 func (rga CameraRuleGroupArg) IsPolygonInUse(polygonId string) bool { var rules []CameraRuleGroupArg if err := db.Table(rga.TableName()).Where("polygon_id=?", polygonId).Find(&rules).Error; err != nil { return true } if len(rules) == 0 { return false } return true } func (rga CameraRuleGroupArg) ExistRunningTask(cameraId string) bool { var arr []CameraRuleGroupArg if err = db.Raw("select a.* from "+CameraRuleGroupArg{}.TableName()+" a join "+CameraRuleGroup{}.TableName()+" g on a.group_id=g.id where a.camera_id=? and g.enable=1",cameraId).Scan(&arr).Error;err !=nil{ return false } return len(arr) >0 }