You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
811 B
45 lines
811 B
package mtproto |
|
|
|
import ( |
|
"fmt" |
|
|
|
"go.uber.org/zap" |
|
"go.uber.org/zap/zapcore" |
|
|
|
"github.com/gotd/td/bin" |
|
) |
|
|
|
type logType struct { |
|
ID uint32 |
|
Name string |
|
} |
|
|
|
func (l logType) MarshalLogObject(e zapcore.ObjectEncoder) error { |
|
typeIDStr := fmt.Sprintf("0x%x", l.ID) |
|
e.AddString("type_id", typeIDStr) |
|
if l.Name != "" { |
|
e.AddString("type_name", l.Name) |
|
} |
|
return nil |
|
} |
|
|
|
func (c *Conn) logWithBuffer(b *bin.Buffer) *zap.Logger { |
|
return c.logWithType(b).With(zap.Int("size_bytes", b.Len())) |
|
} |
|
|
|
func (c *Conn) logWithType(b *bin.Buffer) *zap.Logger { |
|
id, err := b.PeekID() |
|
if err != nil { |
|
// Type info not available. |
|
return c.log |
|
} |
|
|
|
return c.logWithTypeID(id) |
|
} |
|
|
|
func (c *Conn) logWithTypeID(id uint32) *zap.Logger { |
|
return c.log.With(zap.Inline(logType{ |
|
ID: id, |
|
Name: c.types.Get(id), |
|
})) |
|
}
|
|
|