Merge pull request #532 from redis/feat-rueidishook-error-result

feat: add NewErrorResult() to rueidishook

rueidishook/hook.go

@@ -3,6 +3,7 @@ package rueidishook
 import (
 	"context"
 	"time"
+	"unsafe"
 
 	"github.com/redis/rueidis"
 )
@@ -148,3 +149,25 @@ func (e *extended) Dedicate() (client rueidis.DedicatedClient, cancel func()) {
 func (e *extended) Nodes() map[string]rueidis.Client {
 	panic("Nodes() is not allowed with rueidis.DedicatedClient")
 }
+
+type result struct {
+	err error
+	val rueidis.RedisMessage
+}
+
+func NewErrorResult(err error) rueidis.RedisResult {
+	r := result{err: err}
+	return *(*rueidis.RedisResult)(unsafe.Pointer(&r))
+}
+
+type stream struct {
+	p *int
+	w *int
+	e error
+	n int
+}
+
+func NewErrorResultStream(err error) rueidis.RedisResultStream {
+	r := stream{e: err}
+	return *(*rueidis.RedisResultStream)(unsafe.Pointer(&r))
+}

rueidishook/hook_test.go

@@ -287,3 +287,25 @@ func TestForbiddenMethodForDedicatedClient(t *testing.T) {
 		shouldpanic(c.fn, c.msg)
 	}
 }
+
+func TestNewErrorResult(t *testing.T) {
+	e := errors.New("err")
+	r := NewErrorResult(e)
+	if r.Error() != e {
+		t.Fatal("unexpected err")
+	}
+}
+
+func TestNewErrorResultStream(t *testing.T) {
+	e := errors.New("err")
+	r := NewErrorResultStream(e)
+	if r.Error() != e {
+		t.Fatal("unexpected err")
+	}
+	if r.HasNext() {
+		t.Fatal("unexpected err")
+	}
+	if n, err := r.WriteTo(io.Discard); err != e || n != 0 {
+		t.Fatal("unexpected err or n")
+	}
+}