对标题更严格的敏感词过滤

polaris1119 · polaris1119 · commit f1cb0aab0dbc · 2015-11-04T22:34:47.000+08:00
diff --git a/websites/code/studygolang/src/filter/sensitive.go b/websites/code/studygolang/src/filter/sensitive.go
@@ -23,6 +23,8 @@ type SensitiveFilter struct {
 	*mux.EmptyFilter
 }
 
+var titleSensitives = []string{"发票"}
+
 // PreFilter 执行 handler 之前的过滤方法
 func (this *SensitiveFilter) PreFilter(rw http.ResponseWriter, req *http.Request) bool {
 	logger.Debugln("SensitiveFilter PreFilter...")
@@ -33,10 +35,22 @@ func (this *SensitiveFilter) PreFilter(rw http.ResponseWriter, req *http.Request
 		return true
 	}
 
+	curUser, _ := CurrentUser(req)
+
+	// 标题特殊处理
+	for _, s := range titleSensitives {
+		if util.HasSensitiveChar(title, s) {
+			// 把账号冻结
+			service.UpdateUserStatus(curUser["uid"].(int), model.StatusFreeze)
+
+			logger.Infoln("user=", curUser["uid"], "publish ad, title=", title, ";content=", content, ". freeze")
+			return false
+		}
+	}
+
 	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")
diff --git a/websites/code/studygolang/src/util/sensitive.go b/websites/code/studygolang/src/util/sensitive.go
@@ -11,7 +11,11 @@ import (
 )
 
 // HasSensitive 是否有敏感词
-func HasSensitive(content string, sensitive string) bool {
+func HasSensitive(content, sensitive string) bool {
+	if content == "" {
+		return false
+	}
+
 	sensitives := strings.Split(sensitive, ",")
 
 	for _, s := range sensitives {
@@ -22,3 +26,20 @@ func HasSensitive(content string, sensitive string) bool {
 
 	return false
 }
+
+// HasSensitiveChar 是否包含敏感字（多个词都包含）
+func HasSensitiveChar(title, sensitive string) bool {
+	if title == "" || sensitive == "" {
+		return false
+	}
+
+	sensitives := strings.Split(sensitive, "")
+
+	for _, s := range sensitives {
+		if !strings.Contains(title, s) {
+			return false
+		}
+	}
+
+	return true
+}
