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"
	"time"

	"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 []map[string]string) error {
	// 获取当前用户信息
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return err
	}

	// 验证牧场ID
	if userModel.AppPasture.Id <= 0 {
		return xerr.Custom("无效的牧场ID")
	}

	cowList := make([]*model.Cow, 0)

	// 批量处理数据
	for _, row := range data {
		// 验证必要字段
		if _, ok := row["耳号"]; !ok {
			return xerr.Custom("缺少必要字段: 耳号")
		}

		sex := pasturePb.Genders_Female
		if _, ok := row["性别"]; !ok {
			return xerr.Custom("缺少必要字段: 性别")
		} else {
			if row["性别"] == "公" {
				sex = pasturePb.Genders_Male
			}
		}

		// 创建或更新牛只信息
		cow := &model.Cow{
			PastureId: userModel.AppPasture.Id,
			EarNumber: row["耳号"],
			Sex:       sex,
		}

		// 检查牛只是否已存在
		var existingCow model.Cow
		if err = s.DB.Model(new(model.Cow)).
			Where("pasture_id = ? AND ear_number = ?", userModel.AppPasture.Id, cow.EarNumber).
			First(&existingCow).Error; err == nil {

		} else {
			cowList = append(cowList, cow)
		}
	}

	zaplog.Info("ImportExcel", zap.Any("cowList", cowList))

	return nil
}