/* * @Author: liling * @Date: 2022-08-13 14:38:13 * @LastEditors: liling * @LastEditTime: 2022-08-13 14:38:15 * @FilePath: \police_security\controllers\exelController.go * @Description:excel导入导出相关控制器 * * Copyright (c) 2022 by liling/jujutong, All Rights Reserved. */ package controllers import ( "errors" "fmt" "os" "scd_check_tools/excel" "scd_check_tools/global" "scd_check_tools/models/bo" "scd_check_tools/tools" "strings" "time" "github.com/astaxie/beego/orm" ) //文件/附件服务 type ExcelController struct { BaseController } func init() { } //公用数据导出方法 // @Summary Excel数据导出 // @Description Excel数据导出。导出成功时将返回导出文件的下载地址。 // @Tags 文件/附件服务接口 // @Accept x-www-form-urlencoded // @Produce json // @Param code query string true "业务类型。值为:scd-comp-result(差异对比结果)\scl-check-result(SCL校验结果)之一" // @Param caption query string false "数据标题" // @Param expcols query string false "导出的数据列,与expcolnames一一对应。从个数据列之间使用逗号分隔。未设置时将导出业务对应的所有列" // @Param expcolnames query string false "导出的数据列对应的名称,与expcols一一一对应。从个数据列名之间使用逗号分隔。未设置时将导出业务对应的所有列" // @Success 200 {object} ResultOK 成功 // @Failure 500 {object} ResultError 失败 // @router expTableData [post] func (c *ApiController) ExpTableData() { code := c.GetString("code", "") //业务数据类别,由系统定义 if code == "" { c.Data["json"] = c.ResultError("导出的业务数据类别不能为空") c.ServeJSON() return } expCaption := c.GetString("caption", "数据导出-"+time.Now().Format("20060102150405")) expcols := []string{} expcolnames := []string{} c1 := c.GetString("expcols") if c1 != "" { expcols = strings.Split(c1, ",") } c2 := c.GetString("expcolnames") if c2 != "" { expcolnames = strings.Split(c2, ",") } if len(expcols) != len(expcolnames) { c.Data["json"] = c.ResultError("设置了导出列,但列名与中文名数量不一致") c.ServeJSON() return } paras := c.Input() param := map[string]interface{}{"code": code} for k, _ := range paras { param[k] = paras.Get(k) } datalist := []orm.Params{} scdinfo := orm.Params{} var err error switch code { case "log-opt": //操作日志导出 /* datatable := map[string]interface{}{"state": param["state"], "starttime": param["starttime"], "endtime": param["endtime"], "words": param["words"]} queryPara := map[string]interface{}{"datatable": datatable} queryReturn, err := bo.QueryGlobalsyslogList(queryPara) if err == nil { datalist = queryReturn["data"].([]orm.Params) }*/ break case "log-colle": //数据采集日志导出 /* datatable := map[string]interface{}{"starttime": param["starttime"], "endtime": param["endtime"]} queryPara := map[string]interface{}{"datatable": datatable} queryReturn, err := bo.QueryDatalogList(queryPara) if err == nil { datalist = queryReturn["data"].([]orm.Params) }*/ break case "log-report": /* datatable := map[string]interface{}{"state": param["state"], "starttime": param["starttime"], "endtime": param["endtime"], "words": param["words"], "type": param["type"]} queryPara := map[string]interface{}{"datatable": datatable} queryReturn, err := bo.QueryReportlogList(queryPara) if err == nil { datalist = queryReturn["data"].([]orm.Params) }*/ //上报日志导出 break case "scd-comp-result": //SCD一致性比对结果 obj := new(bo.ScdCompare) compid, _ := c.GetInt("comp_id") datalist, err = obj.ExpResultList(compid, c.GetString("nodetype"), c.GetString("nodeid")) break case "scl-check-result": //SCL校验结果 scdid, _ := c.GetInt64("scd_id") scdObj := new(bo.ScdMgr) scdinfo, err = scdObj.One(tools.IsEmpty(scdid)) iedname := c.GetString("ied_name") nodename := c.GetString("node_name") if err == nil { scdname := tools.IsEmpty(scdinfo["scd_name"]) stationinfo, _ := new(bo.BasicArea).One(tools.IsEmpty(scdinfo["station_id"])) if stationinfo != nil { obj := new(bo.ScdNodeRule) obj.ScdID = scdid datalist, _, err = obj.ResultList("", "", c.GetString("alert_level"), iedname, nodename, c.GetString("node_id"), 1, 30000) if iedname != "" { expCaption = fmt.Sprintf("%s-%s-装置(%s)校验结果", stationinfo.AreaName, scdname, iedname) } else if nodename != "" { nodename_zh := "" switch nodename { case "SCLSyntax": nodename_zh = "语法" break case "voltage_level": nodename_zh = "电压等级" key := "global_code_" + c.GetString("node_id") vol, _ := global.GoCahce.Get(key) if vol != nil { nodename_zh = nodename_zh + "(" + tools.IsEmpty(vol.(orm.Params)["name"]) + ")" } break case "area": nodename_zh = "间隔" areaObj := new(bo.ScdAreaMgr) areaInfo, _ := areaObj.One(c.GetString("node_id")) if areaInfo != nil { nodename_zh = nodename_zh + "(" + areaInfo.(bo.T_substation_area).Name + ")" } break default: nodename_zh = nodename break } expCaption = fmt.Sprintf("%s-%s-%s校验结果", stationinfo.AreaName, scdname, nodename_zh) } else { expCaption = fmt.Sprintf("%s-%s全部校验结果", stationinfo.AreaName, scdname) } } } break default: datalist = nil err = errors.New("未定义的导出数据类型:" + code) break } if err != nil { c.Data["json"] = c.ResultError(err.Error()) c.ServeJSON() return } curPath, _ := os.Getwd() //当前绝对路径 excel.InitExcel("exp_data_"+time.Now().Format("20060102150405"), curPath+string(os.PathSeparator)+"static"+string(os.PathSeparator)+"download") filename, errf := excel.ExportExcel(expCaption, expcols, expcolnames, datalist, true) if errf != nil { c.Data["json"] = c.ResultError(errf.Error()) } else { c.Data["json"] = c.ResultOK("static/download/"+filename, 0) } c.ServeJSON() }