Skip to content

Commit

Permalink
feat: add message counter
Browse files Browse the repository at this point in the history
  • Loading branch information
NSEcho committed Dec 3, 2023
1 parent e6fe316 commit 75b7231
Showing 1 changed file with 16 additions and 11 deletions.
27 changes: 16 additions & 11 deletions object.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,21 @@ import (
"regexp"
"strconv"
"strings"
"sync"
)

type ctr struct {
c int
l *sync.Mutex
}

var c = &ctr{l: &sync.Mutex{}}

func PrintData(value any, decode, printHex bool,
whitelist, blacklist, whitelistp, blacklistp []*regexp.Regexp,
logger *Logger) {
msg := 0

val := reflect.ValueOf(value)

data := make(map[string]any)
Expand Down Expand Up @@ -41,18 +51,13 @@ func PrintData(value any, decode, printHex bool,
}
}

if len(whitelist) > 0 {
if !connInList(name, whitelist) {
return
}
} else {
if connInList(name, blacklist) {
return
}
}
c.l.Lock()
msg = c.c
c.c++
c.l.Unlock()

var message string
fnName := fmt.Sprintf("Name: %s\n", data["name"])
fnName := fmt.Sprintf("%d) Name: %s\n", msg, data["name"])
connName := fmt.Sprintf("Connection Name: %s\n", data["connName"])
printData(reflect.ValueOf(data["dictionary"]), "", "", &message)
total := len(fnName) + len(connName) + len(message) + 100
Expand All @@ -66,7 +71,7 @@ func PrintData(value any, decode, printHex bool,
builder.WriteString(message)
builder.WriteString(fmt.Sprintf("\n%s\n", strings.Repeat("=", 80)))

logger.Scriptf("Name: %s", data["name"])
logger.Scriptf("%d) Name: %s", msg, data["name"])
logger.Scriptf("Connection Name: %s", data["connName"])
pid, ok := data["pid"].(float64)
if ok {
Expand Down

0 comments on commit 75b7231

Please sign in to comment.