scd_tools_go/service/models/bo/check_sys_model.go

package bo

import (
	"encoding/json"
	"encoding/xml"
	"errors"
	"fmt"
	"scd_check_tools/logger"
	"scd_check_tools/models/enum"
	"scd_check_tools/tools"
	"strconv"
	"strings"
	"sync"

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

//系统内置模型
type T_data_model_defualt struct {
	Id            int    `orm:"pk"`
	ModelName     string // '模型名称' ,
	AreaType      int    // '所属间隔类型ID' ,
	VolId         int    // '所属电压等级ID' ,
	LineLinkStyle int    // '接线方式' ,
	IedTypes      string // '包含的装置类型' ,
	RelationJson  string // '关系定义内容' ,
	IsSys         int    //是否内置模型 1 内置  2 自定义
	Cr            int    // '创建人' ,
	Ct            string `orm:"-"` // '创建时间' ,
	Ur            int    // '更新人' ,
	Ut            string `orm:"-"` // '更新时间'
}
type ModelNodeStruct struct {
	XMLName xml.Name       `xml:"flow"`
	Nodes   []*INodeStruct `xml:"nodes"`
	Edges   []*IEdges      `xml:"edges"`
}
type INodeStruct struct {
	Id         string       `xml:"id"`
	Type       string       `xml:"type"`
	X          string       `xml:"x"`
	Y          string       `xml:"y"`
	Text       *IText       `xml:"text"`
	Properties *Iproperties `xml:"properties"`
}
type IText struct {
	X     string `xml:"x"`
	Y     string `xml:"y"`
	Value string `xml:"value"`
}
type Iproperties struct {
	IedType string `xml:"ied_type"`
}

type IEdges struct {
	Id           string           `xml:"id"`
	Type         string           `xml:"type"`
	SourceNodeId string           `xml:"sourceNodeId"`
	TargetNodeId string           `xml:"targetNodeId"`
	Properties   *ILineProperties `xml:"properties"`
}
type ILineProperties struct {
	Issv string `xml:"issv"` // SV|GOOSE
}

//内置检测模型管理对象
type SysCheckModelMgr struct {
	Model T_data_model_defualt
	DeviceBaseModel
}

var sysCheckModelDesc = "内置检测模型"

var SysModelList = sync.Map{}

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

//判断名称是否已存在
func (c *SysCheckModelMgr) ExistName(name string) (has bool, id int) {
	db := orm.NewOrm()
	rows := []orm.Params{}
	_, err := db.Raw("select id from t_data_model_defualt where model_name=?", name).Values(&rows)
	if err != nil {
		logger.Logger.Error(err)
		return true, 0
	}
	if len(rows) > 0 {
		id, _ = strconv.Atoi(tools.IsEmpty(rows[0]["id"]))
		return true, id
	}
	return false, 0
}

//保存检测模型信息
func (c *SysCheckModelMgr) Save() (err error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_model
	dblog.Logtype = enum.LogType_Insert
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Hight
	db := orm.NewOrm()
	if c.Model.Id == 0 {
		h, _ := c.ExistName(c.Model.ModelName)
		if h {
			return errors.New("模型名称已存在")
		}
		c.Model.Cr, _ = strconv.Atoi(c.GetUserId())
		newid, err2 := db.Insert(&c.Model)
		if err2 == nil {
			//添加接线方式与模型关系
			lm := new(LinkStyleModelMgr)
			lm.Model.LinkstyleId = c.Model.LineLinkStyle
			lm.Model.ModelId = int(newid)
			lm.SetUserInfo(c.GetUserInfo())
			lm.Save()
			if c.Model.RelationJson != "" {
				m2 := new(SysCheckModelIedRelationMgr)
				m2.SetUserInfo(c.GetUserInfo())
				m2.Model.ModelId = int(newid)
				//err = m2.Save(c.Model.RelationJson)
			}
		}
		if err2 != nil {
			err = err2
		}
	} else {
		tmpObj, err := c.One()
		if err != nil {
			return err
		}
		if c.Model.ModelName != "" {
			h, oldid := c.ExistName(c.Model.ModelName)
			if h && c.Model.Id != oldid {
				return errors.New("模型名称已存在")
			}
		} else {
			c.Model.ModelName = tmpObj.ModelName
		}
		if c.Model.AreaType == 0 {
			c.Model.AreaType = tmpObj.AreaType
		}
		if c.Model.IedTypes == "" {
			c.Model.IedTypes = tmpObj.IedTypes
		}
		if c.Model.LineLinkStyle == 0 {
			c.Model.LineLinkStyle = tmpObj.LineLinkStyle
		} else if c.Model.LineLinkStyle != tmpObj.LineLinkStyle {
			//更新接线方式与模型关系
			lm := new(LinkStyleModelMgr)
			lm.Model.LinkstyleId = tmpObj.LineLinkStyle
			lm.Model.ModelId = c.Model.Id
			lm.SetUserInfo(c.GetUserInfo())
			lm.Delete() //清除原来的关系
			lm.Model.LinkstyleId = c.Model.LineLinkStyle
			lm.Save()
		}
		if c.Model.VolId == 0 {
			c.Model.VolId = tmpObj.VolId
		}
		if c.Model.IsSys == 0 {
			c.Model.IsSys = tmpObj.IsSys
		}

		if c.Model.RelationJson != "" {
			// 解析出模型需要的装置类型
			modelJson := map[string]interface{}{}
			err = json.Unmarshal([]byte(c.Model.RelationJson), &modelJson)
			/* <nodes>
			   <id>5d787390-ebbb-4e97-93d2-87f6a63f7c44</id>
			   <type>rect</type>
			   <x>417</x>
			   <y>202</y>
			   <properties>
			     <ied_type>PM</ied_type>
			   </properties>
			   <text>
			     <x>417</x>
			     <y>202</y>
			     <value>母线保护</value>
			   </text>
			 </nodes>*/
			/*nodes := new(ModelNodeStruct)
			err = xml.Unmarshal([]byte(`<flow>`+c.Model.RelationJson+`</flow>`), &nodes)
			*/
			if err != nil {
				logger.Logger.Error(err)
				dblog.Description = fmt.Sprintf("保存%s信息失败：%s,操作数据：%+v", sysCheckModelDesc, err.Error(), c.Model)
				dblog.Fail2()
				return err
			}
			/*
				if nodes.Nodes != nil && len(nodes.Nodes) > 0 {
					iedtypes := []string{}
					for _, ritem := range nodes.Nodes {
						iedtypes = append(iedtypes, ritem.Properties.IedType)
					}
					c.Model.IedTypes = strings.Join(iedtypes, ",")
				}
			*/
			if modelJson["nodes"] != nil {
				nodes := modelJson["nodes"].([]interface{})
				iedtypes := []string{}
				for _, ritem := range nodes {
					properties := ritem.(map[string]interface{})["properties"].(map[string]interface{})
					iedtypes = append(iedtypes, tools.IsEmpty(properties["ied_type"]))
				}
				c.Model.IedTypes = strings.Join(iedtypes, ",")
			}
			m2 := new(SysCheckModelIedRelationMgr)
			m2.SetUserInfo(c.GetUserInfo())
			m2.Model.ModelId = c.Model.Id
			err = m2.Save(modelJson) //先执行这
		} else {
			c.Model.RelationJson = tmpObj.RelationJson
		}
		if err == nil {
			c.Model.Cr = tmpObj.Cr
			c.Model.Ur, _ = strconv.Atoi(c.GetUserId())
			_, err = db.Update(&c.Model)
		}
	}
	if err != nil {
		logger.Logger.Error(err)
		dblog.Description = fmt.Sprintf("保存%s信息失败：%s,操作数据：%+v", sysCheckModelDesc, err.Error(), c.Model)
		dblog.Fail2()
	} else {
		dblog.Description = fmt.Sprintf("保存%s信息成功,操作数据：%+v", sysCheckModelDesc, c.Model)
		dblog.Success2()
	}
	return err
}

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

//根据model中指定的id删除
func (c *SysCheckModelMgr) Delete() (err error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_model
	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 err != nil {
		logger.Logger.Error(err)
		dblog.Description = fmt.Sprintf("删除%s%s失败：%s", sysCheckModelDesc, c.Model.ModelName, err.Error())
		dblog.Fail2()
	} else {
		SysModelList.Delete(c.Model.Id)
		lsm := new(LinkStyleModelMgr)
		lsm.SetUserInfo(c.GetUserInfo())
		lsm.Model.ModelId = c.Model.Id
		lsm.Delete()
		m2 := new(SysCheckModelIedRelationMgr)
		m2.SetUserInfo(c.GetUserInfo())
		m2.Model.ModelId = c.Model.Id
		m2.Delete()
		m3 := new(SysCheckModelIedFuncMgr)
		m3.Model = T_data_model_func_def{ModelId: c.Model.Id}
		m3.Delete()
		dblog.Description = fmt.Sprintf("删除%s%s成功", sysCheckModelDesc, c.Model.ModelName)
		dblog.Success2()
	}
	return err
}

//复制模型
func (c *SysCheckModelMgr) Copy(newmodelname string) (error, int) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_model
	dblog.Logtype = enum.LogType_Query
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Low
	o := orm.NewOrm()
	o.Read(&c.Model)
	sql := "insert into t_data_model_defualt(model_name,area_type,vol_id,line_link_style,ied_types,relation_json,is_sys)values(?,?,?,?,?,?,?)"
	r, err := o.Raw(sql, []interface{}{newmodelname, c.Model.AreaType, c.Model.VolId, c.Model.LineLinkStyle, c.Model.IedTypes, c.Model.RelationJson, 2}).Exec()
	if err != nil {
		logger.Logger.Error(err)
		dblog.Description = fmt.Sprintf("复制%s%s失败：%s", sysCheckModelDesc, c.Model.ModelName, err.Error())
		dblog.Fail2()
		return err, 0
	} else {
		m2 := new(SysCheckModelIedRelationMgr)
		m2.SetUserInfo(c.GetUserInfo())
		m2.Model.ModelId = c.Model.Id
		newid, _ := r.LastInsertId()
		err, _ = m2.Copy(int(newid))
		if err != nil {
			c.Model.Id = int(newid)
			c.Delete()
			dblog.Description = fmt.Sprintf("复制%s%s失败：%s", sysCheckModelDesc, c.Model.ModelName, err.Error())
			dblog.Fail2()
			return err, 0
		}
		dblog.Description = fmt.Sprintf("复制%s%s成功", sysCheckModelDesc, c.Model.ModelName)
		dblog.Success2()
		id, _ := r.LastInsertId()
		return nil, int(id)
	}
	return nil, 0
}

