过滤敏感词，发广告直接封号

polaris1119 · polaris1119 · commit b5d106f0d754 · 2015-11-04T19:10:47.000+08:00
diff --git a/websites/code/studygolang/src/filter/sensitive.go b/websites/code/studygolang/src/filter/sensitive.go
@@ -0,0 +1,48 @@
+// Copyright 2015 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 filter
+
+import (
+	"model"
+	"net/http"
+	"service"
+
+	"config"
+	"logger"
+	"util"
+
+	"github.com/studygolang/mux"
+)
+
+// SensitiveFilter 敏感词过滤器
+type SensitiveFilter struct {
+	*mux.EmptyFilter
+}
+
+// PreFilter 执行 handler 之前的过滤方法
+func (this *SensitiveFilter) PreFilter(rw http.ResponseWriter, req *http.Request) bool {
+	logger.Debugln("SensitiveFilter PreFilter...")
+
+	content := req.FormValue("content")
+	title := req.FormValue("title")
+	if title == "" && content == "" {
+		return true
+	}
+
+	sensitive := config.Config["sensitive"]
+	if util.HasSensitive(title, sensitive) || util.HasSensitive(content, sensitive) {
+		// 把账号冻结
+		curUser, _ := CurrentUser(req)
+		service.UpdateUserStatus(curUser["uid"].(int), model.StatusFreeze)
+
+		logger.Infoln("user=", curUser["uid"], "publish ad, title=", title, ";content=", content, ". freeze")
+
+		return false
+	}
+
+	return true
+}
diff --git a/websites/code/studygolang/src/server/studygolang/router.go b/websites/code/studygolang/src/server/studygolang/router.go
@@ -12,6 +12,7 @@ import (
 	. "controller"
 	"controller/admin"
 	"filter"
+
 	"github.com/studygolang/mux"
 )
 
@@ -20,6 +21,10 @@ func initRouter() *mux.Router {
 	loginFilter := new(filter.LoginFilter)
 	loginFilterChain := mux.NewFilterChain(loginFilter)
 
+	// 敏感词过滤器
+	sensitiveFilter := new(filter.SensitiveFilter)
+	sensitiveFilterChain := mux.NewFilterChain(sensitiveFilter)
+
 	router := mux.NewRouter()
 	// 所有的页面都需要先检查用户cookie是否存在，以便在没登录时自动登录
 	cookieFilter := new(filter.CookieFilter)
@@ -37,8 +42,8 @@ func initRouter() *mux.Router {
 
 	router.HandleFunc("/topics{view:(|/popular|/no_reply|/last)}", TopicsHandler)
 	router.HandleFunc("/topics/{tid:[0-9]+}", TopicDetailHandler)
-	router.HandleFunc("/topics/new{json:(|.json)}", NewTopicHandler).AppendFilterChain(loginFilterChain)
-	router.HandleFunc("/topics/modify{json:(|.json)}", ModifyTopicHandler).AppendFilterChain(loginFilterChain)
+	router.HandleFunc("/topics/new{json:(|.json)}", NewTopicHandler).AppendFilterChain(loginFilterChain).AppendFilterChain(sensitiveFilterChain)
+	router.HandleFunc("/topics/modify{json:(|.json)}", ModifyTopicHandler).AppendFilterChain(loginFilterChain).AppendFilterChain(sensitiveFilterChain)
 	// 某个节点下的话题
 	router.HandleFunc("/topics/node{nid:[0-9]+}", NodesHandler)
 
@@ -93,8 +98,8 @@ func initRouter() *mux.Router {
 	router.HandleFunc("/resources", ResIndexHandler)
 	router.HandleFunc("/resources/cat/{catid:[0-9]+}", CatResourcesHandler)
 	router.HandleFunc("/resources/{id:[0-9]+}", ResourceDetailHandler)
-	router.HandleFunc("/resources/new{json:(|.json)}", NewResourceHandler).AppendFilterChain(loginFilterChain)
-	router.HandleFunc("/resources/modify{json:(|.json)}", ModifyResourceHandler).AppendFilterChain(loginFilterChain)
+	router.HandleFunc("/resources/new{json:(|.json)}", NewResourceHandler).AppendFilterChain(loginFilterChain).AppendFilterChain(sensitiveFilterChain)
+	router.HandleFunc("/resources/modify{json:(|.json)}", ModifyResourceHandler).AppendFilterChain(loginFilterChain).AppendFilterChain(sensitiveFilterChain)
 
 	// 评论
 	router.HandleFunc("/comment/{objid:[0-9]+}.json", CommentHandler).AppendFilterChain(loginFilterChain)
diff --git a/websites/code/studygolang/src/service/user.go b/websites/code/studygolang/src/service/user.go
@@ -115,6 +115,18 @@ func UpdateUser(form url.Values) (errMsg string, err error) {
 	return
 }
 
+// UpdateUserStatus 更新用户状态
+func UpdateUserStatus(uid, status int) {
+	setClause := "status=" + strconv.Itoa(status)
+	err := model.NewUser().Set(setClause).Where("uid=?", uid).Update()
+	if err != nil {
+		logger.Errorf("更新用户 【%s】 状态失败：%s", uid, err)
+		return
+	}
+
+	return
+}
+
 // 邮件订阅或取消订阅
 func EmailSubscribe(uid, unsubscribe int) {
 	err := model.NewUser().Set("unsubscribe=?", unsubscribe).Where("uid=?", uid).Update()
diff --git a/websites/code/studygolang/src/util/sensitive.go b/websites/code/studygolang/src/util/sensitive.go
@@ -0,0 +1,24 @@
+// Copyright 2015 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 util
+
+import (
+	"strings"
+)
+
+// HasSensitive 是否有敏感词
+func HasSensitive(content string, sensitive string) bool {
+	sensitives := strings.Split(sensitive, ",")
+
+	for _, s := range sensitives {
+		if strings.Contains(content, s) {
+			return true
+		}
+	}
+
+	return false
+}
