HTTP实现xls导出和上传文件处理

更新日期: 2019-04-13阅读: 2.7k标签: 文件

需求

  • 导出节目单[不保存文件到本地]
  • 输出不同的文件类型[csv,xls,xlsx]
  • 上传[csv,xls,xlsx]导入节目单


导出节目单

import (
    "bytes"
    "encoding/csv"
    "fmt"
    "net/http"
    "strings"

    "github.com/tealeg/xlsx"
)

//导出节目单的handler
func Export(w http.ResponseWriter, r *http.Request) {
    q := r.URL.Query()
    format := strings.ToLower(q.Get("format"))
    // 获取节目单
    ps := dbqueryprograms()

    // 设置文件头
    w.Header().Set("ContentType", "application/octet-stream")
    w.Header().Set("Content-Disposition", fmt.Sprintf("attachment;filename=节目单.%s", format))
    if format == "csv" {
        b := &bytes.Buffer{}
        wr := csv.NewWriter(b)
        wr.WriteString("\xEF\xBB\xBF")
        wr.Write([]string{"节目名称", "开始时间", "结束时间", "播出类型", "素材ID"})
        for _, p := range ps {
            wr.Write([]string{p.Name, p.StartTime, p.EndTime, p.Type, p.VideoId})
        }

        wr.Flush()
        w.Write(b.Bytes())
        return
    } else if format == "xls" || format == "xlsx" {
        file := xlsx.NewFile()
        sheet, err := file.AddSheet("sheet1")
        if err != nil {
            DoHttpError(w, "文件创建失败")
            return
        }

        title := &rowContent{"节目名称", "开始时间", "结束时间", "播出类型", "素材ID"}
        writeRow(sheet, title)

        for _, p := range ps {
            line := &rowContent{
                Name:        p.Name,
                Start:       p.StartTime,
                End:         p.EndTime,
                PlayingType: p.Type,
                VideoID:     p.VideoId,
            }
            writeRow(sheet, line)
        }
        // 直接输出到response
        file.Write(w)
        return
    }

    DoHttpError(w, "不支持的文件类型")
    return
}

func writeRow(sheet *xlsx.Sheet, c *rowContent) {
    row := sheet.AddRow()
    nameCell := row.AddCell()
    nameCell.Value = c.Name

    startCell := row.AddCell()
    startCell.Value = c.Start

    endCell := row.AddCell()
    endCell.Value = c.End

    typeCell := row.AddCell()
    typeCell.Value = c.PlayingType

    videoCell := row.AddCell()
    videoCell.Value = c.VideoID
}

type rowContent struct {
    Name, Start, End, PlayingType, VideoID string
}


文件上传

import (
    "encoding/csv"
    "io"
    "mime/multipart"
    "net/http"
    "path"
    "strings"

    "github.com/pkg/errors"
    "github.com/tealeg/xlsx"
)

func ImportHandler(w http.ResponseWriter, r *http.Request) {
    file, handler, err := r.FormFile("file")
    if err != nil {
        DoHttpError(w, "获取上传文件失败")
        return
    }
    defer file.Close()

    if handler.Size > 100*1024*1024 {
        DoHttpError(w, "上传不能超过100M")
        return
    }

    fileExt := strings.ToLower(path.Ext(handler.Filename))
    var content [][]string
    if fileExt == ".csv" {
        content, err = parseCsv(file)
        if err != nil {
            // handle err and log
            return
        }
    } else if fileExt == ".xls" || fileExt == ".xlsx" {
        content, err = parseXlsx(file, handler.Size)
        if err != nil {
            // handle err and log
            return
        }
    } else {
        DoHttpError(w, "不支持的类型")
        return
    }

    // db and handle err log
    // err := DbCreatePrograms(content)
    DoHttpSuccess(w, "解析成功", nil)
    return
}

// 解析csv文件
func parseCsv(file multipart.File) ([][]string, error) {
    result := [][]string{}
    r := csv.NewReader(file)
    for {
        record, err := r.Read()
        if err == io.EOF {
            break
        } else if err != nil {
            return nil, errors.Wrap(err, "Read csv failed")
        }

        var st = []string{}
        for i, j := range record {
            // 头信息过滤掉
            if i == 0 {
                continue
            }
            st = append(st, j)
        }
        result = append(result, st)
    }
    return result, nil
}

