123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- package http
- import (
- "bytes"
- "encoding/csv"
- "fmt"
- "net/http"
- "go-common/app/admin/main/aegis/model"
- "go-common/library/log"
- "github.com/pkg/errors"
- )
- // CSVContentType
- var (
- CSVContentType = []string{"application/csv"}
- _ Render = CSV{}
- )
- // Render http reponse render.
- type Render interface {
- Render(http.ResponseWriter) error
- WriteContentType(w http.ResponseWriter)
- }
- // CSV str.
- type CSV struct {
- Content []byte
- Title string
- }
- // WriteContentType fn
- func (j CSV) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, CSVContentType, j.Title, "csv")
- }
- func writeContentType(w http.ResponseWriter, value []string, title, ext string) {
- header := w.Header()
- if val := header["Content-Type"]; len(val) == 0 {
- header["Content-Type"] = value
- }
- header["Content-Disposition"] = append(header["Content-Disposition"], fmt.Sprintf("attachment; filename=%s.%s", title, ext))
- }
- // Render (JSON) writes data with json ContentType.
- func (j CSV) Render(w http.ResponseWriter) (err error) {
- if _, err = w.Write(j.Content); err != nil {
- err = errors.WithStack(err)
- }
- return
- }
- // FormatCSV format csv data.
- func FormatCSV(records [][]string) (res []byte) {
- buf := new(bytes.Buffer)
- w := csv.NewWriter(buf)
- for _, record := range records {
- if err := w.Write(record); err != nil {
- log.Error("error writing record to csv:", err)
- return
- }
- }
- w.Flush()
- res = buf.Bytes()
- return
- }
- func formatReport(wl map[string][]*model.ReportFlowItem) (data [][]string, err error) {
- if len(wl) < 0 {
- return
- }
- tab := []string{"审核员"}
- for k, v := range wl {
- var fields []string
- fields = append(fields, k)
- for _, item := range v {
- fields = append(fields, fmt.Sprintf("%d/%d", item.Out, item.In))
- if len(tab) != len(v)+1 {
- tab = append(tab, item.Hour)
- }
- }
- data = append(data, fields)
- }
- data = append([][]string{tab}, data...)
- return
- }
- func formatReportTaskSubmit(res *model.ReportSubmitRes) (data [][]string, err error) {
- if res == nil || len(res.Order) <= 0 || len(res.Header) <= 0 || len(res.Rows) <= 0 {
- return
- }
- llen := len(res.Order)
- header := make([]string, llen)
- for i, k := range res.Order {
- header[i] = res.Header[k]
- }
- data = [][]string{header}
- for _, list := range res.Rows {
- for _, one := range list {
- fields := make([]string, llen)
- for i, k := range res.Order {
- fields[i] = one[k]
- }
- data = append(data, fields)
- }
- }
- return
- }
|