kpt-pasture/module/backend/upload_file.go

package backend

import (
	"context"
	"fmt"
	"kpt-pasture/config"
	"kpt-pasture/model"
	"kpt-pasture/util"
	"mime/multipart"
	"os"
	"path/filepath"
	"strconv"
	"time"

	"gorm.io/gorm"

	"gitee.com/xuyiping_admin/pkg/logger/zaplog"
	"go.uber.org/zap"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"

	"gitee.com/xuyiping_admin/pkg/xerr"
)

func (s *StoreEntry) Photos(ctx context.Context, files []*multipart.FileHeader) ([]string, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, err
	}

	workDir := fmt.Sprintf("%s", config.WorkDir)
	pathDir := fmt.Sprintf("/files/photos/%d/%s", userModel.AppPasture.Id, time.Now().Local().Format("20060102"))
	saveDir := filepath.Join(workDir, pathDir)
	if _, err = os.Stat(saveDir); os.IsNotExist(err) {
		if err = os.MkdirAll(saveDir, 0755); err != nil {
			return nil, xerr.Customf("创建目录失败: %s", err.Error())
		}
	}

	// 处理每个文件
	filePaths := make([]string, len(files))
	for i, file := range files {
		contentType := file.Header.Get("Content-Type")
		if contentType != "image/jpeg" && contentType != "image/png" && contentType != "image/gif" {
			return nil, xerr.Customf("图片格式错误: %s", file.Filename)

		}
		if file.Size > 1024*1024*5 {
			return nil, xerr.Customf("单个图片文件不能超过5MB")

		}
		ext := filepath.Ext(file.Filename)
		if ext == "" {
			switch contentType {
			case "image/jpeg":
				ext = ".jpg"
			case "image/png":
				ext = ".png"
			case "image/gif":
				ext = ".gif"
			default:
				ext = ".jpg" // 默认
			}
		}
		randomName := util.GenerateRandomNumberString(32)
		finalFilename := randomName + ext
		fPath := filepath.Join(saveDir, finalFilename)
		urlPath := filepath.Join(pathDir, finalFilename)

		if err = util.SaveUploadedFile(file, fPath); err != nil {
			return nil, xerr.Customf("保存文件失败: %s", err.Error())
		}
		filePaths[i] = urlPath
	}
	return filePaths, nil
}

func (s *StoreEntry) ImportExcel(ctx context.Context, data [][]string) error {
	// 获取当前用户信息
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return err
	}

	penMap := model.PenMap
	if userModel.AppPasture.Id <= 0 {
		return xerr.Custom("无效的牧场ID")
	}

	// 处理Excel数据
	//headers := data[0]
	eventEnterList := make([]*pasturePb.EventEnterRequest, 0)
	for _, row := range data[1:] {
		if len(row) <= 0 {
			continue
		}
		ts := &pasturePb.EventEnterRequest{
			OperationId:   33,
			OperationName: "kpt_admin",
			MessengerId:   33,
			MessengerName: "kpt_admin",
		}
		zaplog.Error("row", zap.Any("row", row))
		for j, d := range row {
			switch j {
			case 0:
				continue
			case 1:
				continue
			case 2:
				ts.EarNumber = d
			case 3:
				if pn, ok := penMap[d]; ok {
					ts.PenId = pn
					ts.PenName = d
				}
			case 4:
				ts.Sex = pasturePb.Genders_Female
				if d == "公" {
					ts.Sex = pasturePb.Genders_Male
				}
			case 5:
				if d == "成母牛" {
					ts.CowType = pasturePb.CowType_Breeding_Calf
				} else if d == "犊牛" {
					ts.CowType = pasturePb.CowType_Lactating_Calf
				} else if d == "青年牛" {
					ts.CowType = pasturePb.CowType_Youth_Calf
				} else if d == "育成牛" {
					ts.CowType = pasturePb.CowType_Reserve_Calf
				}
			case 6:
				continue
			case 7:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !bat.IsZero() {
					ts.BirthAt = int32(bat.Local().Unix())
				}
			case 8:
				lact, _ := strconv.Atoi(d)
				ts.Lact = int32(lact)
			case 9:
				eat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !eat.IsZero() {
					ts.EnterAt = int32(eat.Local().Unix())
				}
			case 10:
				ts.FatherNumber = d
			case 11:
				ts.MotherNumber = d
			case 12:
				mat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !mat.IsZero() {
					ts.MatingAt = int32(mat.Local().Unix())
				}
			case 13:
				continue
			case 14:
				continue
			case 15:
				mt, _ := strconv.Atoi(d)
				ts.MatingTimes = int32(mt)
			case 16:
				continue
			case 17:
				pat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !pat.IsZero() {
					ts.PregnancyCheckAt = int32(pat.Local().Unix())
				}
			case 18:
				continue
			case 19:
				continue
			case 20:
				cat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !cat.IsZero() {
					ts.CalvingAt = int32(cat.Local().Unix())
				}
			case 21:
				continue
			case 22:
				continue
			case 23:
				continue
			case 24:
				ts.CowKind = pasturePb.CowKind_AGSN
			}
		}
		eventEnterList = append(eventEnterList, ts)
		break
	}

	if len(eventEnterList) <= 0 {
		return nil
	}

	pastureId := userModel.AppPasture.Id
	if pastureId != 4 {
		pastureId = 4
	}

	return s.ExecExcelData(ctx, pastureId, eventEnterList)
}

func (s *StoreEntry) ExecExcelData(ctx context.Context, pastureId int64, dataList []*pasturePb.EventEnterRequest) error {
	if len(dataList) <= 0 {
		return nil
	}

	return s.DB.Transaction(func(tx *gorm.DB) error {
		for _, data := range dataList {
			var count int64
			if err := tx.Model(new(model.Cow)).
				Where("pasture_id = ?", pastureId).
				Where("ear_number = ?", data.EarNumber).
				Count(&count).Error; err != nil {
				return err
			}
			if count > 0 {
				continue
			}

			if err := s.CreateEnter(ctx, data); err != nil {
				return err
			}
		}
		return nil
	})
}