sugarBecomer
51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package schedule
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/gob"
|
|
"likedream-goframe/internal/framework/logger"
|
|
"likedream-goframe/internal/framework/model/email"
|
|
"likedream-goframe/internal/utils"
|
|
|
|
"github.com/hibiken/asynq"
|
|
"github.com/samber/oops"
|
|
"github.com/wneessen/go-mail"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type EmailTaskHandler struct {
|
|
}
|
|
|
|
func (h *EmailTaskHandler) ProcessTask(ctx context.Context, task *asynq.Task) (err error) {
|
|
defer utils.Pass(&err)
|
|
|
|
var payload email.SendPayload
|
|
|
|
e := gob.NewDecoder(bytes.NewReader(task.Payload())).Decode(&payload)
|
|
utils.OkOrPanic(e, oops.WithContext(ctx).Wrap)
|
|
|
|
message := mail.NewMsg()
|
|
|
|
e = message.FromFormat(payload.FromName, payload.FromEmail)
|
|
utils.OkOrPanic(e, oops.WithContext(ctx).Wrap)
|
|
|
|
e = message.AddTo(payload.To)
|
|
utils.OkOrPanic(e, oops.WithContext(ctx).Wrap)
|
|
|
|
message.Subject(payload.Subject)
|
|
|
|
message.SetBodyString(mail.TypeTextHTML, payload.Body)
|
|
|
|
client, e := mail.NewClient(payload.SMTPHost, mail.WithPort(payload.SMTPPort), mail.WithSMTPAuth(mail.SMTPAuthAutoDiscover),
|
|
mail.WithUsername(payload.SMTPUsername), mail.WithPassword(payload.SMTPPassword))
|
|
utils.OkOrPanic(e, oops.WithContext(ctx).Wrap)
|
|
|
|
defer utils.LogError(ctx, "关闭邮件客户端失败", client.Close())
|
|
|
|
e = client.DialAndSendWithContext(ctx, message)
|
|
utils.OkOrPanic(e, oops.WithContext(ctx).Wrap)
|
|
logger.Logger.Info("邮件发送完成", zap.String("邮箱", payload.To))
|
|
return
|
|
}
|