根据用户访问网站的情况动态更新活跃用户数据

polaris1119 · polaris1119 · commit 0349ccf1d6a7 · 2014-01-18T20:46:04.000+08:00
diff --git a/websites/code/studygolang/conf/config.json.example b/websites/code/studygolang/conf/config.json.example
@@ -1,15 +1,15 @@
 {
-    "domain": "dev.studygolang.com",
-    "host": "127.0.0.1:8080",
-    "wshost": "127.0.0.1:6090",
-    "drive_name": "mysql",
-    "dsn": "root:@tcp(localhost:3306)/studygolang?charset=utf8",
-    "cookie_secret": "2fwefefwf",
-    "smtp_username": "example_username",
+	"domain": "dev.studygolang.com",
+	"host": "127.0.0.1:8080",
+	"wshost": "127.0.0.1:6090",
+	"drive_name": "mysql",
+	"dsn": "root:@tcp(localhost:3306)/studygolang?charset=utf8",
+	"cookie_secret": "2fwefefwf",
+	"smtp_username": "example_username",
 	"smtp_password": "example_password",
 	"smtp_host": "smtp.example.com",
 	"smtp_addr": "smtp.example.com:25",
 	"from_email": "example@example.com",
-    "data": "data/max_online_num",
-    "pid": "pid/studygolang.pid"
+	"data": "data/max_online_num",
+	"pid": "pid/studygolang.pid"
 }
diff --git a/websites/code/studygolang/src/model/dao.go b/websites/code/studygolang/src/model/dao.go
@@ -192,6 +192,7 @@ func (this *Dao) FindAll(selectCol ...string) (*sql.Rows, error) {
 	this.selectCols = "`" + strings.Join(selectCol, "`,`") + "`"
 	strSql := util.SelectSql(this)
 	logger.Debugln("FindAll sql:", strSql)
+	logger.Debugln("FindAll bind params:", this.whereVal)
 	err := this.Open()
 	if err != nil {
 		return nil, err
@@ -219,6 +220,16 @@ func (this *Dao) Scan(rows *sql.Rows, colNum int, colFieldMap map[string]interfa
 	return rows.Scan(scanInterface...)
 }
 
+// 执行 insert、update、delete 等操作
+func (this *Dao) Exec(strSql string, args ...interface{}) (sql.Result, error) {
+	err := this.Open()
+	if err != nil {
+		return nil, err
+	}
+	defer this.Close()
+	return this.DB.Exec(strSql, args...)
+}
+
 /*
 // FindAllEx 查找多条数据
 func (this *Dao) FindAllEx(selectCol ...string) (reflect.Value, error) {
diff --git a/websites/code/studygolang/src/model/user.go b/websites/code/studygolang/src/model/user.go
@@ -45,6 +45,30 @@ func (this *UserLogin) Find(selectCol ...string) error {
 	return this.Dao.Find(this.colFieldMap(), selectCol...)
 }
 
+func (this *UserLogin) FindAll(selectCol ...string) ([]*UserLogin, error) {
+	if len(selectCol) == 0 {
+		selectCol = util.MapKeys(this.colFieldMap())
+	}
+	rows, err := this.Dao.FindAll(selectCol...)
+	if err != nil {
+		return nil, err
+	}
+	// TODO:
+	userList := make([]*UserLogin, 0, 10)
+	logger.Debugln("selectCol", selectCol)
+	colNum := len(selectCol)
+	for rows.Next() {
+		user := NewUserLogin()
+		err = this.Scan(rows, colNum, user.colFieldMap(), selectCol...)
+		if err != nil {
+			logger.Errorln("UserLogin FindAll Scan Error:", err)
+			continue
+		}
+		userList = append(userList, user)
+	}
+	return userList, nil
+}
+
 // 为了支持连写
 func (this *UserLogin) Where(condition string) *UserLogin {
 	this.Dao.Where(condition)
diff --git a/websites/code/studygolang/src/service/user.go b/websites/code/studygolang/src/service/user.go
@@ -351,3 +351,24 @@ func IncUserWeight(where string, weight int) {
 		logger.Errorln("UserActive update Error:", err)
 	}
 }
+
+func DecrUserWeight(where string, divide int) {
+	if divide <= 0 {
+		return
+	}
+
+	strSql := "UPDATE user_active SET weight=weight/" + strconv.Itoa(divide) + " WHERE " + where
+	if result, err := model.NewUserActive().Exec(strSql); err != nil {
+		logger.Errorln("UserActive update Error:", err)
+	} else {
+		n, _ := result.RowsAffected()
+		logger.Debugln(strSql, "affected num:", n)
+	}
+}
+
+// 获取 loginTime 之前没有登录的用户
+func FindNotLoginUsers(loginTime string) (userList []*model.UserLogin, err error) {
+	userLogin := model.NewUserLogin()
+	userList, err = userLogin.Where("login_time<" + loginTime).FindAll()
+	return
+}
diff --git a/websites/code/studygolang/src/studygolang/background.go b/websites/code/studygolang/src/studygolang/background.go
@@ -0,0 +1,56 @@
+// Copyright 2013 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	studygolang@gmail.com
+
+package main
+
+import (
+	"logger"
+	"service"
+	"time"
+	"util"
+)
+
+// 后台运行的任务
+func ServeBackGround() {
+	for {
+		go decrUserActiveWeight()
+		time.Sleep(24 * time.Hour)
+	}
+}
+
+func decrUserActiveWeight() {
+	logger.Debugln("start decr user active weight...")
+
+	loginTime := time.Now().Add(-72 * time.Hour)
+	userList, err := service.FindNotLoginUsers(loginTime.Format(util.TIME_LAYOUT_OFTEN))
+	if err != nil {
+		logger.Errorln("获取最近未登录用户失败：", err)
+		return
+	}
+
+	logger.Debugln("need dealing users:", len(userList))
+
+	for _, user := range userList {
+		divide := 5
+
+		lastLoginTime, err := util.TimeParseOften(user.LoginTime)
+		if err == nil {
+			hours := (loginTime.Sub(lastLoginTime) / 24).Hours()
+			if hours < 24 {
+				divide = 2
+			} else if hours < 48 {
+				divide = 3
+			} else if hours < 72 {
+				divide = 4
+			}
+		}
+
+		logger.Debugln("decr user weight, username:", user.Username, "divide:", divide)
+		service.DecrUserWeight("username='"+user.Username+"'", divide)
+	}
+
+	logger.Debugln("end decr user active weight...")
+}
diff --git a/websites/code/studygolang/src/studygolang/main.go b/websites/code/studygolang/src/studygolang/main.go
@@ -32,6 +32,8 @@ func main() {
 
 	go ServeWebSocket()
 
+	go ServeBackGround()
+
 	router := initRouter()
 	http.Handle("/", router)
 	log.Fatal(http.ListenAndServe(Config["host"], nil))
diff --git a/websites/code/studygolang/src/util/time.go b/websites/code/studygolang/src/util/time.go
@@ -0,0 +1,19 @@
+// Copyright 2013 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	studygolang@gmail.com
+
+package util
+
+import (
+	"time"
+)
+
+const TIME_LAYOUT_OFTEN = "2006-01-02 15:04:05"
+
+// 解析常用的日期时间格式：2014-01-11 16:18:00，东八区
+func TimeParseOften(value string) (time.Time, error) {
+	local, _ := time.LoadLocation("Local")
+	return time.ParseInLocation(TIME_LAYOUT_OFTEN, value, local)
+}
