scd_tools_go/service/models/bo/task_model.go

package bo

import (
	"errors"
	"fmt"
	"scd_check_tools/arrayex"
	"scd_check_tools/logger"
	"scd_check_tools/models/enum"
	"scd_check_tools/tools"
	"strconv"
	"strings"

	"github.com/astaxie/beego/orm"
)

//检测任务模型
type T_data_task_model struct {
	Id      int    `orm:"pk"`
	TaskId  int    // '任务ID' ,
	AreaId  int    // 间隔ID
	ModelId int    // '模型ID' ,
	Cr      int    // '创建人' ,
	Ct      string `orm:"-"` // '创建时间' ,
	Ur      int    // '更新人' ,
	Ut      string `orm:"-"` // '更新时间'
}

//检测任务管理对象
type TaskModelMgr struct {
	Model T_data_task_model
	DeviceBaseModel
}

var modelDesc2 = "检测任务关联模型"

func init() {
	orm.RegisterModel(new(T_data_task_model))
}

//保存任务模型信息
func (c *TaskModelMgr) Save(modelids []string) (err error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_task
	dblog.Logtype = enum.LogType_Insert
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Hight
	//获取检测任务原来的模型
	oldModels := []string{}
	rowset := []orm.Params{}
	db := orm.NewOrm()
	db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&rowset)
	for _, row := range rowset {
		oldModels = append(oldModels, tools.IsEmpty(row["model_id"]))
	}
	c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
	m1 := new(SysCheckModelMgr)
	m1.SetUserInfo(c.GetUserInfo())
	for _, item := range oldModels {
		//判断模型有没有删减，如果有删除模型则同步删除对应的模型配置数据
		if arrayex.IndexOf(modelids, item) == -1 {
			c.Model.ModelId, _ = strconv.Atoi(item)
			c.Delete()
		}
	}
	for _, ids := range modelids {
		if ids == "" {
			continue
		}
		//判断是否新增了模型
		if arrayex.IndexOf(oldModels, ids) == -1 {
			logger.Logger.Debug(fmt.Sprintf("正在从模型%s复制新模型", ids))
			//将选择的系统模型复制为检测模型
			m1.Model.ModelName = fmt.Sprintf("[%d]", c.Model.TaskId) //生成不唯一的模型名称，规则：检测任务id作为复制源模型的名称前缀
			tmpid, _ := strconv.Atoi(ids)
			err, newid := m1.Copy(tmpid)
			if err != nil {
				logger.Logger.Error(err)
				break
			}
			if newid == 0 {
				continue
			}
			c.Model.ModelId = newid
			_, err = db.Insert(&c.Model)
			if err != nil {
				break
			}
		}
	}
	if err != nil {
		logger.Logger.Error(err)
		dblog.Description = fmt.Sprintf("保存%s信息失败：%s,操作数据：%+v", modelDesc2, err.Error(), c.Model)
		dblog.Fail2()
	} else {
		dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据：%+v", modelDesc2, c.Model)
		dblog.Success2()
	}
	return err
}

func (c *TaskModelMgr) One() (T_data_task_model, error) {
	if c.Model.Id == 0 {
		return c.Model, errors.New("未指定id")
	}
	o := orm.NewOrm()
	err := o.Read(&c.Model)
	if err != nil {
		logger.Logger.Error(err)
	}
	return c.Model, err
}

//根据model中指定的id删除检测任务模型对应关系
func (c *TaskModelMgr) Delete() (err error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_task
	dblog.Logtype = enum.LogType_Delete
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Hight
	db := orm.NewOrm()
	if c.Model.Id > 0 {
		db.Read(&c.Model)
		_, err = db.Delete(&c.Model)
	}
	models := []orm.Params{}
	if c.Model.TaskId > 0 {
		if c.Model.ModelId > 0 {
			_, err = db.Raw("delete from t_data_task_model where task_id=? and model_id=?", c.Model.TaskId, c.Model.ModelId).Exec()
		} else {
			_, err = db.Raw("select model_id from t_data_task_model where task_id=?", c.Model.TaskId).Values(&models)
			_, err = db.Raw("delete from t_data_task_model where task_id=?", c.Model.TaskId).Exec()
		}
	} else {
		if c.Model.ModelId > 0 {
			_, err = db.Raw("delete from t_data_task_model where model_id=?", c.Model.ModelId).Exec()
		}
	}
	if err != nil {
		logger.Logger.Error(err)
		dblog.Description = fmt.Sprintf("删除%s%d失败：%s", modelDesc2, c.Model.TaskId, err.Error())
		dblog.Fail2()
	} else {
		if c.Model.TaskId > 0 && c.Model.ModelId == 0 {
			for _, r := range models {
				id, _ := strconv.Atoi(tools.IsEmpty(r["model_id"]))
				m1 := new(SysCheckModelMgr)
				m1.SetUserInfo(c.GetUserInfo())
				m1.Model = T_data_model_defualt{Id: id, IsSys: 2}
				m1.Delete()
			}
		} else {
			m1 := new(SysCheckModelMgr)
			m1.SetUserInfo(c.GetUserInfo())
			m1.Model = T_data_model_defualt{Id: c.Model.ModelId, IsSys: 2}
			m1.Delete()
		}
		dblog.Description = fmt.Sprintf("删除%s%d成功", modelDesc2, c.Model.TaskId)
		dblog.Success2()
	}
	return err
}

//根据model中的指定过滤属性条件查询任务列表
func (c *TaskModelMgr) List(pageno, pagesize int) ([]orm.Params, int, error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_task
	dblog.Logtype = enum.LogType_Query
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Low
	o := orm.NewOrm()
	sqlParamters := []interface{}{}
	sql := "select t.*,t1.model_name from t_data_task_model t,t_data_model_defualt t1  where t.model_id=t1.id "

	if c.Model.Id > 0 {
		sql += " and t.id=?"
		sqlParamters = append(sqlParamters, c.Model.Id)
	}
	if c.Model.TaskId > 0 {
		sql += " and t.task_id=?"
		sqlParamters = append(sqlParamters, c.Model.TaskId)
	}
	limit := fmt.Sprintf(" order by t.ct desc limit %d,%d", (pageno-1)*pagesize, pagesize)
	r := []orm.Params{}
	_, err := o.Raw(sql+limit, sqlParamters).Values(&r)
	dblog.Description = fmt.Sprintf("SQL：%s 参数：%+v", sql+limit, sqlParamters)
	if err != nil {
		logger.Logger.Error(err, dblog.Description)
		dblog.Fail2()
		return nil, 0, err
	}
	dblog.Success2()
	total := []orm.Params{}
	_, err = o.Raw(strings.Replace(sql, "t.*,t1.model_name", "count(1) cnt", 1), sqlParamters).Values(&total)
	if err != nil {
		logger.Logger.Error(err)
		return nil, 0, err
	}
	totalCnt := 0
	if len(total) > 0 {
		totalCnt, _ = strconv.Atoi(tools.IsEmpty(total[0]["cnt"]))
	}
	return r, totalCnt, err
}