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.
46 lines
811 B
46 lines
811 B
3 years ago
|
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),
|
||
|
|
}))
|
||
|
|
}
|