mirror of
https://github.com/Gouryella/drip.git
synced 2026-02-23 21:00:44 +00:00
aead68bb62
enhancements - Add adaptive HTTP response handling with automatic streaming for large responses (>1MB) - Implement zero-copy streaming using buffer pools for better performance - Add compression module for reduced bandwidth usage - Add GitHub Container Registry workflow for automated Docker builds - Add production-optimized Dockerfile and docker-compose configuration - Simplify background mode with -d flag and improved daemon management - Update documentation with new command syntax and deployment guides - Clean up unused code and improve error handling - Fix lipgloss style usage (remove unnecessary .Copy() calls)
97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package protocol
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
|
|
"drip/internal/shared/pool"
|
|
)
|
|
|
|
// encodeDataPayload encodes a data header and payload into a frame payload.
|
|
func encodeDataPayload(header DataHeader, data []byte) ([]byte, error) {
|
|
streamIDLen := len(header.StreamID)
|
|
requestIDLen := len(header.RequestID)
|
|
|
|
totalLen := binaryHeaderMinSize + streamIDLen + requestIDLen + len(data)
|
|
payload := make([]byte, totalLen)
|
|
|
|
flags := uint8(header.Type) & 0x07
|
|
if header.IsLast {
|
|
flags |= 0x08
|
|
}
|
|
payload[0] = flags
|
|
|
|
binary.BigEndian.PutUint16(payload[1:3], uint16(streamIDLen))
|
|
binary.BigEndian.PutUint16(payload[3:5], uint16(requestIDLen))
|
|
|
|
offset := binaryHeaderMinSize
|
|
copy(payload[offset:], header.StreamID)
|
|
offset += streamIDLen
|
|
copy(payload[offset:], header.RequestID)
|
|
offset += requestIDLen
|
|
copy(payload[offset:], data)
|
|
|
|
return payload, nil
|
|
}
|
|
|
|
// EncodeDataPayloadPooled encodes with adaptive allocation based on load.
|
|
// Returns payload slice and pool buffer pointer (may be nil).
|
|
func EncodeDataPayloadPooled(header DataHeader, data []byte) (payload []byte, poolBuffer *[]byte, err error) {
|
|
streamIDLen := len(header.StreamID)
|
|
requestIDLen := len(header.RequestID)
|
|
totalLen := binaryHeaderMinSize + streamIDLen + requestIDLen + len(data)
|
|
|
|
dynamicThreshold := GetAdaptiveThreshold()
|
|
|
|
if totalLen < dynamicThreshold {
|
|
regularPayload, err := encodeDataPayload(header, data)
|
|
return regularPayload, nil, err
|
|
}
|
|
|
|
if totalLen > pool.SizeLarge {
|
|
regularPayload, err := encodeDataPayload(header, data)
|
|
return regularPayload, nil, err
|
|
}
|
|
|
|
poolBuffer = pool.GetBuffer(totalLen)
|
|
payload = (*poolBuffer)[:totalLen]
|
|
|
|
flags := uint8(header.Type) & 0x07
|
|
if header.IsLast {
|
|
flags |= 0x08
|
|
}
|
|
payload[0] = flags
|
|
|
|
binary.BigEndian.PutUint16(payload[1:3], uint16(streamIDLen))
|
|
binary.BigEndian.PutUint16(payload[3:5], uint16(requestIDLen))
|
|
|
|
offset := binaryHeaderMinSize
|
|
copy(payload[offset:], header.StreamID)
|
|
offset += streamIDLen
|
|
copy(payload[offset:], header.RequestID)
|
|
offset += requestIDLen
|
|
copy(payload[offset:], data)
|
|
|
|
return payload, poolBuffer, nil
|
|
}
|
|
|
|
// DecodeDataPayload decodes a frame payload into header and data.
|
|
func DecodeDataPayload(payload []byte) (DataHeader, []byte, error) {
|
|
if len(payload) < binaryHeaderMinSize {
|
|
return DataHeader{}, nil, errors.New("invalid payload: too short")
|
|
}
|
|
|
|
var header DataHeader
|
|
if err := header.UnmarshalBinary(payload); err != nil {
|
|
return DataHeader{}, nil, err
|
|
}
|
|
|
|
headerSize := header.Size()
|
|
if len(payload) < headerSize {
|
|
return DataHeader{}, nil, errors.New("invalid payload: data missing")
|
|
}
|
|
|
|
data := payload[headerSize:]
|
|
return header, data, nil
|
|
}
|