package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-10-25 07:50:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Oarkflow
2025-09-03 19:08:04 +05:45
parent 2b5a50a4f4
commit 53584d0ddc
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
codec/codec.go
View File

@@ -184,7 +184,7 @@ func (m *Message) Validate(config *Config) error {
} }
if len(m.Payload) > int(config.MaxMessageSize) { if len(m.Payload) > int(config.MaxMessageSize) {
return ErrMessageTooLarge return fmt.Errorf("%v %v, payload size: %d", ErrMessageTooLarge, config.MaxMessageSize, len(m.Payload))
} }
return nil return nil
@@ -264,7 +264,7 @@ func (c *Codec) sendRawMessage(ctx context.Context, conn net.Conn, msg *Message)
// Check message size // Check message size
if len(data) > int(c.config.MaxMessageSize) { if len(data) > int(c.config.MaxMessageSize) {
c.incrementErrors() c.incrementErrors()
return ErrMessageTooLarge return fmt.Errorf("%v %v, payload: %d", ErrMessageTooLarge, c.config.MaxMessageSize, len(data))
} }
// Prepare buffer // Prepare buffer
@@ -337,7 +337,7 @@ func (c *Codec) ReadMessage(ctx context.Context, conn net.Conn) (*Message, error
// Validate message size // Validate message size
if length > c.config.MaxMessageSize { if length > c.config.MaxMessageSize {
c.incrementErrors() c.incrementErrors()
return nil, ErrMessageTooLarge return nil, fmt.Errorf("%v %v, payload size: %d", ErrMessageTooLarge, c.config.MaxMessageSize, length)
} }
if length == 0 { if length == 0 {
@@ -449,7 +449,7 @@ func validateHeaders(headers map[string]string) error {
for k, v := range headers { for k, v := range headers {
totalSize += len(k) + len(v) totalSize += len(k) + len(v)
if totalSize > MaxHeaderSize { if totalSize > MaxHeaderSize {
return ErrMessageTooLarge return fmt.Errorf("%v %v, payload size: %d", ErrMessageTooLarge, MaxHeaderSize, totalSize)
} }
} }
return nil return nil