Skip to content
This repository was archived by the owner on Feb 1, 2023. It is now read-only.

Commit 8c7bf92

Browse files
committed
feat: normalize message logging
1 parent 484399b commit 8c7bf92

File tree

2 files changed

+39
-14
lines changed

2 files changed

+39
-14
lines changed

internal/messagequeue/messagequeue.go

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -466,23 +466,43 @@ func (mq *MessageQueue) simulateDontHaveWithTimeout(wantlist []bsmsg.Entry) {
466466

467467
func (mq *MessageQueue) logOutgoingMessage(wantlist []bsmsg.Entry) {
468468
// Save some CPU cycles and allocations if log level is higher than debug
469-
if ce := sflog.Check(zap.DebugLevel, "Bitswap -> send wants"); ce == nil {
469+
if ce := sflog.Check(zap.DebugLevel, "sent message"); ce == nil {
470470
return
471471
}
472472

473473
self := mq.network.Self()
474474
for _, e := range wantlist {
475475
if e.Cancel {
476476
if e.WantType == pb.Message_Wantlist_Have {
477-
log.Debugw("Bitswap -> cancel-have", "local", self, "to", mq.p, "cid", e.Cid)
477+
log.Debugw("sent message",
478+
"type", "CANCEL_WANT_HAVE",
479+
"cid", e.Cid,
480+
"local", self,
481+
"to", mq.p,
482+
)
478483
} else {
479-
log.Debugw("Bitswap -> cancel-block", "local", self, "to", mq.p, "cid", e.Cid)
484+
log.Debugw("sent message",
485+
"type", "CANCEL_WANT_BLOCK",
486+
"cid", e.Cid,
487+
"local", self,
488+
"to", mq.p,
489+
)
480490
}
481491
} else {
482492
if e.WantType == pb.Message_Wantlist_Have {
483-
log.Debugw("Bitswap -> want-have", "local", self, "to", mq.p, "cid", e.Cid)
493+
log.Debugw("sent message",
494+
"type", "WANT_HAVE",
495+
"cid", e.Cid,
496+
"local", self,
497+
"to", mq.p,
498+
)
484499
} else {
485-
log.Debugw("Bitswap -> want-block", "local", self, "to", mq.p, "cid", e.Cid)
500+
log.Debugw("sent message",
501+
"type", "WANT_BLOCK",
502+
"cid", e.Cid,
503+
"local", self,
504+
"to", mq.p,
505+
)
486506
}
487507
}
488508
}

workers.go

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -67,35 +67,40 @@ func (bs *Bitswap) taskWorker(ctx context.Context, id int) {
6767
}
6868

6969
func (bs *Bitswap) logOutgoingBlocks(env *engine.Envelope) {
70-
if ce := sflog.Check(zap.DebugLevel, "Bitswap -> send blocks"); ce == nil {
70+
if ce := sflog.Check(zap.DebugLevel, "sent message"); ce == nil {
7171
return
7272
}
7373

74+
self := bs.network.Self()
75+
7476
for _, blockPresence := range env.Message.BlockPresences() {
7577
c := blockPresence.Cid
7678
switch blockPresence.Type {
7779
case pb.Message_Have:
78-
log.Debugw("sending message",
80+
log.Debugw("sent message",
7981
"type", "HAVE",
8082
"cid", c,
81-
"peer", env.Peer,
83+
"local", self,
84+
"to", env.Peer,
8285
)
8386
case pb.Message_DontHave:
84-
log.Debugw("sending message",
87+
log.Debugw("sent message",
8588
"type", "DONT_HAVE",
8689
"cid", c,
87-
"peer", env.Peer,
90+
"local", self,
91+
"to", env.Peer,
8892
)
8993
default:
9094
panic(fmt.Sprintf("unrecognized BlockPresence type %v", blockPresence.Type))
9195
}
9296

9397
}
9498
for _, block := range env.Message.Blocks() {
95-
log.Debugw("sending message",
99+
log.Debugw("sent message",
96100
"type", "BLOCK",
97101
"cid", block.Cid(),
98-
"peer", env.Peer,
102+
"local", self,
103+
"to", env.Peer,
99104
)
100105
}
101106
}
@@ -105,8 +110,6 @@ func (bs *Bitswap) sendBlocks(ctx context.Context, env *engine.Envelope) {
105110
// throughout the network stack
106111
defer env.Sent()
107112

108-
bs.logOutgoingBlocks(env)
109-
110113
err := bs.network.SendMessage(ctx, env.Peer, env.Message)
111114
if err != nil {
112115
log.Debugw("failed to send blocks message",
@@ -116,6 +119,8 @@ func (bs *Bitswap) sendBlocks(ctx context.Context, env *engine.Envelope) {
116119
return
117120
}
118121

122+
bs.logOutgoingBlocks(env)
123+
119124
dataSent := 0
120125
blocks := env.Message.Blocks()
121126
for _, b := range blocks {

0 commit comments

Comments
 (0)