91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package service
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"encoding/csv"
|
||
"io"
|
||
"strconv"
|
||
"time"
|
||
|
||
coinclient "go-common/app/service/main/coin/api"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
// FormatCSV format csv data.
|
||
func (s *Service) 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(%+v) writing record to csv:", err)
|
||
return
|
||
}
|
||
}
|
||
w.Flush()
|
||
res = buf.Bytes()
|
||
return
|
||
}
|
||
|
||
// Upload http upload file.
|
||
func (s *Service) Upload(c context.Context, fileName, fileType string, expire int64, body io.Reader) (location string, err error) {
|
||
if location, err = s.uploadDao.Upload(c, fileName, fileType, expire, body); err != nil {
|
||
log.Error("s.upload.Upload() error(%v)", err)
|
||
}
|
||
return
|
||
}
|
||
|
||
// AnnualCoins .
|
||
func (s *Service) AnnualCoins(c context.Context, reader *csv.Reader) (fmid []int64) {
|
||
for {
|
||
record, err := reader.Read()
|
||
if err == io.EOF {
|
||
log.Warn("AnnualCoins is over!")
|
||
err = nil
|
||
break
|
||
}
|
||
if err != nil {
|
||
log.Error("AnnualCoins(%+v) Error(%+v)", record, err)
|
||
err = nil
|
||
continue
|
||
}
|
||
log.Info("AnnualCoins record(%+v)", record)
|
||
if len(record) < 2 {
|
||
log.Error("AnnualCoins record(%+v) len<2", record)
|
||
continue
|
||
}
|
||
mid, err := strconv.ParseInt(record[0], 10, 64)
|
||
if err != nil {
|
||
log.Error("AnnualCoins strconv.ParseInt mid(%+v) err(%+v)", record, err)
|
||
err = nil
|
||
continue
|
||
}
|
||
coins, err := strconv.ParseInt(record[1], 10, 64)
|
||
if err != nil {
|
||
log.Error("AnnualCoins strconv.ParseInt coins(%+v) err(%+v)", record, err)
|
||
err = nil
|
||
continue
|
||
}
|
||
arg := &coinclient.ModifyCoinsReq{
|
||
Mid: mid,
|
||
Count: float64(coins),
|
||
Reason: "风纪委奖励",
|
||
IP: "",
|
||
Operator: "credit",
|
||
CheckZero: 1,
|
||
Ts: time.Now().Unix(),
|
||
}
|
||
_, err = s.coinClient.ModifyCoins(context.Background(), arg)
|
||
if err != nil {
|
||
fmid = append(fmid, mid)
|
||
log.Error("ModifyCoins arg(%+v), err(%+v)", arg, err)
|
||
continue
|
||
}
|
||
if err = s.msgDao.SendSysMsg(context.Background(), mid, "您的风纪委硬币礼包已到账", "Hi 风纪委员你好，你的2018年风纪委硬币礼包已到账！快进入你的硬币账户看看吧！"); err != nil {
|
||
log.Error("SendSysMsg mid(%d), err(%+v)", mid, err)
|
||
}
|
||
time.Sleep(time.Duration(50) * time.Millisecond)
|
||
}
|
||
return
|
||
}
|