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.
120 lines
3.0 KiB
120 lines
3.0 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"encoding/json" |
|
"net/url" |
|
|
|
"go-common/app/interface/main/web/model" |
|
"go-common/library/cache/redis" |
|
"go-common/library/ecode" |
|
"go-common/library/log" |
|
"go-common/library/net/metadata" |
|
) |
|
|
|
const ( |
|
_wxHotURI = "hot-weixin.json" |
|
_wxCacheKey = "wx_hot" |
|
_wxBkCacheKey = _keyBakPrefix + _wxCacheKey |
|
) |
|
|
|
// WxHot get wx hot aids. |
|
func (d *Dao) WxHot(c context.Context) (aids []int64, err error) { |
|
ip := metadata.String(c, metadata.RemoteIP) |
|
var res struct { |
|
Code int `json:"code"` |
|
List []*model.NewArchive `json:"list"` |
|
} |
|
if err = d.httpBigData.Get(c, d.wxHotURL, ip, url.Values{}, &res); err != nil { |
|
log.Error("d.httpBigData.Get(%s) error(%v)", d.wxHotURL, err) |
|
return |
|
} |
|
if res.Code != ecode.OK.Code() { |
|
log.Error("d.httpBigData.Get(%s) error(%v)", d.wxHotURL, err) |
|
err = ecode.Int(res.Code) |
|
return |
|
} |
|
for _, v := range res.List { |
|
if v.Aid > 0 { |
|
aids = append(aids, v.Aid) |
|
} |
|
} |
|
return |
|
} |
|
|
|
// WxHotCache get wx hot cache. |
|
func (d *Dao) WxHotCache(c context.Context) (arcs []*model.WxArchive, err error) { |
|
key := _wxCacheKey |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
arcs, err = wxHotCache(conn, key) |
|
return |
|
} |
|
|
|
// WxHotBakCache get wx hot bak cache. |
|
func (d *Dao) WxHotBakCache(c context.Context) (arcs []*model.WxArchive, err error) { |
|
key := _wxBkCacheKey |
|
conn := d.redisBak.Get(c) |
|
defer conn.Close() |
|
arcs, err = wxHotCache(conn, key) |
|
return |
|
} |
|
|
|
func wxHotCache(conn redis.Conn, key string) (res []*model.WxArchive, err error) { |
|
var value []byte |
|
if value, err = redis.Bytes(conn.Do("GET", key)); err != nil { |
|
if err == redis.ErrNil { |
|
err = nil |
|
} else { |
|
log.Error("conn.Do(GET, %s) error(%v)", key, err) |
|
} |
|
return |
|
} |
|
res = []*model.WxArchive{} |
|
if err = json.Unmarshal(value, &res); err != nil { |
|
log.Error("json.Unmarshal(%v) error(%v)", value, err) |
|
} |
|
return |
|
} |
|
|
|
// SetWxHotCache set wx hot to cache. |
|
func (d *Dao) SetWxHotCache(c context.Context, arcs []*model.WxArchive) (err error) { |
|
key := _wxCacheKey |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
if err = d.setWxHotCache(c, conn, key, d.redisWxHotExpire, arcs); err != nil { |
|
return |
|
} |
|
key = _wxBkCacheKey |
|
connBak := d.redisBak.Get(c) |
|
err = d.setWxHotCache(c, connBak, key, d.redisWxHotBakExpire, arcs) |
|
connBak.Close() |
|
return |
|
} |
|
|
|
func (d *Dao) setWxHotCache(c context.Context, conn redis.Conn, key string, expire int32, arcs []*model.WxArchive) (err error) { |
|
var bs []byte |
|
if bs, err = json.Marshal(arcs); err != nil { |
|
log.Error("json.Marshal(%v) error (%v)", arcs, err) |
|
return |
|
} |
|
if err = conn.Send("SET", key, bs); err != nil { |
|
log.Error("conn.Send(SET, %s, %s) error(%v)", key, string(bs), err) |
|
return |
|
} |
|
if err = conn.Send("EXPIRE", key, expire); err != nil { |
|
log.Error("conn.Send(Expire, %s, %d) error(%v)", key, expire, err) |
|
return |
|
} |
|
if err = conn.Flush(); err != nil { |
|
log.Error("conn.Flush error(%v)", err) |
|
return |
|
} |
|
for i := 0; i < 2; i++ { |
|
if _, err = conn.Receive(); err != nil { |
|
log.Error("conn.Receive() error(%v)", err) |
|
return |
|
} |
|
} |
|
return |
|
}
|
|
|