implement io.ByteReader in socket transport (#180)

ortuman · Dec 13, 2021 · 17a2237 · 17a2237
1 parent f7f39f7
commit 17a2237
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pkg/transport/socket.go b/pkg/transport/socket.go
@@ -66,6 +66,19 @@ func (s *socketTransport) Read(p []byte) (n int, err error) {
 	return s.rd.Read(p)
 }
 
+func (s *socketTransport) ReadByte() (byte, error) {
+	var p [1]byte
+	n, err := s.rd.Read(p[:])
+	switch {
+	case n == 1 && err == nil:
+		return p[0], nil
+	case err != nil:
+		return 0, err
+	default:
+		return 0, nil
+	}
+}
+
 func (s *socketTransport) Write(p []byte) (n int, err error) {
 	if s.bw == nil {
 		s.grabBuffWriter()

diff --git a/pkg/transport/transport.go b/pkg/transport/transport.go
@@ -52,6 +52,7 @@ const (
 // Transport represents a stream transport mechanism.
 type Transport interface {
 	io.ReadWriteCloser
+	io.ByteReader
 
 	// Type returns transport type value.
 	Type() Type