//获取指定电压等级下的所有模型
func (c *SysCheckModelMgr) GetModelsByVolid() ([]orm.Params, error) {
	o := orm.NewOrm()
	sqlParamters := []interface{}{}
	sql := "select t.id, t.model_name,t.area_type,c.code area_type_code,c.name area_type_name ,t.ied_types,t.is_sys ,c1.`code` vol_code,c1.`name` vol_name from t_data_model_defualt t INNER JOIN global_const_code c on c.id=t.area_type and c.parentcode='area_type' INNER JOIN global_const_code c1 on t.vol_id=c1.id and c1.parentcode='voltage_level' "
	rowset := []orm.Params{}
	_, err := o.Raw(sql, sqlParamters).Values(&rowset)
	if err != nil {
		logger.Logger.Error(err)
	}
	return rowset, err
}

//根据model中的指定过滤属性条件查询列表
func (c *SysCheckModelMgr) List(pageno, pagesize int) ([]orm.Params, int, error) {
	dblog := new(SystemLog)
	dblog.SetUserInfo(c.GetUserInfo())
	dblog.Audittype = enum.AuditType_check_model
	dblog.Logtype = enum.LogType_Query
	dblog.Eventtype = enum.OptEventType_Bus
	dblog.Eventlevel = enum.OptEventLevel_Low
	o := orm.NewOrm()
	sqlParamters := []interface{}{}
	sql := "select t.*,t1.name area_type_name,s1.name line_link_style_name,v1.name voltage_level_name from t_data_model_defualt t left join global_const_code t1 on t.area_type=t1.id and t1.parentcode='area_type' left join global_const_code v1 on t.vol_id=v1.id and v1.parentcode='voltage_level' left join t_data_link_style s1 on t.line_link_style=s1.id  where 1=1 "

	if c.Model.Id > 0 {
		sql += " and t.id=?"
		sqlParamters = append(sqlParamters, c.Model.Id)
	}
	if c.Model.VolId > 0 {
		sql += " and t.vol_id=?"
		sqlParamters = append(sqlParamters, c.Model.VolId)
	}
	if c.Model.LineLinkStyle > 0 {
		sql += " and t.line_link_style=?"
		sqlParamters = append(sqlParamters, c.Model.LineLinkStyle)
	}
	if c.Model.AreaType > 0 {
		sql += " and t.area_type=?"
		sqlParamters = append(sqlParamters, c.Model.AreaType)
	}
	if c.Model.ModelName != "" {
		sql += " and t.model_name like ?"
		sqlParamters = append(sqlParamters, "%"+c.Model.ModelName+"%")
	}
	if c.Model.IsSys > 0 {
		sql += " and t.is_sys = ?"
		sqlParamters = append(sqlParamters, c.Model.IsSys)
	}
	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.name area_type_name,s1.name line_link_style_name,v1.name voltage_level_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
}