BlockMessageLog现在支持正则表达式匹配

meta.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"path"
 	"reflect"
+	"regexp"
 )
 
 // 消息元信息
@@ -107,6 +108,25 @@ func MessageMetaByFullName(name string) *MessageMeta {
 	return nil
 }
 
+func MessageMetaVisit(nameRule string, callback func(meta *MessageMeta) bool) error {
+	exp, err := regexp.Compile(nameRule)
+	if err != nil {
+		return err
+	}
+
+	for name, meta := range metaByFullName {
+		if exp.MatchString(name) {
+
+			if !callback(meta) {
+				return nil
+			}
+
+		}
+	}
+
+	return nil
+}
+
 // 根据类型查找消息元信息
 func MessageMetaByType(t reflect.Type) *MessageMeta {
 
@@ -171,6 +191,10 @@ func MessageToID(msg interface{}) int {
 
 func MessageSize(msg interface{}) int {
 
+	if msg == nil {
+		return 0
+	}
+
 	// 获取消息元信息
 	meta := MessageMetaByType(reflect.TypeOf(msg))
 	if meta == nil {

msglog/blocker.go

@@ -1,7 +1,6 @@
 package msglog
 
 import (
-	"errors"
 	"github.com/davyxu/cellnet"
 	"sync"
 )
@@ -17,18 +16,15 @@ func IsBlockedMessageByID(msgid int) bool {
 	return ok
 }
 
-var (
-	ErrMessageNotFound = errors.New("msg not exists")
-)
+func BlockMessageLog(nameRule string) (err error, matchCount int) {
 
-func BlockMessageLog(msgName string) error {
-	meta := cellnet.MessageMetaByFullName(msgName)
+	err = cellnet.MessageMetaVisit(nameRule, func(meta *cellnet.MessageMeta) bool {
 
-	if meta == nil {
-		return ErrMessageNotFound
-	}
+		blockedMsgByID.Store(int(meta.ID), meta)
+		matchCount++
 
-	blockedMsgByID.Store(int(meta.ID), meta)
+		return true
+	})
 
-	return nil
+	return
 }