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

时间: 2019-04-13阅读: 224标签: 文件

需求

  • 导出节目单[不保存文件到本地]
  • 输出不同的文件类型[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
}

PHP读取文件内容的方法

下面我们就为大家详细介绍PHP读取文件内容的两种方法。第一种方法:fread函数;第二种方法:file_get_contents函数,这个方法就非常简单了,我们直接使用file_get_contents函数来读取本地文件内容。

HTML5中FileReader的使用

HTML5定义了FileReader作为文件API的重要成员用于读取文件,FileReader接口提供了读取文件的方法和包含读取结果的事件模型。 FileReader 的实例拥有 4 个方法,其中 3 个用以读取文件,另一个用来中断读取。

前端实现文件下载功能

通过window.open()打开新页面下载文件;通过a标签打开新页面下载文件;通过文件流的方式下载;如何实现批量下载,且打包文件

CSS文件里引入另一个CSS文件

在HTML中引入css的其中的两个方法:使用链接式、使用导入式;使用链接式时,会在装载页面主体部分 之前装载css文件,这样现实出来的页面从一开始就是带有样式效果的

js下载文件,判断文件是否返回

下载之前用的window.location.href下载的,但是这个判别不了文件是否返回,小文件倒还好,大的文件长时间没有下载也没有加载条什么的,用户有时会点击下载好几下,大的数据很容易将服务拖垮

vue读取本地的excel文件并显示在网页上

我想实现读取一个本地的xlsx文件(task_list.xlsx)然后显示在网页上, 一开始选择的方法是建个express server, 通过发送axios请求来实现, 但是觉得只是读取一个本地文件还要搞个server太复杂了

node.js 中的fs.rename()模块

node.js的核心模块 原生模块。修改文件名称,可更改文件的存放路径,方法说明 : 修改文件名称,可更改文件的存放路径.

html5中利用FileReader来读取文件

利用FileReader来读取文件的能够来实现即时预览的效果,这个也是在html5中才有的功能,readAsText():读取文本文件(可以使用TxT打开的文件),返回文本字符串,默认编码是UTF-8

PHP下载大文件失败,并限制下载速度

PHP在使用readfile函数定义下载文件时候,文件不可以过大,否则会下载失败,文件损坏且不报错;这个是因为readfile读取文件的时候会把文件放入缓存,导致内存溢出;解决:分段下载,并限制下载速度;

vue在index.html中引入静态文件不生效

出现该问题的标志如下,控制台warning(Resource interpreted as Stylesheet but transferred with MIME type text/html),第一种可能出现原因就是引入的静态文件在src文件夹内,这种的解决办法就是把资源引入静态资源的目录static

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

广告赞助文章投稿关于web前端网站点搜索站长推荐网站地图站长QQ:522607023

小程序专栏: 土味情话心理测试脑筋急转弯幽默笑话段子句子语录成语大全