Add examples/singleapp/iotest/errreader

Author: devlights

examples/singleapp/iotest/errreader/Taskfile.yml

@@ -0,0 +1,8 @@
+# https://taskfile.dev
+
+version: '3'
+
+tasks:
+  default:
+    cmds:
+      - go test -v .

examples/singleapp/iotest/errreader/errreader_test.go

@@ -0,0 +1,51 @@
+package errreader
+
+import (
+	"errors"
+	"strings"
+	"syscall"
+	"testing"
+	"testing/iotest"
+)
+
+func TestReadOk(t *testing.T) {
+	var (
+		s   = "hello world"
+		r   = strings.NewReader(s)
+		p   = make([]byte, len(s))
+		err error
+	)
+
+	err = read(r, p)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if string(p) != s {
+		t.Fatalf("[want] equal\t[got] not equal (%s, %s)", string(p), s)
+	}
+}
+
+func TestReadTooManyEAGAIN(t *testing.T) {
+	//
+	// iotest.ErrReader() は、指定したエラーを返す io.Reader を返してくれる。
+	// io.Readerを使って何らかの処理を行う関数などを実装している場合のエラーテストに便利。
+	// (例えば、ノンブロッキング処理をしている実装で、ずっとEAGAINが返ってくる場合のテストなど)
+	//
+	// - https://pkg.go.dev/testing/iotest@go1.23.0#ErrReader
+	//
+	var (
+		s = "hello world"
+		r = iotest.ErrReader(syscall.EAGAIN)
+		p = make([]byte, len(s))
+	)
+
+	err := read(r, p)
+	if err == nil {
+		t.Fatal("[want] err \t[got] nil")
+	}
+
+	if !errors.Is(err, ErrTooManyEAGAIN) {
+		t.Fatalf("[want] ErrRetryOver\t[got] %v", err)
+	}
+}

examples/singleapp/iotest/errreader/read.go

@@ -0,0 +1,55 @@
+package errreader
+
+import (
+	"errors"
+	"io"
+	"syscall"
+	"time"
+)
+
+var (
+	Interval         = 10 * time.Millisecond
+	ErrTooManyEAGAIN = errors.New("retry over (EAGAIN)")
+	MaxRetryCount    = 5
+)
+
+func read(r io.Reader, p []byte) error {
+	var (
+		buf     = make([]byte, 2)
+		numRead int
+		offset  int
+		count   int
+		err     error
+	)
+
+	for count = 0; count < MaxRetryCount; {
+		if len(p) <= offset {
+			break
+		}
+
+		clear(buf)
+
+		numRead, err = r.Read(buf)
+		if err != nil {
+			switch {
+			case errors.Is(err, io.EOF):
+				break
+			case errors.Is(err, syscall.EAGAIN):
+				time.Sleep(Interval)
+				count++
+				continue
+			default:
+				return err
+			}
+		}
+
+		copy(p[offset:offset+numRead], buf[:numRead])
+		offset += numRead
+	}
+
+	if MaxRetryCount <= count {
+		return ErrTooManyEAGAIN
+	}
+
+	return nil
+}