email_custom.go

package email

import (
	"github.com/thejackrabbit/aero/conf"
	"github.com/thejackrabbit/aero/ds"
	"github.com/thejackrabbit/aero/panik"
	"github.com/thejackrabbit/aero/que"
	"time"
)

func (e *Email) Enque() {

	if mails == nil { // initialization
		startLoop()
	}

	go func() {
		d, err := ds.ToBytes(e, false)
		panik.On(err)
		mails <- d
	}()
}

func (e *Email) SendLater(after string) {
	d, err := time.ParseDuration(after)
	panik.On(err)
	e.SendAt(time.Now().Add(d))
}

func (e *Email) SendAt(t time.Time) {
	u := t.Unix()
	if time.Now().Unix() < u {
		panik.Do("Send time %s is earlier than current time", t)
	}
	e.SendAfter = &u
	e.Enque()
}

var Queue que.Queue

var mails chan []byte
var size int = 100

func init() {
	if conf.Exists("email.queue.default") {
		Queue = que.NewQueue("email.queue.default")
	}
}

func startLoop() {
	if mails != nil {
		return
	}

	mails = make(chan []byte, conf.Int(size, "email", "buffer"))
	panik.If(Queue == nil, "Queue is not initialized")

	go func() {
		for {
			select {
			case mb := <-mails:
				err := Queue.Push(mb)
				panik.On(err)
			}
		}
	}()
}