先本地生成,后再读取输出
csv 文件的创建和生成请参考我的文章 《Golang CSV操作》这里是先在本地生成 abc.csv 文件,然后再去读取csv中的内容来生成可下载文件,这个下载方法不加锁,是不安全的
package main
import (
"encoding/csv"
"io"
"net/http"
"os"
)
func main() {
http.HandleFunc("/", func(w http.ResponseWriter, request *http.Request) {
filename := "abc.csv"
b, err := os.Create(filename)
if err != nil {
panic(err)
}
_, _ = b.WriteString("\xEF\xBB\xBF") //写入UTF-8 BOM 防止乱码
writer := csv.NewWriter(b)
_ = writer.Write([]string{"name", "age", "sex"})
_ = writer.Write([]string{"kiki", "29", "man"})
writer.Flush()
w.Header().Set("Content-Type", "application/octet-stream")
w.Header().Set("Content-Disposition", "attachment; filename=\""+filename+"\"")
open, err := os.Open(filename)
if err != nil {
panic(err)
}
all, err := io.ReadAll(open)
if err != nil {
panic(err)
}
_, _ = w.Write(all)
})
_ = http.ListenAndServe(":8080", nil)
}
直接输出CSV
这个就是直接输出csv,不在本地存储要下载的文件
package main
import (
"encoding/csv"
"net/http"
"strings"
)
func main() {
http.HandleFunc("/", func(w http.ResponseWriter, request *http.Request) {
var b strings.Builder
_, _ = b.WriteString("\xEF\xBB\xBF") //写入UTF-8 BOM 防止乱码
writer := csv.NewWriter(&b)
_ = writer.Write([]string{"name", "age", "sex"})
_ = writer.Write([]string{"kiki", "29", "man"})
writer.Flush()
w.Header().Set("Content-Type", "application/octet-stream")
w.Header().Set("Content-Disposition", "attachment; filename=\"abc.csv\"")
_, _ = w.Write([]byte(b.String()))
})
_ = http.ListenAndServe(":8080", nil)
}