You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
2.4 KiB
113 lines
2.4 KiB
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 |
|
}
|
|
|