使用自定义流程实现网关协议解析转发功能
This commit is contained in:
22
pkg/utils/utils.go
Normal file
22
pkg/utils/utils.go
Normal file
@@ -0,0 +1,22 @@
|
||||
package utils
|
||||
|
||||
import "io"
|
||||
|
||||
func ReadByte(reader io.Reader) (byte, error) {
|
||||
buffer, err := ReadBuffer(reader, 1)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return buffer[0], nil
|
||||
}
|
||||
|
||||
func ReadBuffer(reader io.Reader, size int) ([]byte, error) {
|
||||
buffer := make([]byte, size)
|
||||
_, err := io.ReadFull(reader, buffer)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return buffer, nil
|
||||
}
|
||||
Reference in New Issue
Block a user