Merge pull request #3 from edadeal/feature/must_publish

MustPublish method, State Stringer

Author: bbrodriges

lepus.go

@@ -2,6 +2,7 @@ package lepus
 
 import (
 	"errors"
+	"fmt"
 	"strconv"
 	"sync"
 	"sync/atomic"
@@ -10,18 +11,6 @@ import (
 	"github.com/streadway/amqp"
 )
 
-// State indicates publishing state of message
-type State int32
-
-// states
-const (
-	StateUnknown State = iota
-	StatePublished
-	StateReturned
-	StateTimeout
-	StateClosed
-)
-
 type info struct {
 	state int32
 	err   error
@@ -234,3 +223,16 @@ func (d *Delivery) NackDelayed(multiple, mandatory, immediate bool) (State, erro
 		Body:            d.Delivery.Body,
 	})
 }
+
+// MustPublish can be used as a wrapper around `PublishAndWait` and
+// `NackDelayed` methods if you didn't want to process error and state
+// separately.
+func MustPublish(s State, err error) error {
+	if err != nil {
+		return err
+	}
+	if s != StatePublished {
+		return fmt.Errorf("Message publishing failed. Result state: %s", s)
+	}
+	return nil
+}

state.go

@@ -0,0 +1,26 @@
+package lepus
+
+import "strconv"
+
+// State indicates publishing state of message
+type State int32
+
+// states
+const (
+	StateUnknown   State = iota // Unknown
+	StatePublished              // Published
+	StateReturned               // Returned
+	StateTimeout                // Timeout
+	StateClosed                 // Closed
+)
+
+const _State_name = "UnknownPublishedReturnedTimeoutClosed"
+
+var _State_index = [...]uint8{0, 7, 16, 24, 31, 37}
+
+func (i State) String() string {
+	if i < 0 || i >= State(len(_State_index)-1) {
+		return "State(" + strconv.FormatInt(int64(i), 10) + ")"
+	}
+	return _State_name[_State_index[i]:_State_index[i+1]]
+}

state_test.go

@@ -0,0 +1,25 @@
+package lepus
+
+import "testing"
+
+var stateStringerCases = []struct {
+	name     string
+	state    State
+	expected string
+}{
+	{"Unknown", StateUnknown, "Unknown"},
+	{"Published", StatePublished, "Published"},
+	{"Returned", StateReturned, "Returned"},
+	{"Timeout", StateTimeout, "Timeout"},
+	{"Closed", StateClosed, "Closed"},
+}
+
+func TestStateStringer(t *testing.T) {
+	for _, c := range stateStringerCases {
+		t.Run(c.name, func(t *testing.T) {
+			if c.state.String() != c.expected {
+				t.Fatalf("Expecting '%s', got '%s'", c.expected, c.state.String())
+			}
+		})
+	}
+}