102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"crypto/md5"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"net/http"
|
||
"net/url"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"go-common/app/interface/openplatform/article/model"
|
||
"go-common/library/log"
|
||
"go-common/library/xstr"
|
||
)
|
||
|
||
var _queryStr = `{"select":[{"name":"tid"},{"name":"oid"},{"name":"log_date"}],"where":{"tid":{"in":[%s]}},"page":{"limit":10,"skip":0}}`
|
||
|
||
// BerserkerTagArts .
|
||
func (d *Dao) BerserkerTagArts(c context.Context, tags []int64) (aids []int64, err error) {
|
||
var (
|
||
query string
|
||
res struct {
|
||
Code int
|
||
Msg string
|
||
Result []struct {
|
||
Tid int64 `json:"tid"`
|
||
Oid string `json:"oid"`
|
||
LogDate string `json:"log_date"`
|
||
}
|
||
}
|
||
tmps = make(map[int64]bool)
|
||
aid int64
|
||
date time.Time
|
||
now = time.Now()
|
||
)
|
||
query = fmt.Sprintf(_queryStr, xstr.JoinInts(tags))
|
||
if err = d.berserkerQuery(c, query, &res); err != nil {
|
||
return
|
||
}
|
||
if res.Code != 200 {
|
||
log.Error("s.BerserkerTagArts.query code(%d) msg(%s)", res.Code, res.Msg)
|
||
return
|
||
}
|
||
for _, v := range res.Result {
|
||
if date, err = time.Parse("20060102", v.LogDate); err != nil {
|
||
log.Error("s.BerserkerTagArts.time.Parse(%s) error(%+v)", v.LogDate, err)
|
||
return
|
||
}
|
||
if now.Sub(date) > time.Hour*60 {
|
||
continue
|
||
}
|
||
ids := strings.Split(v.Oid, "，")
|
||
var ts []int64
|
||
for _, id := range ids {
|
||
if aid, err = strconv.ParseInt(id, 10, 64); err != nil {
|
||
log.Error("s.BerserkerTagArts.ParseInt(%s) error(%+v)", id, err)
|
||
return
|
||
}
|
||
if !tmps[aid] {
|
||
aids = append(aids, aid)
|
||
tmps[aid] = true
|
||
}
|
||
ts = append(ts, aid)
|
||
}
|
||
d.AddCacheAidsByTag(c, v.Tid, &model.TagArts{Tid: v.Tid, Aids: ts})
|
||
}
|
||
return
|
||
}
|
||
|
||
func (d *Dao) berserkerQuery(c context.Context, query string, res interface{}) (err error) {
|
||
var (
|
||
params = url.Values{}
|
||
now = time.Now().Format("2006-01-02 15:04:05")
|
||
sign string
|
||
req *http.Request
|
||
)
|
||
sign = d.sign(now)
|
||
params.Set("appKey", d.c.Berserker.AppKey)
|
||
params.Set("signMethod", "md5")
|
||
params.Set("timestamp", now)
|
||
params.Set("version", "1.0")
|
||
params.Set("query", query)
|
||
params.Set("sign", sign)
|
||
|
||
req, err = http.NewRequest(http.MethodGet, d.c.Berserker.URL+"?"+params.Encode(), nil)
|
||
if err != nil {
|
||
log.Error("d.berserkerQuery.NewRequest error(%+v)", err)
|
||
return
|
||
}
|
||
return d.httpClient.Do(c, req, res)
|
||
}
|
||
|
||
// Sign calc appkey and appsecret sign.
|
||
func (d *Dao) sign(ts string) string {
|
||
str := d.c.Berserker.AppSecret + "appKey" + d.c.Berserker.AppKey + "timestamp" + ts + "version1.0" + d.c.Berserker.AppSecret
|
||
mh := md5.Sum([]byte(str))
|
||
return strings.ToUpper(hex.EncodeToString(mh[:]))
|
||
}
|