// 解析xlsx文件
func parseXlsx(file multipart.File, size int64) ([][]string, error) {
    result := [][]string{}
    xlFile, err := xlsx.OpenReaderAt(file, size)
    if err != nil {
        return nil, errors.Wrap(err, "Read file failed;")
    }

    for _, sheet := range xlFile.Sheets {
        // 不读隐藏的工作区
        if sheet.Hidden {
            continue
        }

        for i, row := range sheet.Rows {
            // 过滤头信息
            if i == 0 {
                continue
            }
            var st = []string{}
            for j, cell := range row.Cells {
                c, err := cell.FormattedValue()
                if err != nil {
                    return nil, errors.Wrapf(err, "Get (%v, %v) value failed;", i, j)
                }
                st = append(st, c)
            }
            result = append(result, st)
        }
    }
    return result, nil
}

链接: https://fly63.com/article/detial/2850

h5移动端实现图片文件上传

PC端上传文件多半用插件,引入flash都没关系,但是移动端要是还用各种冗余的插件估计得被喷死,项目里面需要做图片上传的功能,既然H5已经有相关的接口且兼容性良好,当然优先考虑用H5来实现。

前端使用js读取文件操作

首先我们定义一个input标签type=file、然后我们定义一个jsReadFiles的方法将文件作为参数;当上传文件的时候读取这个文件。图片上传成功,只是图片路径变成了base64编码的形式。

HTML5实现文件读取、编辑、保存

HTML5读取文件主要利用的就是FileReader这个API,它的使用需要从一个构造函数开始,保存文件的关键是生成文件对象,可以使用URL.createObjectURL()方法实现,该方法能返回给定对象的URL,用在<a>标签的href属性上就可以创建可下载的文件链接。

血淋淋的事实告诉你:你为什么不应该在JS文件中保存敏感信息

在JavaScript文件中存储敏感数据,不仅是一种错误的实践方式,而且还是一种非常危险的行为,长期以来大家都知道这一点。

在js文件中引入另一个js文件的实现方法总汇

比如我写了一个JS文件,这个文件需要调用另外一个JS文件,该如何实现呢?这篇文章主要介绍:在js文件中引入另一个js文件的实现

使用HTML5来实现本地文件读取和写入

最近有这样一个需求,就是在HTML页面中有个按钮导出,点击它,将构造一个文档并存储到本地文件系统中。另外还有个按钮,点击它,从本地文件系统中读取一个文件并对内容进行分析。

lock文件_我们为什么需要 lock 文件

从 Yarn 横空出世推出 lock 文件以来,已经两年多时间了,npm 也在 5.0 版本加入了类似的功能,lock 文件越来越被开发者们接收和认可。本篇文章想从前端视角探讨一下我们为什么需要 lock 文件,以及它的一些成本与风险,当然其中一些观点对于后端也是适用的

什么是断点续传?前端如何实现文件的断点续传

什么是断点续传?就是下载文件时,不必重头开始下载,而是从指定的位置继续下载,这样的功能就叫做断点续传。前端通过FileList对象获取到相应的文件,按照指定的分割方式将大文件分段,然后一段一段地传给后端,后端再按顺序一段段将文件进行拼接。

form表单文件上传_multipart/form-data文件上传

form表单的enctype属性:规定了form表单数据在发送到服务器时候的编码方式.。application/x-www-form-urlencoded:默认编码方式,multipart/form-data:指定传输数据为二进制数据,例如图片、mp3、文件,text/plain:纯文本的传输。空格转换为“+”,但不支持特殊字符编码。

使用HttpClient发送文件流到服务器端

适用场景: 网络绝对路径的URL文件或图片,不存储到本地,转换成stream,直接使用HTTPClient传送到SpringBoot的服务端,将文件存储下来,并返回一个文件地址。目前分层架构的系统越来越多这种需求,所以记录下来以备不时之需。

点击更多...

内容以共享、参考、研究为目的,不存在任何商业目的。其版权属原作者所有,如有侵权或违规,请与小编联系!情况属实本人将予以删除!