71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package email
|
||
|
||
import (
|
||
"crypto/tls"
|
||
|
||
"go-common/app/job/main/videoup-report/conf"
|
||
"go-common/app/job/main/videoup-report/model/email"
|
||
"go-common/library/cache/redis"
|
||
gomail "gopkg.in/gomail.v2"
|
||
)
|
||
|
||
// Dao is redis dao.
|
||
type Dao struct {
|
||
c *conf.Config
|
||
redis *redis.Pool
|
||
email *gomail.Dialer
|
||
FansAddr map[int16][]string
|
||
emailAddr map[string][]string
|
||
PrivateAddr map[string][]string
|
||
//fast behavior detector
|
||
detector *email.FastDetector
|
||
//快速通道token
|
||
fastChan chan int
|
||
//邮件发送api的频率token，发送邮件5s后插入
|
||
controlChan chan int64
|
||
}
|
||
|
||
// New is new redis dao.
|
||
func New(c *conf.Config) (d *Dao) {
|
||
emailAddr := make(map[string][]string)
|
||
for _, v := range c.Mail.Addr {
|
||
emailAddr[v.Type] = v.Addr
|
||
}
|
||
|
||
privateMail := make(map[string][]string)
|
||
for _, v := range c.Mail.PrivateAddr {
|
||
privateMail[v.Type] = v.Addr
|
||
}
|
||
|
||
d = &Dao{
|
||
c: c,
|
||
redis: redis.NewPool(c.Redis.Mail),
|
||
email: gomail.NewDialer(c.Mail.Host, c.Mail.Port, c.Mail.Username, c.Mail.Password),
|
||
emailAddr: emailAddr,
|
||
PrivateAddr: privateMail,
|
||
detector: email.NewFastDetector(c.Mail.SpeedThreshold, c.Mail.OverspeedThreshold),
|
||
fastChan: make(chan int, 10240),
|
||
controlChan: make(chan int64, 1),
|
||
}
|
||
|
||
d.email.TLSConfig = &tls.Config{
|
||
InsecureSkipVerify: true,
|
||
}
|
||
d.fastChan <- 1
|
||
d.controlChan <- 1
|
||
return d
|
||
}
|
||
|
||
//Close close
|
||
func (d *Dao) Close() (err error) {
|
||
if d.redis != nil {
|
||
err = d.redis.Close()
|
||
}
|
||
return
|
||
}
|
||
|
||
//FastChan get fast channel
|
||
func (d *Dao) FastChan() <-chan int {
|
||
return d.fastChan
|
||
}
|