go-common/app/job/main/tv/dao/ugc/report_cid.go
2019-04-22 18:49:16 +08:00

55 lines
1.4 KiB
Go

package ugc
import (
"bytes"
"context"
"encoding/json"
"fmt"
"net/http"
appDao "go-common/app/job/main/tv/dao/app"
ugcmdl "go-common/app/job/main/tv/model/ugc"
"go-common/library/log"
"go-common/library/xstr"
)
const (
_reported = 1
_finishReport = "UPDATE ugc_video SET mark = %d WHERE cid IN (%s)"
)
// RepCidBatch reports cid info to VideoCloud api
func (d *Dao) RepCidBatch(c context.Context, cidReq []*ugcmdl.CidReq) (err error) {
var (
reportURL = d.conf.UgcSync.Cfg.ReportCidURL
resp = ugcmdl.CidResp{}
jsonBody []byte
)
if jsonBody, err = json.Marshal(cidReq); err != nil {
log.Error("json.Marchal(%v) error(%v)", cidReq, err)
return
}
req, err := http.NewRequest(http.MethodPost, reportURL, bytes.NewBuffer(jsonBody))
if err != nil {
log.Error("http.NewRequest err - %v", err)
return
}
if err = d.client.Do(c, req, &resp); err != nil {
log.Error("ReportCid jsonBody (%s) error(%v)", string(jsonBody), err)
return
}
if resp.Code != 0 {
return fmt.Errorf("ReportCid Error Code:%d, Msg:%s", resp.Code, resp.Message)
}
appDao.PromInfo("ReportCid:Succ")
return
}
// FinishReport change's the ugc_video's state from 0 to 1, means it has already been reported
func (d *Dao) FinishReport(c context.Context, cids []int64) (err error) {
if _, err = d.DB.Exec(c, fmt.Sprintf(_finishReport, _reported, xstr.JoinInts(cids))); err != nil {
log.Error("FinishReport Error: %v", cids, err)
}
return
}