burlesque/queue.go

package main

type (
	Queue struct {
		Name    string
		Counter *Counter
	}
)

var (
	queues = make(map[string]*Queue)
)

func (q *Queue) Push(msg Message) bool {
	var err error

	q.Counter.Write(func(i uint) bool {
		key := NewKey(q.Name, i)
		err = storage.Set(key, msg)
		if err != nil {
			Error(err, "Failed to write %d bytes to record '%s'", len(msg), key)
		}

		return (err == nil)
	})

	return (err == nil)
}

func (q *Queue) TryFetch(abort chan bool) (Message, bool) {
	if q.Counter.Distance() > 0 {
		return q.Fetch(abort)
	} else {
		return Message{}, false
	}
}

func (q *Queue) Fetch(abort chan bool) (Message, bool) {
	var i uint

	select {
	case i = <-q.Counter.Read:
	case <-abort:
		return Message{}, false
	}

	key := NewKey(q.Name, i)
	msg, err := storage.Get(key)
	if err != nil {
		Error(err, "Failed to read record '%s'", key)
		return msg, false
	}

	err = storage.Remove(key)
	if err != nil {
		Error(err, "Failed to delete record '%s'", key)
		return msg, false
	}

	return msg, true
}

func GetQueue(name string) *Queue {
	if _, ok := queues[name]; !ok {
		RegisterQueue(name, 0, 0)
	}
	return queues[name]
}

func RegisterQueue(name string, wi, ri uint) {
	queues[name] = &Queue{
		Name:    name,
		Counter: NewCounter(wi, ri),
	}
}