You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
563 lines
16 KiB
563 lines
16 KiB
package models
|
|
|
|
import (
|
|
"lapp_-wy/db"
|
|
"lapp_-wy/utils"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/360EntSecGroup-Skylar/excelize"
|
|
"strings"
|
|
"xorm.io/core"
|
|
)
|
|
|
|
type Scripttab struct {
|
|
Cid int `xorm:"pk comment('工厂号') Int" json:"scripttab-cid"`
|
|
Script string `xorm:"pk comment('脚本') VARCHAR(50)" json:"scripttab-script"`
|
|
Bez string `xorm:"comment('名称') VARCHAR(50)" json:"scripttab-bez"`
|
|
Scripttable string `xorm:"comment('导入表') VARCHAR(40)" json:"scripttab-scripttable"`
|
|
Scripttype string `xorm:"comment('脚本类型') VARCHAR(20)" json:"scripttab-scripttype"`
|
|
Filepath string `xorm:"comment('文件路径') VARCHAR(50)" json:"scripttab-filepath"`
|
|
Separator string `xorm:"comment('分隔符') VARCHAR(20)" json:"scripttab-separator"`
|
|
Usehead int `xorm:"comment('使用列头') INT" json:"scripttab-usehead"`
|
|
Cleartable int `xorm:"comment('清除表') INT" json:"scripttab-cleartable"`
|
|
Startrow int `xorm:"comment('起始行') INT" json:"scripttab-startrow"`
|
|
Startcol int `xorm:"comment('起始列') INT" json:"scripttab-startcol"`
|
|
Appendnewrecord int `xorm:"comment('附加新纪录') INT" json:"scripttab-appendnewrecord"`
|
|
Updaterecord int `xorm:"comment('更新记录') INT" json:"scripttab-updaterecord"`
|
|
Callspecfunc int `xorm:"comment('调用特殊功能') INT" json:"scripttab-callspecfunc"`
|
|
Callfunction string `xorm:"comment('调用函数') VARCHAR(100)" json:"scripttab-callfunction"`
|
|
Inputpara1 string `xorm:"comment('输入参数1') VARCHAR(255)" json:"scripttab-inputpara1"`
|
|
Para1type int `xorm:"comment('参数1类型') INT" json:"scripttab-para1type"`
|
|
Para1descr string `xorm:"comment('参数1描述') VARCHAR(40)" json:"scripttab-para1descr"`
|
|
Inputpara2 string `xorm:"comment('输入参数2') VARCHAR(255)" json:"scripttab-inputpara2"`
|
|
Para2type int `xorm:"comment('参数2类型') INT" json:"scripttab-para2type"`
|
|
Para2descr string `xorm:"comment('参数2描述') VARCHAR(40)" json:"scripttab-para2descr"`
|
|
Inputpara3 string `xorm:"comment('输入参数3') VARCHAR(255)" json:"scripttab-inputpara3"`
|
|
Para3type int `xorm:"comment('参数3类型') INT" json:"scripttab-para3type"`
|
|
Para3descr string `xorm:"comment('参数3描述') VARCHAR(40)" json:"scripttab-para3descr"`
|
|
Inputpara4 string `xorm:"comment('输入参数4') VARCHAR(255)" json:"scripttab-inputpara4"`
|
|
Para4type int `xorm:"comment('参数4类型') INT" json:"scripttab-para4type"`
|
|
Para4descr string `xorm:"comment('参数4描述') VARCHAR(40)" json:"scripttab-para4descr"`
|
|
Lastmodif string `xorm:"comment('最近一次更改时间') VARCHAR(14)" json:"scripttab-lastmodif"`
|
|
Lastuser string `xorm:"comment('最近一次更改人') VARCHAR(20)" json:"scripttab-lastuser"`
|
|
Credatuz string `xorm:"comment('创建时间') VARCHAR(14)" json:"scripttab-credatuz"`
|
|
Valst []Scriptitemtab `json:"valst" xorm:"-"`
|
|
}
|
|
|
|
func (t *Scripttab) TableName() string {
|
|
return "scripttab"
|
|
}
|
|
|
|
// 清除string字段的右侧空格
|
|
func (t *Scripttab) Clipped() {
|
|
utils.TrimStruct(t, *t)
|
|
}
|
|
|
|
//增
|
|
func (t *Scripttab) Add() error {
|
|
e := db.MasterEngine()
|
|
countrole := new(Scripttab)
|
|
affw, err := e.Table("scripttab").ID(core.PK{t.Cid, t.Script}).Count(countrole)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if affw > 0 {
|
|
return errors.New("数据已经存在！")
|
|
}
|
|
_, err = e.Table("scripttab").Insert(t)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//批量删除
|
|
_, err = e.Table("scriptitemtab").Where("cid = ? and script = ?", t.Cid, t.Script).Delete(&Scriptitemtab{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//批量插入
|
|
alldata := make([]Scriptitemtab, 0)
|
|
for _, v := range t.Valst {
|
|
onedata := Scriptitemtab{}
|
|
onedata.Cid = t.Cid
|
|
onedata.Script = t.Script
|
|
onedata.Tabmapcol = v.Tabmapcol
|
|
onedata.Colno = v.Colno
|
|
onedata.Rowno = v.Rowno
|
|
onedata.Colfieldtype = v.Colfieldtype
|
|
onedata.Colname = v.Colname
|
|
onedata.Rowname = v.Rowname
|
|
onedata.Colformat = v.Colformat
|
|
onedata.Defaultvalue = v.Defaultvalue
|
|
alldata = append(alldata, onedata)
|
|
}
|
|
_, err = e.Table("scriptitemtab").Insert(&alldata)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//删
|
|
func (t *Scripttab) Del() bool {
|
|
e := db.MasterEngine()
|
|
_, err := e.ID(core.PK{t.Cid, t.Script}).Delete(&Scripttab{})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
countrole := new(Scriptitemtab)
|
|
affw, _ := e.Table("scriptitemtab").Where("cid = ? and script = ?", t.Cid, t.Script).Count(countrole)
|
|
if affw > 0 {
|
|
_, err = e.Where("cid = ? and script = ?", t.Cid, t.Script).Delete(&Scriptitemtab{})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
//改
|
|
func (t *Scripttab) Update() bool {
|
|
e := db.MasterEngine()
|
|
_, err := e.ID(core.PK{t.Cid, t.Script}).Cols("usehead", "scripttype", "startcol", "startrow", "scripttable", "cleartable", "appendnewrecord", "updaterecord", "callspecfunc").Update(t)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
//批量删除
|
|
_, err = e.Table("scriptitemtab").Where("cid = ? and script = ?", t.Cid, t.Script).Delete(&Scriptitemtab{})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
//批量插入
|
|
alldata := make([]Scriptitemtab, 0)
|
|
for _, v := range t.Valst {
|
|
onedata := Scriptitemtab{}
|
|
onedata.Cid = t.Cid
|
|
onedata.Script = t.Script
|
|
onedata.Tabmapcol = v.Tabmapcol
|
|
onedata.Colno = v.Colno
|
|
onedata.Rowno = v.Rowno
|
|
onedata.Colfieldtype = v.Colfieldtype
|
|
onedata.Colname = v.Colname
|
|
onedata.Rowname = v.Rowname
|
|
onedata.Colformat = v.Colformat
|
|
onedata.Defaultvalue = v.Defaultvalue
|
|
alldata = append(alldata, onedata)
|
|
}
|
|
_, err = e.Table("scriptitemtab").Insert(&alldata)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
//查
|
|
func (t *Scripttab) SelectOne() (Scripttab, error) {
|
|
e := db.MasterEngine()
|
|
var data Scripttab
|
|
_, err := e.ID(core.PK{t.Cid, t.Script}).Get(&data)
|
|
if err != nil {
|
|
return data, err
|
|
}
|
|
data.Clipped()
|
|
var datalist []Scriptitemtab
|
|
err = e.Where("cid = ? and script = ?", t.Cid, t.Script).Find(&datalist)
|
|
if err != nil {
|
|
return data, err
|
|
}
|
|
for k, _ := range datalist {
|
|
datalist[k].Clipped()
|
|
}
|
|
data.Valst = datalist
|
|
return data, nil
|
|
}
|
|
|
|
//分页
|
|
func (t *Scripttab) GetPage(pageSize int, pageIndex int) ([]Scripttab, int, error) {
|
|
data := make([]Scripttab, 0)
|
|
e := db.MasterEngine()
|
|
query := e.Table("scripttab").Where("cid = ? ", t.Cid)
|
|
table := e.Table("scripttab").Where("cid = ? ", t.Cid)
|
|
if !utils.ValueIsEmpty(t.Script) {
|
|
query = query.And("script = ?", t.Script)
|
|
table = table.And("script = ?", t.Script)
|
|
}
|
|
if !utils.ValueIsEmpty(t.Scripttype) {
|
|
query = query.And("scripttype = ?", t.Scripttype)
|
|
table = table.And("scripttype = ?", t.Scripttype)
|
|
}
|
|
Offset := (pageIndex - 1) * pageSize
|
|
err := query.Limit(pageSize, Offset).Desc("credatuz").Find(&data)
|
|
pcount := new(Scripttab)
|
|
count, err := table.Count(pcount)
|
|
if err != nil {
|
|
return data, 0, err
|
|
}
|
|
for k, _ := range data {
|
|
data[k].Clipped()
|
|
}
|
|
return data, int(count), nil
|
|
}
|
|
|
|
type TablePk struct {
|
|
TabName string
|
|
Colname string
|
|
}
|
|
|
|
// 一维表导入表功能
|
|
func (t *Scripttab) ReadXlsx(path string) bool {
|
|
e := db.MasterEngine()
|
|
|
|
xlsx, err := excelize.OpenFile(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
rows := xlsx.GetRows("Sheet1")
|
|
|
|
//第一步:获取表名称
|
|
tablename := t.Scripttable
|
|
|
|
//新增新纪录
|
|
if t.Appendnewrecord == 1 {
|
|
//第二步构建sql语句
|
|
for startrow, row := range rows {
|
|
//起始行导入
|
|
if (startrow + 1) <= t.Startrow {
|
|
continue
|
|
}
|
|
keystr := ""
|
|
valstr := ""
|
|
|
|
//构造map
|
|
tem := map[int]interface{}{}
|
|
for k, v := range row {
|
|
key := k + 1
|
|
val := v
|
|
tem[key] = val
|
|
}
|
|
|
|
//第三步：拼接
|
|
for _, vv := range t.Valst {
|
|
keystr += vv.Tabmapcol + ","
|
|
val, ok := tem[vv.Colno]
|
|
if ok {
|
|
valstr += "'" + strings.TrimSpace(utils.ValueToString(val, "")) + "'" + ","
|
|
} else {
|
|
if !utils.ValueIsEmpty(vv.Defaultvalue) {
|
|
valstr += "' " + utils.ValueToString(vv.Defaultvalue, "") + " '" + ","
|
|
} else {
|
|
//根据数据类型添加默认值
|
|
if vv.Colfieldtype == 1 {
|
|
valstr += "'0'" + ","
|
|
} else if vv.Colfieldtype == 5 {
|
|
valstr += "'0.0'" + ","
|
|
} else {
|
|
valstr += "' '" + ","
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
//去掉最后边的逗号
|
|
keystr = strings.Trim(keystr, ",")
|
|
valstr = strings.Trim(valstr, ",")
|
|
//2.开始导入
|
|
_, err := e.Exec("INSERT INTO " + tablename + " (" + keystr + ") values (" + valstr + ")")
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
//更新数据表
|
|
if t.Updaterecord == 1 {
|
|
TablePk := make([]TablePk, 0)
|
|
//获取当前导入表的主键
|
|
err := e.SQL("SELECT TABLE_NAME as tabname,COLUMN_NAME as colname FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE WHERE TABLE_NAME=?", t.Scripttable).Find(&TablePk)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
temPk := make(map[string]interface{})
|
|
for _, v := range TablePk {
|
|
key := v.Colname
|
|
temPk[key] = v
|
|
}
|
|
//第二步构建sql语句
|
|
for startrow, row := range rows {
|
|
//起始行导入
|
|
if (startrow + 1) <= t.Startrow {
|
|
continue
|
|
}
|
|
wherestr := ""
|
|
valstr := ""
|
|
|
|
//构造map
|
|
tem := map[int]interface{}{}
|
|
for k, v := range row {
|
|
key := k + 1
|
|
val := v
|
|
tem[key] = val
|
|
}
|
|
|
|
//第三步：拼接
|
|
for _, vv := range t.Valst {
|
|
_, res := temPk[vv.Tabmapcol]
|
|
val, ok := tem[vv.Colno]
|
|
if ok {
|
|
if res {
|
|
wherestr += vv.Tabmapcol + "= '" + utils.ValueToString(val, "") + "'" + " and "
|
|
}
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(val, "") + "'" + ","
|
|
} else {
|
|
if res {
|
|
wherestr += vv.Tabmapcol + "= '" + utils.ValueToString(vv.Defaultvalue, "") + "'" + " and "
|
|
}
|
|
if !utils.ValueIsEmpty(vv.Defaultvalue) {
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(vv.Defaultvalue, "") + " '" + ","
|
|
}
|
|
|
|
}
|
|
}
|
|
//去掉最后边的逗号
|
|
wherestr = strings.Trim(wherestr, " and ")
|
|
valstr = strings.Trim(valstr, ",")
|
|
//2.开始导入
|
|
//更新
|
|
_, err = e.Exec("UPDATE " + tablename + " SET " + valstr + " WHERE " + wherestr)
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
continue
|
|
}
|
|
}
|
|
|
|
}
|
|
//清除数据表
|
|
if t.Cleartable == 1 {
|
|
//先删除
|
|
_, err := e.Exec("delete from " + tablename)
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
type TabData struct {
|
|
Tabmapcol string
|
|
Coltype string
|
|
}
|
|
|
|
//新增数据（根据表明查询）
|
|
func (t *Scripttab) SelectArr() ([]Scriptitemtab, error) {
|
|
e := db.MasterEngine()
|
|
datalist := make([]TabData, 0)
|
|
data := make([]Scriptitemtab, 0)
|
|
err := e.SQL("SELECT COLUMN_NAME as tabmapcol,DATA_TYPE as coltype FROM INFORMATION_SCHEMA.columns WHERE TABLE_NAME=?", t.Scripttable).Find(&datalist)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, v := range datalist {
|
|
dataone := Scriptitemtab{}
|
|
dataone.Tabmapcol = v.Tabmapcol
|
|
if v.Coltype == "int" {
|
|
dataone.Colfieldtype = 1
|
|
} else if v.Coltype == "decimal" {
|
|
dataone.Colfieldtype = 5
|
|
} else {
|
|
dataone.Colfieldtype = 8
|
|
}
|
|
dataone.Clipped()
|
|
data = append(data, dataone)
|
|
}
|
|
return data, err
|
|
}
|
|
|
|
// 二维表导入表功能
|
|
func (t *Scripttab) ReadDoubleXlsx(path string) bool {
|
|
e := db.MasterEngine()
|
|
|
|
xlsx, err := excelize.OpenFile(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
rows := xlsx.GetRows("Sheet1")
|
|
|
|
//第一步:获取表名称
|
|
tablename := t.Scripttable
|
|
|
|
//定义一个数组，记录attrcode,并且与k关联起来
|
|
tem := map[int]string{}
|
|
|
|
//新增新纪录
|
|
if t.Appendnewrecord == 1 {
|
|
//第二步构建sql语句
|
|
for startrow, row := range rows {
|
|
|
|
if t.Startrow == (startrow + 1) {
|
|
//记录attrcode,并且与k关联起来
|
|
for kk, vv := range row {
|
|
tem[kk] = vv
|
|
}
|
|
continue
|
|
}
|
|
|
|
//起始行导入
|
|
if (startrow + 1) <= t.Startrow {
|
|
continue
|
|
}
|
|
|
|
//构造map
|
|
temrow := map[int]interface{}{}
|
|
|
|
for k, v := range row {
|
|
//记录表头
|
|
if k < t.Startcol {
|
|
key := k + 1
|
|
val := v
|
|
temrow[key] = val
|
|
continue
|
|
}
|
|
//第三步：拼接
|
|
for _, vv := range t.Valst {
|
|
keystr := ""
|
|
valstr := ""
|
|
//记录字段
|
|
keystr += vv.Tabmapcol + ","
|
|
val, ok := temrow[vv.Colno]
|
|
if ok {
|
|
valstr += "'" + utils.ValueToString(val, "") + "'" + ","
|
|
} else if t.Startrow == vv.Rowno && vv.Rowno > 0 {
|
|
valstr += "'" + utils.ValueToString(tem[k], "") + "'" + ","
|
|
} else if t.Startcol == vv.Rowno && vv.Rowno > 0 {
|
|
if utils.ValueIsEmpty(v) {
|
|
continue
|
|
}
|
|
valstr += "'" + utils.ValueToString(v, "") + "'" + ","
|
|
} else {
|
|
if !utils.ValueIsEmpty(vv.Defaultvalue) {
|
|
valstr += "' " + utils.ValueToString(vv.Defaultvalue, "") + " '" + ","
|
|
} else {
|
|
//根据数据类型添加默认值
|
|
if vv.Colfieldtype == 1 {
|
|
valstr += "'0'" + ","
|
|
} else if vv.Colfieldtype == 5 {
|
|
valstr += "'0.0'" + ","
|
|
} else {
|
|
valstr += "' '" + ","
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
//去掉最后边的逗号
|
|
keystr = strings.Trim(keystr, ",")
|
|
valstr = strings.Trim(valstr, ",")
|
|
//2.开始导入
|
|
_, err := e.Exec("INSERT INTO " + tablename + " (" + keystr + ") values (" + valstr + ")")
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
continue
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
//更新数据表
|
|
if t.Updaterecord == 1 {
|
|
TablePk := make([]TablePk, 0)
|
|
//获取当前导入表的主键
|
|
err := e.SQL("SELECT TABLE_NAME as tabname,COLUMN_NAME as colname FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE WHERE TABLE_NAME=?", t.Scripttable).Find(&TablePk)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
temPk := make(map[string]interface{})
|
|
for _, v := range TablePk {
|
|
key := v.Colname
|
|
temPk[key] = v
|
|
}
|
|
//第二步构建sql语句
|
|
for startrow, row := range rows {
|
|
|
|
if t.Startrow == (startrow + 1) {
|
|
//记录attrcode,并且与k关联起来
|
|
for kk, vv := range row {
|
|
tem[kk] = vv
|
|
}
|
|
continue
|
|
}
|
|
|
|
//起始行导入
|
|
if (startrow + 1) <= t.Startrow {
|
|
continue
|
|
}
|
|
|
|
//构造map
|
|
temrow := map[int]interface{}{}
|
|
|
|
for k, v := range row {
|
|
//记录表头
|
|
if k < t.Startcol {
|
|
key := k + 1
|
|
val := v
|
|
temrow[key] = val
|
|
continue
|
|
}
|
|
//第三步：拼接
|
|
for _, vv := range t.Valst {
|
|
wherestr := ""
|
|
valstr := ""
|
|
_, res := temPk[vv.Tabmapcol]
|
|
val, ok := temrow[vv.Colno]
|
|
if ok {
|
|
if res {
|
|
wherestr += vv.Tabmapcol + "= '" + utils.ValueToString(val, "") + "'" + " and "
|
|
}
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(val, "") + "'" + ","
|
|
} else if t.Startrow == vv.Rowno && vv.Rowno > 0 {
|
|
if res {
|
|
wherestr += vv.Tabmapcol + "= '" + utils.ValueToString(vv.Defaultvalue, "") + "'" + " and "
|
|
}
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(tem[k], "") + "'" + ","
|
|
} else if t.Startcol == vv.Rowno && vv.Rowno > 0 {
|
|
if utils.ValueIsEmpty(v) {
|
|
continue
|
|
}
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(v, "") + "'" + ","
|
|
} else {
|
|
if res {
|
|
wherestr += vv.Tabmapcol + "= '" + utils.ValueToString(vv.Defaultvalue, "") + "'" + " and "
|
|
}
|
|
if !utils.ValueIsEmpty(vv.Defaultvalue) {
|
|
valstr += vv.Tabmapcol + "= '" + utils.ValueToString(vv.Defaultvalue, "") + " '" + ","
|
|
}
|
|
}
|
|
//去掉最后边的逗号
|
|
wherestr = strings.Trim(wherestr, " and ")
|
|
valstr = strings.Trim(valstr, ",")
|
|
//2.开始导入
|
|
//更新
|
|
_, err = e.Exec("UPDATE " + tablename + " SET " + valstr + " WHERE " + wherestr)
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
continue
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
//清除数据表
|
|
if t.Cleartable == 1 {
|
|
//先删除
|
|
_, err := e.Exec("delete from " + tablename)
|
|
if err != nil {
|
|
fmt.Printf("err is :%v", err)
|
|
}
|
|
}
|
|
return true
|
|
}
|