123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193 |
- package backend
- import (
- "context"
- "fmt"
- "kpt-pasture/config"
- "kpt-pasture/model"
- "kpt-pasture/util"
- "mime/multipart"
- "os"
- "path/filepath"
- "strconv"
- "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 [][]string) error {
- // 获取当前用户信息
- userModel, err := s.GetUserModel(ctx)
- if err != nil {
- return err
- }
- penMap := model.PenMap
- // 验证牧场ID
- 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{}
- 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
- }
- zaplog.Error("eventEnterList", zap.Any("eventEnterList", eventEnterList))
- return nil
- }
|