Make the AskOnce return error.

Author: johnteee

actor.go

@@ -1,9 +1,14 @@
 package fpgo
 
 import (
+	"fmt"
 	"time"
 )
 
+var (
+	ErrActorAskTimeout = fmt.Errorf("ErrActorAskTimeout")
+)
+
 // ActorHandle A target could send messages
 type ActorHandle interface {
 	Send(message interface{})
@@ -144,20 +149,22 @@ func AskNewByOptionsGenerics(message interface{}, ioCh *chan interface{}) *AskDe
 }
 
 // AskOnce Sender Ask
-func (askSelf *AskDef) AskOnce(target ActorHandle, timeout *time.Duration) interface{} {
+func (askSelf *AskDef) AskOnce(target ActorHandle, timeout *time.Duration) (interface{}, error) {
 	ch := askSelf.AskChannel(target)
+	defer close(*ch)
 	var result interface{}
+	// var err error
 	if timeout == nil {
 		result = <-*ch
 	} else {
 		select {
 		case result = <-*ch:
 		case <-time.After(*timeout):
+			return result, ErrActorAskTimeout
 		}
 	}
-	close(*ch)
 
-	return result
+	return result, nil
 }
 
 // AskChannel Sender Ask

actor_test.go

@@ -87,6 +87,8 @@ func TestActorCommon(t *testing.T) {
 func TestActorAsk(t *testing.T) {
 	var expectedInt int
 	var actual int
+	var err error
+	var result interface{}
 
 	// Testee
 	actorRoot := Actor.New(func(self *ActorDef, input interface{}) {
@@ -112,12 +114,14 @@ func TestActorAsk(t *testing.T) {
 	// Normal cases
 	actual = 0
 	expectedInt = 10
-	actual, _ = Maybe.Just(AskNewGenerics(1).AskOnce(actorRoot, nil)).ToInt()
+	result, _ = Ask.New(1).AskOnce(actorRoot, nil)
+	actual, _ = Maybe.Just(result).ToInt()
 	assert.Equal(t, expectedInt, actual)
 	// Ask with Timeout
 	actual = 0
 	expectedInt = 20
-	actual, _ = Maybe.Just(AskNewGenerics(2).AskOnce(actorRoot, &timeout)).ToInt()
+	result, _ = Ask.New(2).AskOnce(actorRoot, &timeout)
+	actual, _ = Maybe.Just(result).ToInt()
 	assert.Equal(t, expectedInt, actual)
 	// Ask channel
 	actual = 0
@@ -130,6 +134,8 @@ func TestActorAsk(t *testing.T) {
 	// Timeout cases
 	actual = 9999
 	expectedInt = 0
-	actual, _ = Maybe.Just(AskNewGenerics(-1).AskOnce(actorRoot, &timeout)).ToInt()
+	result, err = Ask.New(-1).AskOnce(actorRoot, &timeout)
+	actual, _ = Maybe.Just(result).ToInt()
 	assert.Equal(t, expectedInt, actual)
+	assert.Equal(t, ErrActorAskTimeout, err)
 }