gistfile1.txt

package workers

type Action interface {
	Call(queue string, message *Msg, next func() bool) bool
}

type Middlewares struct {
	actions []Action
}

func (m *Middlewares) Append(action Action) {
	m.actions = append(m.actions, action)
}

func (m *Middlewares) Prepend(action Action) {
	actions := make([]Action, len(m.actions)+1)
	actions[0] = action
	copy(actions[1:], m.actions)
	m.actions = actions
}

func (m *Middlewares) call(queue string, message *Msg, final func()) bool {
	return continuation(m.actions, queue, message, final)()
}

func continuation(actions []Action, queue string, message *Msg, final func()) func() bool {
	if Running {
		return func() (acknowledge bool) {
			if len(actions) > 0 {
				acknowledge = actions[0].Call(
					queue,
					message,
					continuation(actions[1:], queue, message, final),
				)

				if !acknowledge {
					return
				}
			} else {
				final()
			}

			return true
		}
	} else {
		return func() (acknowledge bool) {
			return false
		}
	}
}

func NewMiddleware(actions ...Action) *Middlewares {
	return &Middlewares{actions}
}