-
-
Notifications
You must be signed in to change notification settings - Fork 121
/
queue.go
92 lines (75 loc) · 1.6 KB
/
queue.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package email
import (
"fmt"
"sync"
"github.com/ArtalkJS/Artalk/internal/config"
"github.com/ArtalkJS/Artalk/internal/log"
)
type EmailConf struct {
config.EmailConf
Sender Sender
OnSendSuccess func(email *Email)
}
type EmailQueue struct {
conf EmailConf
sender Sender
ch chan *Email
mux sync.Mutex
closed bool
}
// Initialize Email Queue
func NewQueue(conf EmailConf) *EmailQueue {
queue := &EmailQueue{
conf: conf,
}
// init email queue
queue.ch = make(chan *Email, conf.Queue.BufferSize)
log.Debug("[Email] Email Queue initialize complete")
// init email sender
if conf.Sender != nil {
queue.sender = conf.Sender
} else {
if sender, err := NewSender(queue.conf); err == nil {
queue.sender = sender
} else {
log.Error("[Email] Email Sender initialize failed: ", err)
}
}
// init queue worker
go func() {
for email := range queue.ch {
queue.handleEmail(email)
}
}()
return queue
}
func (q *EmailQueue) handleEmail(email *Email) {
log.Debug(fmt.Sprintf("[Email] Sending an email %+v: ", email))
if q.sender == nil {
log.Error("[Email] Email Sender is nil")
return
}
if isOK := q.sender.Send(email); isOK {
if q.conf.OnSendSuccess != nil {
q.conf.OnSendSuccess(email)
}
} else {
log.Errorf("[Email] Failed send email to addr: %s", email.ToAddr)
}
}
// Add an email to the sending queue
func (q *EmailQueue) Push(email *Email) {
if q.closed {
log.Error("[Email] Queue closed, dropping email")
return
}
q.ch <- email
}
func (q *EmailQueue) Close() {
q.mux.Lock()
defer q.mux.Unlock()
if !q.closed {
close(q.ch)
q.closed = true
}
}