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.
44 lines
1.1 KiB
44 lines
1.1 KiB
package http |
|
|
|
import ( |
|
"bytes" |
|
"encoding/csv" |
|
"strconv" |
|
|
|
"go-common/app/admin/main/creative/model/whitelist" |
|
"go-common/library/log" |
|
) |
|
|
|
// 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 formatWhilteList(wl []*whitelist.Whitelist) (data [][]string, err error) { |
|
if len(wl) < 0 { |
|
return |
|
} |
|
data = append(data, []string{"MID", "昵称", "AdminMID", "备注", "粉丝数", "等级", "创建时间"}) |
|
for _, v := range wl { |
|
var fields []string |
|
fields = append(fields, strconv.FormatInt(int64(v.MID), 10)) |
|
fields = append(fields, v.Name) |
|
fields = append(fields, strconv.FormatInt(int64(v.AdminMID), 10)) |
|
fields = append(fields, v.Comment) |
|
fields = append(fields, strconv.FormatInt(int64(v.Fans), 10)) |
|
fields = append(fields, strconv.FormatInt(int64(v.CurrentLevel), 10)) |
|
fields = append(fields, v.Ctime) |
|
data = append(data, fields) |
|
} |
|
return |
|
}
|
|
|