阅读排行榜：完成数据计算和存储

zhima1234 · zhima1234 · commit 1112d9a796af · 2017-05-25T20:20:53.000+08:00
diff --git a/config/env.sample.ini b/config/env.sample.ini
@@ -27,6 +27,20 @@ show_sql = true
 ; 0-debug, 1-info, 2-warning, 3-error, 4-off, 5-unknow
 log_level = 0
 
+[redis]
+host = 127.0.0.1
+port = 6379
+password = 
+; 连接超时
+conn_timeout = 2
+read_timeout = 2
+write_timeout = 2
+
+; 最大空闲连接数
+max_idle = 2
+
+prefix = studygolang:
+
 ; 用于注册发送激活码等
 [email]
 smtp_username = xxxx@studygolang.com
diff --git a/src/logic/rank.go b/src/logic/rank.go
@@ -0,0 +1,80 @@
+// Copyright 2017 The StudyGolang Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author:polaris	polaris@studygolang.com
+
+package logic
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/polaris1119/logger"
+	"github.com/polaris1119/nosql"
+	"github.com/polaris1119/times"
+)
+
+type RankLogic struct{}
+
+var DefaultRank = RankLogic{}
+
+func (self RankLogic) GenDayRank(objtype, objid, num int) {
+	redisClient := nosql.NewRedisClient()
+	key := self.getDayRankKey(objtype, times.Format("ymd"))
+	err := redisClient.ZINCRBY(key, num, objid)
+	if err != nil {
+		logger.Errorln("view redis ZINCRBY error:", err)
+	}
+	redisClient.EXPIRE(key, 2*30*86400)
+}
+
+// GenWeekRank 过去 7 天排行榜
+func (self RankLogic) GenWeekRank(objtype int) {
+	redisClient := nosql.NewRedisClient()
+	dest := self.getWeekRankKey(objtype)
+
+	keys := self.getMultiKey(objtype, 7)
+
+	err := redisClient.ZUNIONSTORE(dest, 7, keys, nil)
+	if err != nil {
+		logger.Errorln("GenWeekRank ZUNIONSTORE error:", err)
+	}
+}
+
+// GenMonthRank 过去 30 天排行榜
+func (self RankLogic) GenMonthRank(objtype int) {
+	redisClient := nosql.NewRedisClient()
+	dest := self.getMonthRankKey(objtype)
+
+	keys := self.getMultiKey(objtype, 30)
+
+	err := redisClient.ZUNIONSTORE(dest, 30, keys, nil)
+	if err != nil {
+		logger.Errorln("GenMonthRank ZUNIONSTORE error:", err)
+	}
+}
+
+func (self RankLogic) getMultiKey(objtype, num int) []string {
+	today := time.Now()
+
+	keys := make([]string, num)
+	for i := 0; i < num; i++ {
+		ymd := times.Format("ymd", today.Add(time.Duration(-(i+1)*86400)*time.Second))
+		keys[i] = self.getDayRankKey(objtype, ymd)
+	}
+
+	return keys
+}
+
+func (RankLogic) getDayRankKey(objtype int, ymd string) string {
+	return fmt.Sprintf("view:type-%d:rank:%s", objtype, ymd)
+}
+
+func (RankLogic) getWeekRankKey(objtype int) string {
+	return fmt.Sprintf("view:type-%d:rank:last-week", objtype)
+}
+
+func (RankLogic) getMonthRankKey(objtype int) string {
+	return fmt.Sprintf("view:type-%d:rank:last-month", objtype)
+}
diff --git a/src/logic/view.go b/src/logic/view.go
@@ -18,7 +18,7 @@ import (
 	"github.com/polaris1119/logger"
 )
 
-// 话题/文章/资源的浏览数
+// 话题/文章/资源/图书等的浏览数
 // 避免每次写库，同时避免刷屏
 type view struct {
 	objtype int // 对象类型（model/comment 中的 type 常量）
@@ -60,6 +60,8 @@ func (this *view) flush() {
 		session.Incr("viewnum", this.num).Update(new(model.Book))
 	}
 
+	DefaultRank.GenDayRank(this.objtype, this.objid, this.num)
+
 	this.num = 0
 }
 
diff --git a/src/server/studygolang/background.go b/src/server/studygolang/background.go
@@ -10,6 +10,7 @@ import (
 	"db"
 	"global"
 	"logic"
+	"model"
 	"time"
 
 	"github.com/polaris1119/logger"
@@ -34,6 +35,9 @@ func ServeBackGround() {
 	// 每天对非活跃用户降频
 	c.AddFunc("@daily", decrUserActiveWeight)
 
+	// 生成阅读排行榜
+	c.AddFunc("@daily", genViewRank)
+
 	// 两分钟刷一次浏览数（TODO：重启丢失问题？信号控制重启？）
 	c.AddFunc("@every 2m", logic.Views.Flush)
 
@@ -101,3 +105,18 @@ func decrUserActiveWeight() {
 
 	logger.Debugln("end decr user active weight...")
 }
+
+func genViewRank() {
+	needRankTypes := []int{
+		model.TypeTopic,
+		model.TypeResource,
+		model.TypeProject,
+		model.TypeArticle,
+		model.TypeBook,
+	}
+
+	for _, objtype := range needRankTypes {
+		logic.DefaultRank.GenWeekRank(objtype)
+		logic.DefaultRank.GenMonthRank(objtype)
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ import (`
`18`	`18`	`"github.com/polaris1119/logger"`
`19`	`19`	`)`
`20`	`20`
`21`		`-// 话题/文章/资源的浏览数`
	`21`	`+// 话题/文章/资源/图书等的浏览数`
`22`	`22`	`// 避免每次写库，同时避免刷屏`
`23`	`23`	`type view struct {`
`24`	`24`	`objtype int // 对象类型（model/comment 中的 type 常量）`
`@@ -60,6 +60,8 @@ func (this *view) flush() {`
`60`	`60`	`session.Incr("viewnum", this.num).Update(new(model.Book))`
`61`	`61`	`}`
`62`	`62`
	`63`	`+ DefaultRank.GenDayRank(this.objtype, this.objid, this.num)`
	`64`	`+`
`63`	`65`	`this.num = 0`
`64`	`66`	`}`
`65`	`67`