1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- package dao
- import (
- "context"
- recsys "go-common/app/service/bbq/recsys/api/grpc/v1"
- "go-common/app/service/bbq/recsys/model"
- "go-common/app/service/bbq/recsys/service/retrieve"
- "go-common/library/cache/redis"
- "go-common/library/log"
- "strconv"
- "strings"
- )
- //DownGradeRecall ...
- func (d *Dao) DownGradeRecall(c context.Context) (response *recsys.RecsysResponse, err error) {
- conn := d.redis.Get(c)
- defer conn.Close()
- key := retrieve.RecallHotDefault
- var str string
- if str, err = redis.String(conn.Do("GET", key)); err != nil {
- if err == redis.ErrNil {
- err = nil
- } else {
- log.Errorw(c, "recall", "get hot recall error", "err", err)
- }
- }
- response = new(recsys.RecsysResponse)
- response.Message = make(map[string]string)
- records := make([]*recsys.RecsysRecord, 0)
- response.Message[model.ResponseDownGrade] = "2"
- for _, svidStr := range strings.Split(str, ",") {
- svid, _ := strconv.ParseInt(svidStr, 10, 64)
- record := &recsys.RecsysRecord{
- Svid: svid,
- Score: 0,
- Map: make(map[string]string),
- }
- record.Map[model.RecallClasses] = retrieve.HotRecall
- records = append(records, record)
- }
- key = retrieve.RecallOpVideoKey
- if str, err = redis.String(conn.Do("GET", key)); err != nil {
- if err == redis.ErrNil {
- err = nil
- } else {
- log.Errorw(c, "recall", "get selection recall error", "err", err)
- }
- }
- response = new(recsys.RecsysResponse)
- response.Message = make(map[string]string)
- for _, svidStr := range strings.Split(str, ",") {
- svid, _ := strconv.ParseInt(svidStr, 10, 64)
- record := &recsys.RecsysRecord{
- Svid: svid,
- Score: 0,
- Map: make(map[string]string),
- }
- record.Map[model.RecallClasses] = retrieve.SelectionRecall
- records = append(records, record)
- }
- response.List = records
- return
- }
|