Add mail

Author: minhduc140583

client/health.go

@@ -0,0 +1,63 @@
+package client
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net"
+	"net/http"
+	"time"
+)
+
+type HealthChecker struct {
+	name    string
+	url     string
+	timeout time.Duration
+}
+
+func NewHealthChecker(name, url string, options ...time.Duration) *HealthChecker {
+	if len(options) >= 1 && options[0] > 0 {
+		return &HealthChecker{name, url, options[0]}
+	} else {
+		return &HealthChecker{name, url, 4 * time.Second}
+	}
+}
+
+func (s *HealthChecker) Name() string {
+	return s.name
+}
+
+func (s *HealthChecker) Check(ctx context.Context) (map[string]interface{}, error) {
+	res := make(map[string]interface{})
+	client := http.Client{
+		Timeout: s.timeout,
+		// never follow redirects
+		CheckRedirect: func(*http.Request, []*http.Request) error {
+			return http.ErrUseLastResponse
+		},
+	}
+	resp, err := client.Get(s.url)
+	if e, ok := err.(net.Error); ok && e.Timeout() {
+		return res, fmt.Errorf("time out: %w", e)
+	} else if err != nil {
+		return res, err
+	}
+	_, _ = io.Copy(ioutil.Discard, resp.Body)
+	_ = resp.Body.Close()
+	if resp.StatusCode >= 500 {
+		return res, fmt.Errorf("status code is: %d", resp.StatusCode)
+	}
+	return res, nil
+}
+
+func (s *HealthChecker) Build(ctx context.Context, data map[string]interface{}, err error) map[string]interface{} {
+	if err == nil {
+		return data
+	}
+	if data == nil {
+		data = make(map[string]interface{}, 0)
+	}
+	data["error"] = err.Error()
+	return data
+}

mail/default_simple_mail_sender.go

@@ -0,0 +1,30 @@
+package mail
+
+type DefaultSimpleMailSender struct {
+	MailSender MailSender
+}
+
+func NewSimpleMailSender(mailService MailSender) *DefaultSimpleMailSender {
+	return &DefaultSimpleMailSender{MailSender: mailService}
+}
+
+func (s *DefaultSimpleMailSender) Send(m SimpleMail) error {
+	var contents = make([]Content, len(m.Content))
+	for i, content := range m.Content {
+		contents[i] = content
+	}
+	mail := NewMailInit(m.From, m.Subject, m.To, m.Cc, contents...)
+	return s.MailSender.Send(*mail)
+}
+
+func NewSimpleHtmlMail(mailFrom Email, subject string, mailTo []Email, cc *[]Email, htmlContent string) *SimpleMail {
+	html := NewContent("text/html", htmlContent)
+	s := SimpleMail{
+		From:    mailFrom,
+		To:      mailTo,
+		Cc:      cc,
+		Subject: subject,
+		Content: []Content{*html},
+	}
+	return &s
+}

mail/default_template_loader.go

@@ -0,0 +1,20 @@
+package mail
+
+import "context"
+
+type DefaultTemplateLoader struct {
+	Subject string
+	Body    string
+}
+
+func NewTemplateLoader(subject string, body string) *DefaultTemplateLoader {
+	return &DefaultTemplateLoader{subject, body}
+}
+
+func NewTemplateLoaderByConfig(c TemplateConfig) *DefaultTemplateLoader {
+	return &DefaultTemplateLoader{c.Subject, c.Body}
+}
+
+func (s *DefaultTemplateLoader) Load(ctx context.Context, id string) (string, string, error) {
+	return s.Subject, s.Body, nil
+}