文档：https://xuri.me/excelize/zh-hans/cell.html?q=
读取本地文件与文件流：https://blog.csdn.net/qq_41119959/article/details/122128547

excel文件流批量插入实例

  1. const (
  2.     ExcelTitle_Phone     = "手机号"
  3.     ExcelTitle_Name      = "姓名"
  4.     ExcelTitle_Place     = "位置"
  5. )
  7. func CreateInBatches(devId int, file *multipart.FileHeader) error {
  8.        var (
  9.         wg sync.WaitGroup    
  10.     )
  11.     // 1.处理参数
  13.     // 2.为文件提供阅读器(将保存在内存中)
  14.     excelFile, err := file.Open()
  15.     if err != nil {
  16.         return err
  17.     }
  18.     // 3.流式读取(默认读取第一个sheet),返回所有的单元格内容
  19.     rows, err := utils.ReadExcelFile(excelFile)
  20.     if err != nil {
  21.         return err
  22.     }
  23.     // 4.初始化一个map
  24.     // 存放excel的对应内容及其索引，例如 手机号:0，代表第一列为手机号
  25.     excelMap := make(map[string]int)
  26.     // 从excel中取出所有的标题及其索引
  27.     for i, key := range rows[0] {
  28.         excelMap[key] = i
  29.     }
  31.     // 判断标题除外的数据有多少行
  32.     valuelen := len(rows[1:])
  33.     // 每一千行为一批次，向上取整
  34.     batch := int(math.Ceil(float64(valuelen) / float64(1000)))
  35.     // 初始化一个map，记录每个批次的索引及其数据量
  36.     batchMap := make(map[int]int)
  37.     for i := 0; i < batch; i++ {
  38.         // 记录每一个批次对应的数据界限，例如 batchMap[0] = 1000 batchMap[1] = 2000
  39.         // 如果不足1000条就以具体的行数valuelen为底线
  40.         if batch == 1 || i == batch-1 {
  41.             // <= 1000 行数据 || 例如 5290 中的 290
  42.             batchMap[i] = valuelen
  43.         } else {
  44.             // > 1000 行数据
  45.             batchMap[i] = (i + 1) * 1000
  46.         }
  48.         wg.Add(1)
  49.         go func(i int, batchMap map[int]int) {
  50.             defer wg.Done()
  51.             var userScheduleList []schedule.UserSchedule
  53.             // 5.从excel中取出指定范围的所有的值
  54.             // 每次循环起点也需要变化，第一次为0，后面就是上一次的界限batchMap[i-1]
  55.             // 因为go没有三元表达式，用这个代替
  56.             startIndex := map[bool]int{true: 0, false: batchMap[i-1]}[i == 0]
  58.             // 因为切片[1:n]只能取到[1,n)，所以需要给 batchMap[i]+1
  59.             for _, row := range rows[1+startIndex : batchMap[i]+1] {
  60.                 user := schedule.UserSchedule{
  61.                     // 从索引中取到各个标题对应的值
  62.                     PhoneNum: row[excelMap[ExcelTitle_Phone]],
  63.                     Name:     row[excelMap[ExcelTitle_Name]],
  64.                     Place:    row[excelMap[ExcelTitle_Place]],
  65.                     DevId:    &devId,
  66.                 }
  67.                 userScheduleList = append(userScheduleList, user)
  68.             }
  69.             // 6.做数据库的批量插入操作
  70.             // 创建db
  71.             db := global.GVA_DB.Model(&schedule.UserSchedule{})
  72.             // 每次限制批量插入1000条
  73.             if err = db.CreateInBatches(userScheduleList, 1000).Error; err != nil {
  74.                 // handle log
  75.                 return
  76.             }
  77.         }(i,batchMap)
  78.     }
  79.     wg.Wait()
  81.     if err != nil {
  82.         return err
  83.     }
  85.     return nil
  86. }

ReadExcelFile

  1. func ReadExcelFile(file File, tableName ...string) ([][]string, error) {
  2.     f, err := excelize.OpenReader(file)
  3.     if err != nil {
  4.         return nil, err
  5.     }
  6.     //默认读取第一个
  7.     firstSheet := ""
  8.     if len(tableName) > 0 {
  9.         firstSheet = tableName[0]
  10.     } else {
  11.         firstSheet = f.GetSheetName(0)
  12.     }
  13.     rows, err := f.GetRows(firstSheet)
  14.     return rows, err
  15. }