蝙蝠岛资源网 Design By www.hbtsch.com
                                业务背景
业务需求要求开发一个异步上传文件的接口,并支持上传进度的查询。
需求分析
ZIP压缩包中,包含一个csv文件和一个图片文件夹,要求:解析csv数据存入mongo,将图片文件夹中的图片信息对应上csv中的人员信息。
ZIP压缩包解压
使用golang自带的 "archive/zip" 包解压。
func decompressZip(filePath, dest string) (string, string, error) {
  var csvName string
  imageFolder := path.Base(filePath)
  ext := path.Ext(filePath)
  folderName := strings.TrimSuffix(imageFolder, ext)
  src, err := os.Open(filePath)
  if err != nil {
    return "", "", err
  }
  defer src.Close()
  zipFile, err := zip.OpenReader(src.Name())
  if err != nil {
    return "", "", err
  }
  defer zipFile.Close()
  err = os.MkdirAll(path.Join(dest, folderName), os.ModePerm)
  for _, innerFile := range zipFile.File {
    info := innerFile.FileInfo()
    if info.IsDir() {
      continue
    }
    dst, err := os.Create(path.Join(dest, folderName, info.Name()))
    if err != nil {
      fmt.Println(err.Error())
      continue
    }
    src, err := innerFile.Open()
    if err != nil {
      fmt.Println(err.Error())
      continue
    }
    io.Copy(dst, src)
  }
  destPath, err := ioutil.ReadDir(path.Join(dest, folderName))
  if err != nil {
    return "", "", err
  }
  for _, v := range destPath {
    if path.Ext(v.Name()) == ".csv" {
      csvName = path.Join(dest, folderName, v.Name())
    }
  }
  return folderName, csvName, nil
}
在这个解压的过程中,压缩包的树结构只能到2层
import.zip ┝┅┅import.csv ┖┅┅images
在解压后,所有的文件都会在同一个目录下,既images中的图片会变成和.csv文件同级
验证csv文件编码格式是否为UTF-8
func ValidUTF8(buf []byte) bool {
  nBytes := 0
  for i := 0; i < len(buf); i++ {
    if nBytes == 0 {
      if (buf[i] & 0x80) != 0 { //与操作之后不为0,说明首位为1
        for (buf[i] & 0x80) != 0 {
          buf[i] <<= 1 //左移一位
          nBytes++   //记录字符共占几个字节
        }
        if nBytes < 2 || nBytes > 6 { //因为UTF8编码单字符最多不超过6个字节
          return false
        }
        nBytes-- //减掉首字节的一个计数
      }
    } else { //处理多字节字符
      if buf[i]&0xc0 != 0x80 { //判断多字节后面的字节是否是10开头
        return false
      }
      nBytes--
    }
  }
  return nBytes == 0
}
后续支持utf-8转码
这个utf8编码判断方法是网上down下来的,后续优化一下
主逻辑
type LineWrong struct {
  LineNumber int64 `json:"line_number"`
  Msg    string `json:"msg"`
}
func Import(/*自定义参数*/){
  // decompress zip file to destination address
  folder, csvName, err := Decompress(path.Join(constant.FolderPrefix, req.FilePath), dest)
  if err != nil {
    fmt.Println(err.Error())
  }
  // check if the file encoding is utf8
  b, err := ioutil.ReadFile(csvName)
  if err != nil {
    fmt.Println(err.Error())
  }
  if !utils.ValidUTF8(b) {
    fmt.Println(errors.New("数据编码错误,请使用utf-8格式csv!"))
  }
  // create goroutine to analysis data into mongodb
  var wg sync.WaitGroup
  wg.Add(1)
  // used to interrupt goroutine
  resultChan := make(chan error)
  // used to record wrong row in csv
  lW := make(chan []LineWrong)
  go func(ctx *gin.Context, Name, csvPath, dir, folder string) {
    defer wg.Done()
    tidT, ciT, lwT, err := importCsv(ctx, Name, csvPath, dir, folder)
    resultChan <- err
    if err != nil {
      fmt.Println(err.Error())
    }
    lW <- lwT
    if len(lwT) == 0 {
      importClassData(ctx, tidT, ciT)
    }
  }(ctx, req.Name, csvName, dest, folder)
  err = <-resultChan
  lineWrong := <-lW
  close(lW)
  ···
}
// pre-analysis data in csv and through wrong data with line numbers and information
func importCsv()(){
  ···
}
// analysis data again and save data into mongodb, if is there any error,through them same as import()
func importClassData()(){
  ···
  conn, err := connect()
  if err != nil {
    return err
  }
  defer conn.Close()
  conn.Do("hset", taskId, "task_id", (curLine*100)/totalLines)
  ···
}
将错误信息以channel接收,使用 package "sync" 的 sync.WaitGroup 控制异步协程。在入库的过程中,将当前的进度存入redis。
查询进度接口
func QueryImport()(){
  conn, err := connect()
  if err != nil {
    return nil, err
  }
  defer conn.Close()
  progress, _ := conn.Do("hget", key, field)
  if pro, ok := progress.([]uint8); ok {
    ba := []byte{}
    for _, b := range pro {
      ba = append(ba, byte(b))
    }
    progress,_ = strconv.Atoi(string(ba))
  }
  return progress
}
从redis中取出来的数据是[]uint8类型数据,先断言,然后转类型返回。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
蝙蝠岛资源网 Design By www.hbtsch.com
                            
                                广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
                        免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
蝙蝠岛资源网 Design By www.hbtsch.com
                        暂无Golang实现异步上传文件支持进度条查询的方法的评论...
                                    更新日志
2025年10月31日
                                2025年10月31日
                    - 小骆驼-《草原狼2(蓝光CD)》[原抓WAV+CUE]
- 群星《欢迎来到我身边 电影原声专辑》[320K/MP3][105.02MB]
- 群星《欢迎来到我身边 电影原声专辑》[FLAC/分轨][480.9MB]
- 雷婷《梦里蓝天HQⅡ》 2023头版限量编号低速原抓[WAV+CUE][463M]
- 群星《2024好听新歌42》AI调整音效【WAV分轨】
- 王思雨-《思念陪着鸿雁飞》WAV
- 王思雨《喜马拉雅HQ》头版限量编号[WAV+CUE]
- 李健《无时无刻》[WAV+CUE][590M]
- 陈奕迅《酝酿》[WAV分轨][502M]
- 卓依婷《化蝶》2CD[WAV+CUE][1.1G]
- 群星《吉他王(黑胶CD)》[WAV+CUE]
- 齐秦《穿乐(穿越)》[WAV+CUE]
- 发烧珍品《数位CD音响测试-动向效果(九)》【WAV+CUE】
- 邝美云《邝美云精装歌集》[DSF][1.6G]
- 吕方《爱一回伤一回》[WAV+CUE][454M]
 
                        