103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package archive
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"go-common/app/service/main/archive/api"
|
|
"go-common/library/cache/memcache"
|
|
"go-common/library/log"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
_prefix = "a3p_"
|
|
_prxfixSt = "stp_"
|
|
)
|
|
|
|
func keyArc(aid int64) string {
|
|
return _prefix + strconv.FormatInt(aid, 10)
|
|
}
|
|
|
|
func keySt(aid int64) string {
|
|
return _prxfixSt + strconv.FormatInt(aid, 10)
|
|
}
|
|
|
|
// statsCache get stat cache by aids
|
|
func (d *Dao) statsCache(c context.Context, aids []int64) (cached map[int64]*api.Stat, missed []int64, err error) {
|
|
cached = make(map[int64]*api.Stat, len(aids))
|
|
var (
|
|
conn = d.mc.Get(c)
|
|
keys = make([]string, 0, len(aids))
|
|
rs map[string]*memcache.Item
|
|
)
|
|
defer conn.Close()
|
|
for _, aid := range aids {
|
|
keys = append(keys, keySt(aid))
|
|
}
|
|
if rs, err = conn.GetMulti(keys); err != nil {
|
|
err = errors.Wrapf(err, "%v", keys)
|
|
return
|
|
}
|
|
for _, item := range rs {
|
|
var st = &api.Stat{}
|
|
if err = conn.Scan(item, st); err != nil {
|
|
log.Error("conn.Scan(%s) error(%v)", item.Value, err)
|
|
err = nil
|
|
continue
|
|
}
|
|
cached[st.Aid] = st
|
|
}
|
|
if len(cached) == len(aids) {
|
|
return
|
|
}
|
|
for _, aid := range aids {
|
|
if _, ok := cached[aid]; !ok {
|
|
missed = append(missed, aid)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// arcsCache get archives cache.
|
|
func (d *Dao) arcsCache(c context.Context, aids []int64) (cached map[int64]*api.Arc, missed []int64, err error) {
|
|
var (
|
|
keys = make([]string, 0, len(aids))
|
|
conn = d.mc.Get(c)
|
|
aidmap = make(map[string]int64, len(aids))
|
|
rs map[string]*memcache.Item
|
|
a *api.Arc
|
|
)
|
|
cached = make(map[int64]*api.Arc, len(aids))
|
|
defer conn.Close()
|
|
for _, aid := range aids {
|
|
k := keyArc(aid)
|
|
if _, ok := aidmap[k]; !ok {
|
|
keys = append(keys, k)
|
|
aidmap[k] = aid
|
|
}
|
|
}
|
|
if rs, err = conn.GetMulti(keys); err != nil {
|
|
err = errors.Wrapf(err, "%v", keys)
|
|
return
|
|
}
|
|
for k, r := range rs {
|
|
a = &api.Arc{}
|
|
if err = conn.Scan(r, a); err != nil {
|
|
log.Error("conn.Scan(%s) error(%v)", r.Value, err)
|
|
err = nil
|
|
continue
|
|
}
|
|
cached[aidmap[k]] = a
|
|
// delete hit key
|
|
delete(aidmap, k)
|
|
}
|
|
// missed key
|
|
missed = make([]int64, 0, len(aidmap))
|
|
for _, aid := range aidmap {
|
|
missed = append(missed, aid)
|
|
}
|
|
return
|
|
}
|