228 lines
7.2 KiB
Go
228 lines
7.2 KiB
Go
package data
|
||
|
||
import (
|
||
"context"
|
||
"encoding/binary"
|
||
"strconv"
|
||
"time"
|
||
|
||
"go-common/app/admin/main/up/model/datamodel"
|
||
"go-common/app/admin/main/up/util/hbaseutil"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
|
||
"github.com/tsuna/gohbase/hrpc"
|
||
)
|
||
|
||
// reverse for string.
|
||
func reverseString(s string) string {
|
||
rs := []rune(s)
|
||
l := len(rs)
|
||
for f, t := 0, l-1; f < t; f, t = f+1, t-1 {
|
||
rs[f], rs[t] = rs[t], rs[f]
|
||
}
|
||
ns := string(rs)
|
||
if l < 10 {
|
||
for i := 0; i < 10-l; i++ {
|
||
ns = ns + "0"
|
||
}
|
||
}
|
||
return ns
|
||
}
|
||
|
||
// 粉丝管理 - up_fans_analysis
|
||
// mid倒置补0 （补满共10位）+0 +yyyyMMdd – 累计数据
|
||
// mid倒置补0 （补满共10位）+1 +yyyyMMdd – 7日数据
|
||
// mid倒置补0 （补满共10位）+2 +yyyyMMdd – 30日数据
|
||
// mid倒置补0 （补满共10位）+3 +yyyyMMdd – 90日数据
|
||
func fansRowKey(id int64, ty int) string {
|
||
idStr := strconv.FormatInt(id, 10)
|
||
s := reverseString(idStr)
|
||
s = s + strconv.Itoa(ty) + time.Now().AddDate(0, 0, -1).Add(-12*time.Hour).Format("20060102")
|
||
return s
|
||
}
|
||
|
||
// 稿件索引表 - up_archive_query
|
||
// [mid倒置补0（10位）] + [投稿年月（4位）] + [原创/转载（1位）]
|
||
func arcQueryKey(id int64, dt string, cp int) string {
|
||
idStr := strconv.FormatInt(id, 10)
|
||
cpStr := strconv.Itoa(cp)
|
||
s := reverseString(idStr)
|
||
s = s + dt + cpStr
|
||
return s
|
||
}
|
||
|
||
// UpFansAnalysis for web fans analysis.
|
||
func (d *Dao) UpFansAnalysis(c context.Context, mid int64, ty int) (res *datamodel.FanInfo, err error) {
|
||
var (
|
||
result *hrpc.Result
|
||
ctx, cancel = context.WithTimeout(c, time.Duration(d.c.HBase.ReadTimeout))
|
||
tableName = HBaseUpFansAnalysis
|
||
//summary = make(map[string]int32) //total & classify proportion
|
||
//rankDr = make(map[string]int32) //播放时长 mid list
|
||
//rankVideoAct = make(map[string]int32) //视频互动 mid list
|
||
//rankDyAct = make(map[string]int32) //动态互动 mid list
|
||
rankMap = make(map[string]map[string]int32) //top 10 rank list
|
||
source = make(map[string]int32) //source proportion
|
||
rowkey = fansRowKey(mid, ty)
|
||
parser = hbaseutil.Parser{}
|
||
)
|
||
res = new(datamodel.FanInfo)
|
||
defer cancel()
|
||
log.Info("UpFansAnalysisForWeb mid(%d)|rowkey(%s)", mid, rowkey)
|
||
if result, err = d.hbase.GetStr(ctx, tableName, rowkey, hrpc.Families(map[string][]string{"f": nil})); err != nil {
|
||
log.Error("d.hbase.GetStr tableName(%s)|mid(%d)|rowkey(%v)|error(%v)", tableName, mid, rowkey, err)
|
||
err = ecode.CreativeDataErr
|
||
return
|
||
}
|
||
if result == nil || len(result.Cells) == 0 {
|
||
log.Warn("UpFansAnalysisForWeb no data tableName(%s)|mid(%d)|rowkey(%v)", tableName, mid, rowkey)
|
||
return
|
||
}
|
||
var summary datamodel.FanSummaryData
|
||
err = parser.Parse(result.Cells, &summary)
|
||
if err != nil {
|
||
log.Error("convert from hbase fail, err=%v", err)
|
||
return
|
||
}
|
||
|
||
/*
|
||
for _, c := range result.Cells {
|
||
if c == nil {
|
||
continue
|
||
}
|
||
if string(c.Family) == "f" {
|
||
|
||
k := string(c.Qualifier[:])
|
||
var v int32
|
||
if len(c.Value) == 4 {
|
||
v = byteToInt32(c.Value)
|
||
switch k {
|
||
case "all": //总粉丝
|
||
summary.Total = v
|
||
//summary["total"] = v
|
||
case "inc": //新增粉丝
|
||
summary.Inc = v
|
||
case "act": //活跃粉丝
|
||
summary.Active = v
|
||
// summary["active"] = v
|
||
//case "mdl": //领取勋章粉丝
|
||
// summary["medal"] = v
|
||
//case "elec": //充电粉丝
|
||
// summary["elec"] = v
|
||
//case "act_diff": //活跃粉丝（增量）
|
||
// summary["act_diff"] = v
|
||
//case "mdl_diff": //领取勋章粉丝（增量）
|
||
// summary["mdl_diff"] = v
|
||
//case "elec_diff": //充电粉丝（增量）
|
||
// summary["elec_diff"] = v
|
||
//case "v": //播放粉丝占比*10000
|
||
// summary["v"] = v
|
||
//case "dm": //弹幕粉丝占比*10000
|
||
// summary["dm"] = v
|
||
//case "r": //评论粉丝占比*10000
|
||
// summary["r"] = v
|
||
//case "c": //投币粉丝占比*10000
|
||
// summary["c"] = v
|
||
// //新增粉丝活跃占比
|
||
//case "inter": //互动活跃度*10000
|
||
// summary["inter"] = v
|
||
//case "vv": //观看活跃度*10000
|
||
// summary["vv"] = v
|
||
//case "da": //弹幕粉丝占比*10000
|
||
// summary["da"] = v
|
||
//case "re": //评论粉丝占比*10000
|
||
// summary["re"] = v
|
||
//case "co": //投币粉丝占比*10000
|
||
// summary["co"] = v
|
||
//case "fv": //收藏粉丝占比*10000
|
||
// summary["fv"] = v
|
||
//case "sh": //分享粉丝占比*10000
|
||
// summary["sh"] = v
|
||
//case "lk": //点赞粉丝占比*10000
|
||
// summary["lk"] = v
|
||
}
|
||
} else {
|
||
log.Error("UpFansAnalysisForWeb family[f] get dirty tableName(%s)|mid(%d)|rowkey(%s)|value(%+v)", tableName, mid, rowkey, c.Value)
|
||
}
|
||
}
|
||
if string(c.Family) == "t" {
|
||
var v int32
|
||
if len(c.Value) == 4 {
|
||
v = int32(binary.BigEndian.Uint32(c.Value))
|
||
} else {
|
||
log.Error("UpFansAnalysisForWeb family t get dirty tableName(%s)|mid(%d)|rowkey(%s)|value(%+v)", tableName, mid, fansRowKey(mid, ty), c.Value)
|
||
}
|
||
if strings.Contains(string(c.Qualifier[:]), "dr") {
|
||
rankDr[string(c.Qualifier[:])] = v
|
||
}
|
||
if strings.Contains(string(c.Qualifier[:]), "act") {
|
||
rankVideoAct[string(c.Qualifier[:])] = v
|
||
}
|
||
if strings.Contains(string(c.Qualifier[:]), "dy") {
|
||
rankDyAct[string(c.Qualifier[:])] = v
|
||
}
|
||
}
|
||
if string(c.Family) == "s" {
|
||
var v int32
|
||
if len(c.Value) == 4 {
|
||
v = int32(binary.BigEndian.Uint32(c.Value))
|
||
} else {
|
||
log.Error("UpFansAnalysisForWeb family[t] get dirty data tableName(%s)|mid(%d)|rowkey(%s)|value(%+v)", tableName, mid, rowkey, c.Value)
|
||
}
|
||
if strings.Contains(string(c.Qualifier[:]), "pf") {
|
||
if pk, ok := fanSource[string(c.Qualifier[:])]; ok {
|
||
source[pk] = v
|
||
}
|
||
}
|
||
}
|
||
}
|
||
for _, k := range fanSource { //粉丝来源页面占比如果数据平台未返回,则设置对应的key
|
||
if _, ok := source[k]; !ok {
|
||
source[k] = 0
|
||
}
|
||
}
|
||
*/
|
||
log.Info("UpFansAnalysisForWebRankMap mid(%d)|rowkey(%s)|summary(%+v)|rankMap(%+v)|source(%+v)", mid, rowkey, summary, rankMap, source)
|
||
res.Summary = summary
|
||
return
|
||
}
|
||
|
||
// UpArcQuery for play aids by mid.
|
||
func (d *Dao) UpArcQuery(c context.Context, mid int64, dt string, cp int) (res []int64, err error) {
|
||
var (
|
||
result *hrpc.Result
|
||
ctx, cancel = context.WithTimeout(c, time.Duration(d.c.HBase.ReadTimeout))
|
||
tableName = HBaseUpArcQuery
|
||
rowkey = arcQueryKey(mid, dt, cp)
|
||
)
|
||
defer cancel()
|
||
log.Info("UpArcQuery mid(%d)|rowkey(%s)", mid, rowkey)
|
||
if result, err = d.hbase.GetStr(ctx, tableName, rowkey); err != nil {
|
||
log.Error("d.hbase.GetStr tableName(%s)|mid(%d)|rowkey(%s)|error(%v)", tableName, mid, rowkey, err)
|
||
err = ecode.CreativeDataErr
|
||
return
|
||
}
|
||
if result == nil || len(result.Cells) == 0 {
|
||
log.Warn("UpArcQuery no data tableName(%s)|mid(%d)|rowkey(%s)", tableName, mid, rowkey)
|
||
return
|
||
}
|
||
res = make([]int64, 0)
|
||
for _, c := range result.Cells {
|
||
if c == nil {
|
||
continue
|
||
}
|
||
if string(c.Family) == "av" {
|
||
var v int64
|
||
if len(c.Value) == 4 {
|
||
v = int64(binary.BigEndian.Uint32(c.Value))
|
||
} else {
|
||
log.Error("UpArcQuery family[av] get dirty value tableName(%s)|rowkey(%s)", tableName, rowkey)
|
||
}
|
||
res = append(res, v)
|
||
}
|
||
}
|
||
log.Info("UpArcQuery mid(%d)|rowkey(%s)|res(%+v)", mid, rowkey, res)
|
||
return
|
||
}
|