265 lines
7.6 KiB
Go
265 lines
7.6 KiB
Go
package models
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/codinl/go-logger"
|
||
"github.com/xuri/excelize/v2"
|
||
"reflect"
|
||
)
|
||
|
||
type CategoryExcel struct {
|
||
FirstCategory string `json:"first_category" binding:"required"` // 一级分类
|
||
SecondCategory string `json:"second_category"` // 二级分类
|
||
ThreeCategory string `json:"three_category"` // 三级分类
|
||
}
|
||
|
||
type CommodityExcel struct {
|
||
Category string `json:"category" binding:"required"` // 商品所属分类
|
||
Name string `json:"name" binding:"required"` // 商品名称
|
||
IMEIType string `json:"imei_type" binding:"required"` // 是否串码 1-无串码 2-串码（系统生成） 3-串码（手动添加）
|
||
SysGenerate string `json:"sys_generate" binding:"required"` // 系统生成串码
|
||
SupplierName string `json:"supplier_name" binding:"required"` // 供应商名称
|
||
SellBrokerage string `json:"sell_brokerage" binding:"required"` // 销售毛利提成
|
||
StaffBrokerage string `json:"staff_brokerage" binding:"required"` // 员工毛利提成
|
||
MemberDiscount string `json:"member_discount" binding:"required"` // 会员优惠
|
||
RetailPrice string `json:"retail_price" binding:"required"` // 指导零售价
|
||
MinRetailPrice string `json:"min_retail_price" binding:"required"` // 最低零售价
|
||
WholesalePrice string `json:"wholesale_price" binding:"required"` // 指导采购价
|
||
StaffCostPrice string `json:"staff_cost_price" binding:"required"` // 员工成本价加价
|
||
Origin string `json:"origin"` // 产地
|
||
Remark string `json:"remark"` // 备注
|
||
}
|
||
|
||
// 获取struct的tag标签，匹配excel导入数据
|
||
func getJSONTagNames(s interface{}) []string {
|
||
valueType := reflect.TypeOf(s)
|
||
if valueType.Kind() != reflect.Struct {
|
||
fmt.Println("传入的不是结构体")
|
||
return nil
|
||
}
|
||
|
||
var tagNames []string
|
||
for i := 0; i < valueType.NumField(); i++ {
|
||
field := valueType.Field(i)
|
||
tag := field.Tag.Get("json")
|
||
if tag != "" {
|
||
tagNames = append(tagNames, tag)
|
||
}
|
||
}
|
||
|
||
return tagNames
|
||
}
|
||
|
||
// 预览excel数据（不做必填项校验）
|
||
func FileExcelReader(d []byte, cols []string) ([]byte, []map[string]interface{}, error) {
|
||
reader, err := excelize.OpenReader(bytes.NewReader(d))
|
||
if err != nil {
|
||
logger.Error("open reader err:", err)
|
||
return nil, nil, err
|
||
}
|
||
|
||
sheetList := reader.GetSheetList()
|
||
if len(sheetList) == 0 {
|
||
logger.Error("sheet list nil")
|
||
return nil, nil, err
|
||
}
|
||
rows, err := reader.Rows(sheetList[0])
|
||
if err != nil {
|
||
logger.Error("reader rows err:", err)
|
||
return nil, nil, err
|
||
}
|
||
sheetCols, err := reader.GetCols(sheetList[0])
|
||
if err != nil {
|
||
logger.Error("reader get cols err:", err)
|
||
return nil, nil, err
|
||
}
|
||
if len(sheetCols) == 0 {
|
||
logger.Error("get cols null err")
|
||
return nil, nil, err
|
||
}
|
||
colsMap := make([]map[string]interface{}, 0)
|
||
if len(cols) == 0 {
|
||
if sheetList[0] == "导分类" {
|
||
cols = getJSONTagNames(CategoryExcel{})
|
||
} else if sheetList[0] == "导商品" {
|
||
cols = getJSONTagNames(CommodityExcel{})
|
||
} else {
|
||
cols = make([]string, len(sheetCols))
|
||
for i := 0; i < len(sheetCols); i++ {
|
||
cols[i] = fmt.Sprintf("c%02d", i)
|
||
}
|
||
}
|
||
} else {
|
||
fmt.Println("cols", len(cols))
|
||
fmt.Println("sheetCols", len(sheetCols))
|
||
if len(cols) != len(sheetCols) {
|
||
logger.Error("cols length not equal columns")
|
||
return nil, nil, errors.New("cols length not equal columns")
|
||
}
|
||
}
|
||
|
||
for rows.Next() {
|
||
columns, err := rows.Columns()
|
||
if err != nil {
|
||
logger.Error("rows columns err:", err)
|
||
return nil, nil, err
|
||
}
|
||
|
||
columnMap := make(map[string]interface{}, 0)
|
||
for i, _ := range cols {
|
||
columnMap[cols[i]] = ""
|
||
}
|
||
for i, _ := range columns {
|
||
columnMap[cols[i]] = columns[i]
|
||
}
|
||
fmt.Println("columnMap:", columnMap)
|
||
colsMap = append(colsMap, columnMap)
|
||
}
|
||
|
||
fmt.Println("colsMap:", colsMap)
|
||
mCols, err := json.Marshal(colsMap)
|
||
if err != nil {
|
||
logger.Error("marshal err:", err)
|
||
return mCols, nil, err
|
||
}
|
||
|
||
return mCols, colsMap, nil
|
||
}
|
||
|
||
// 导入excel数据（校验必填项）
|
||
func FileExcelImport(d []byte, cols []string) ([]byte, []map[string]interface{}, error) {
|
||
reader, err := excelize.OpenReader(bytes.NewReader(d))
|
||
if err != nil {
|
||
logger.Error("open reader err:", err)
|
||
return nil, nil, err
|
||
}
|
||
|
||
sheetList := reader.GetSheetList()
|
||
if len(sheetList) == 0 {
|
||
logger.Error("sheet list nil")
|
||
return nil, nil, err
|
||
}
|
||
rows, err := reader.Rows(sheetList[0])
|
||
if err != nil {
|
||
logger.Error("reader rows err:", err)
|
||
return nil, nil, err
|
||
}
|
||
sheetCols, err := reader.GetCols(sheetList[0])
|
||
if err != nil {
|
||
logger.Error("reader get cols err:", err)
|
||
return nil, nil, err
|
||
}
|
||
if len(sheetCols) == 0 {
|
||
logger.Error("get cols null err")
|
||
return nil, nil, err
|
||
}
|
||
colsMap := make([]map[string]interface{}, 0)
|
||
if len(cols) == 0 {
|
||
if sheetList[0] == "导分类" {
|
||
//校验商品分类导入规则
|
||
err := checkCategoryExcel(sheetCols)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
cols = getJSONTagNames(CategoryExcel{})
|
||
} else if sheetList[0] == "导商品" {
|
||
//校验商品资料导入规则
|
||
err := checkCommodityExcel(sheetCols)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
cols = getJSONTagNames(CommodityExcel{})
|
||
} else {
|
||
cols = make([]string, len(sheetCols))
|
||
for i := 0; i < len(sheetCols); i++ {
|
||
cols[i] = fmt.Sprintf("c%02d", i)
|
||
}
|
||
}
|
||
} else {
|
||
fmt.Println("cols", len(cols))
|
||
fmt.Println("sheetCols", len(sheetCols))
|
||
if len(cols) != len(sheetCols) {
|
||
logger.Error("cols length not equal columns")
|
||
return nil, nil, errors.New("cols length not equal columns")
|
||
}
|
||
}
|
||
|
||
for rows.Next() {
|
||
columns, err := rows.Columns()
|
||
if err != nil {
|
||
logger.Error("rows columns err:", err)
|
||
return nil, nil, err
|
||
}
|
||
|
||
columnMap := make(map[string]interface{}, 0)
|
||
for i, _ := range cols {
|
||
columnMap[cols[i]] = ""
|
||
for j, _ := range columns {
|
||
columnMap[cols[j]] = columns[j]
|
||
}
|
||
}
|
||
//for i, _ := range columns {
|
||
// columnMap[cols[i]] = columns[i]
|
||
//}
|
||
|
||
fmt.Println("columnMap:", columnMap)
|
||
colsMap = append(colsMap, columnMap)
|
||
}
|
||
|
||
fmt.Println("colsMap:", colsMap)
|
||
mCols, err := json.Marshal(colsMap)
|
||
if err != nil {
|
||
logger.Error("marshal err:", err)
|
||
return mCols, nil, err
|
||
}
|
||
|
||
return mCols, colsMap, nil
|
||
}
|
||
|
||
// 校验商品分类导入规则
|
||
// 导入商品分类校验报错： （1）只有3级，没有2级或1级 （2）有2，3级，但没有1级
|
||
func checkCategoryExcel(sheetCols [][]string) error {
|
||
if len(sheetCols) != 3 {
|
||
return errors.New("模版错误，请检查文件")
|
||
}
|
||
|
||
if len(sheetCols[0]) < len(sheetCols[1]) || len(sheetCols[0]) < len(sheetCols[2]) {
|
||
return errors.New("格式错误，缺少一级分类")
|
||
} else if len(sheetCols[1]) < len(sheetCols[2]) {
|
||
return errors.New("格式错误，缺少二级分类")
|
||
}
|
||
|
||
for i := 1; i < len(sheetCols[2]); i++ {
|
||
if sheetCols[0][i] == "" && (sheetCols[1][i] != "" || sheetCols[2][i] != "") {
|
||
return errors.New("格式错误，缺少一级分类")
|
||
} else if sheetCols[2][i] != "" && sheetCols[1][i] == "" {
|
||
return errors.New("格式错误，缺少二级分类")
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// 校验商品资料导入规则
|
||
// 导入商品资料校验报错：必填项缺少数据（除了产地、备注，其他都是必填项）
|
||
func checkCommodityExcel(sheetCols [][]string) error {
|
||
if len(sheetCols) != 14 {
|
||
return errors.New("模版错误，请检查文件")
|
||
}
|
||
|
||
for i := 0; i < len(sheetCols)-2; i++ {
|
||
if len(sheetCols[i]) < len(sheetCols[i+1]) {
|
||
return errors.New("格式错误，有必填项未录入")
|
||
}
|
||
for _, v := range sheetCols[i] {
|
||
if v == "" {
|
||
return errors.New("格式错误，有必填项未录入")
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|