scd_tools_go/service/models/bo/task_model.go

package bo

import (
	"errors"
	"fmt"
	"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 [][]int) (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
	err = c.Delete()
	if err != nil {
		return err
	}
	db := orm.NewOrm()
	c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
	for _, ids := range modelids {
		c.Model.AreaId = (ids[0])
		c.Model.ModelId = (ids[1])
		_, 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)
	}
	if c.Model.TaskId > 0 {
		_, err = db.Raw("delete from t_data_task_model where task_id=?", c.Model.TaskId).Exec()
	}
	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 {
		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
}