herenshan112
4 months ago
4 changed files with 229 additions and 12 deletions
@ -0,0 +1,189 @@ |
|||
package taskmanagement |
|||
|
|||
import ( |
|||
"appPlatform/api/version1/customerform" |
|||
"appPlatform/models/customerForm" |
|||
"appPlatform/models/modelAppPlatform" |
|||
"appPlatform/models/modelshr" |
|||
"appPlatform/overall" |
|||
"appPlatform/overall/publicmethod" |
|||
"encoding/csv" |
|||
"errors" |
|||
"fmt" |
|||
"mime/multipart" |
|||
"net/http" |
|||
"regexp" |
|||
"strconv" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/gin-gonic/gin" |
|||
"github.com/xuri/excelize/v2" |
|||
) |
|||
|
|||
func (a *ApiMethod) ImportFormListFile(ctx *gin.Context) { |
|||
formId := ctx.Request.FormValue("formId") |
|||
pFileHeader, err := ctx.FormFile("file") |
|||
if err != nil { |
|||
ctx.JSON(http.StatusBadRequest, err) |
|||
return |
|||
} |
|||
|
|||
file, err := pFileHeader.Open() |
|||
if err != nil { |
|||
ctx.JSON(http.StatusBadRequest, err) |
|||
return |
|||
} |
|||
defer file.Close() |
|||
|
|||
//获取form的view，以便可以获取具体的table名
|
|||
var formCont modelAppPlatform.CustomerFormView |
|||
err = formCont.GetCont(map[string]interface{}{"`id`": formId}) |
|||
if err != nil || formCont.TableKey == "" { |
|||
ctx.JSON(http.StatusBadGateway, err) |
|||
return |
|||
} |
|||
|
|||
//直接读取数据库获取表的详细结构
|
|||
masterFieldAry, mastErr := customerform.ReadDatabaseForm(formCont.TableKey) |
|||
if mastErr != nil { |
|||
ctx.JSON(http.StatusBadGateway, mastErr) |
|||
return |
|||
} |
|||
|
|||
isCSV := strings.HasSuffix(pFileHeader.Filename, ".csv") |
|||
var ( |
|||
title []string |
|||
data [][]interface{} |
|||
) |
|||
//支持处理excel和csv格式文件
|
|||
title, data, err = handleCSVDataFile(file, masterFieldAry, isCSV) |
|||
|
|||
//构建task记录，每条导入记录绑定一个task
|
|||
context, _ := ctx.Get(overall.MyContJwt) |
|||
var userCont modelshr.ManCont |
|||
userCont.GetLoginCont(context) //当前操作人
|
|||
cureeTime := time.Now().Unix() //写入时间
|
|||
|
|||
valueMap := make(map[string]interface{}) |
|||
valueMap["creater_time"] = cureeTime |
|||
valueMap["creater"] = userCont.Key |
|||
valueMap["createrOrg"] = userCont.AdminOrg |
|||
valueMap["edit_time"] = cureeTime |
|||
i, l := 0, len(title) |
|||
for k := range data { |
|||
if len(data[k]) != l { |
|||
continue |
|||
} |
|||
_uuid := publicmethod.GetUUid(1) //统一识别符
|
|||
valueMap["masters_key"] = _uuid |
|||
for ; i < l; i++ { |
|||
//准备数据
|
|||
valueMap[title[i]] = data[k][i] |
|||
} |
|||
//重置循环
|
|||
i = 0 |
|||
delete(valueMap, "@id") |
|||
//开始插入
|
|||
if err = overall.CONSTANT_DB_CustomerForm.Table(formCont.TableKey).Create(valueMap).Error; err != nil { |
|||
break |
|||
} |
|||
|
|||
var taskCont customerForm.TaskRecord |
|||
taskCont.MastersKey = _uuid |
|||
taskCont.Title = fmt.Sprintf("%v-%v(%v)-%v", formCont.Name, userCont.Name, userCont.Number, publicmethod.UnixTimeToDay(cureeTime, 14)) //标题"`
|
|||
taskCont.AppKey = formCont.Groupid |
|||
taskCont.TableKey = formCont.SignCode |
|||
taskCont.Creater = valueMap["creater"].(int64) //创建人"`
|
|||
taskCont.CreaterTime = valueMap["edit_time"].(int64) //创建时间"`
|
|||
taskCont.EditTime = valueMap["edit_time"].(int64) //编辑时间"`
|
|||
taskCont.Types = formCont.FlowIsOpen //类型（1：流程表单；2：普通表单）"`
|
|||
taskCont.VersionId = formCont.Id //来源于哪个表单"`
|
|||
taskCont.Status = 2 |
|||
taskCont.CreaterOrg = valueMap["createrOrg"].(int64) |
|||
taskCont.MastesForm = formCont.MastesForm |
|||
taskCont.MastesFormJson = formCont.MastesFormJson |
|||
if err = overall.CONSTANT_DB_CustomerForm.Create(&taskCont).Error; err != nil { |
|||
break |
|||
} |
|||
} |
|||
if err != nil { |
|||
ctx.JSON(http.StatusBadGateway, err) |
|||
return |
|||
} |
|||
ctx.JSON(http.StatusOK, nil) |
|||
} |
|||
|
|||
// 处理excel和csv格式文件
|
|||
func handleCSVDataFile(file multipart.File, fields []customerform.Result, iscsv bool) (title []string, result [][]interface{}, err error) { |
|||
var records [][]string |
|||
if iscsv { |
|||
//handle csv file
|
|||
f := csv.NewReader(file) |
|||
records, err = f.ReadAll() |
|||
} else { |
|||
//handle excel file
|
|||
fe, err0 := excelize.OpenReader(file) |
|||
records, err = fe.GetRows("Sheet1") |
|||
if err0 != nil { |
|||
err = errors.Join(err) |
|||
} |
|||
} |
|||
|
|||
if err != nil { |
|||
return |
|||
} |
|||
|
|||
title = records[0] |
|||
reg := regexp.MustCompile(`(\w+)`) |
|||
|
|||
reflects := []string{} |
|||
//根据title，获取对应列的具体存储类型
|
|||
for k, v := range title { |
|||
//(shen1gao1)标题 => shen1gao1
|
|||
v = reg.FindString(v) |
|||
title[k] = v |
|||
for _, f := range fields { |
|||
if v == f.Field { |
|||
if strings.Contains(f.Type, "varchar") { |
|||
reflects = append(reflects, "string") |
|||
} else if strings.Contains(f.Type, "int") { |
|||
reflects = append(reflects, "int") |
|||
} else if strings.Contains(f.Type, "double") || strings.Contains(f.Type, "float") { |
|||
reflects = append(reflects, "number") |
|||
} else { |
|||
reflects = append(reflects, "string") |
|||
} |
|||
} |
|||
} |
|||
} |
|||
max := len(records) |
|||
result = make([][]interface{}, len(title)-1) //丢掉标题行
|
|||
for i := 1; i < max; i++ { |
|||
arr := []interface{}{} |
|||
for k := range records[i] { //csv的第i行数据
|
|||
//根据列数，先转换类型，然后在插入到对应的数组中
|
|||
switch reflects[k] { |
|||
case "int": |
|||
vi, err0 := strconv.Atoi(records[i][k]) |
|||
if err0 != nil { |
|||
err = err0 |
|||
return |
|||
} |
|||
arr = append(arr, vi) |
|||
case "number": |
|||
vf, err0 := strconv.ParseFloat(records[i][k], 64) |
|||
if err0 != nil { |
|||
err = err0 |
|||
return |
|||
} |
|||
arr = append(arr, vf) |
|||
default: |
|||
arr = append(arr, records[i][k]) |
|||
} |
|||
} |
|||
result[i-1] = arr |
|||
} |
|||
|
|||
return |
|||
} |
|||
Loading…
Reference in new issue