Merge branch 'master' of http://gitlab.fjmaimaimai.com/mmm-go/gocomm
正在显示
3 个修改的文件
包含
225 行增加
和
0 行删除
common/mail.go
0 → 100644
| 1 | +package common | ||
| 2 | + | ||
| 3 | +import ( | ||
| 4 | + "bytes" | ||
| 5 | + "crypto/tls" | ||
| 6 | + "fmt" | ||
| 7 | + "log" | ||
| 8 | + "mime" | ||
| 9 | + "net" | ||
| 10 | + "net/smtp" | ||
| 11 | +) | ||
| 12 | +/* | ||
| 13 | +用途 发送邮件 | ||
| 14 | +1.初始化 | ||
| 15 | +InitMailService(&MailConfig{ | ||
| 16 | + Host:"smtp.qq.com", | ||
| 17 | + Port:25, | ||
| 18 | + From:"785410885@qq.com", | ||
| 19 | + Password:"ibfduqhfmgypbffe", //授权码 | ||
| 20 | + IsUseSsl:false, | ||
| 21 | + }) | ||
| 22 | +或者 | ||
| 23 | +InitMailService(&MailConfig{ | ||
| 24 | + Host:"smtp.qq.com", | ||
| 25 | + Port:465, | ||
| 26 | + From:"785410885@qq.com", | ||
| 27 | + Password:"ibfduqhfmgypbffe", //授权码 | ||
| 28 | + IsUseSsl:true, | ||
| 29 | + }) | ||
| 30 | + | ||
| 31 | +2.发送邮件 | ||
| 32 | +SendMail(&MailContent{ | ||
| 33 | + ToMail:"892423867@qq.com", | ||
| 34 | + Subject:"测试邮件", | ||
| 35 | + Body:[]byte("邮件内容..."), | ||
| 36 | + }) | ||
| 37 | + */ | ||
| 38 | +var( | ||
| 39 | + ErrorInvalidMailConfig = fmt.Errorf("mail config error") | ||
| 40 | +) | ||
| 41 | + | ||
| 42 | +var DefaultMail *MailService | ||
| 43 | +//邮件配置 | ||
| 44 | +type MailConfig struct { | ||
| 45 | + Host string | ||
| 46 | + Port int | ||
| 47 | + From string | ||
| 48 | + Password string | ||
| 49 | + IsUseSsl bool | ||
| 50 | +} | ||
| 51 | + | ||
| 52 | +//初始化邮件服务 | ||
| 53 | +func InitMailService(mail *MailConfig){ | ||
| 54 | + DefaultMail = NewMailService(mail) | ||
| 55 | +} | ||
| 56 | +type MailService struct { | ||
| 57 | + Config *MailConfig | ||
| 58 | +} | ||
| 59 | +func NewMailService(config *MailConfig)*MailService{ | ||
| 60 | + return &MailService{ | ||
| 61 | + Config:config, | ||
| 62 | + } | ||
| 63 | +} | ||
| 64 | +//to: 邮件发送目标 多个 | ||
| 65 | +func (mail *MailService)SendMail(to []string, subject string, body []byte)(err error){ | ||
| 66 | + if err =mail.CheckConfig();err!=nil{ | ||
| 67 | + return | ||
| 68 | + } | ||
| 69 | + address :=fmt.Sprintf("%v:%v",mail.Config.Host,mail.Config.Port) | ||
| 70 | + auth := smtp.PlainAuth("", mail.Config.From, mail.Config.Password, mail.Config.Host) | ||
| 71 | + if !mail.Config.IsUseSsl{ //qq 普通发送 端口25 | ||
| 72 | + // hostname is used by PlainAuth to validate the TLS certificate. | ||
| 73 | + err = smtp.SendMail(address, auth, mail.Config.From,to, body) | ||
| 74 | + if err != nil { | ||
| 75 | + return err | ||
| 76 | + } | ||
| 77 | + return | ||
| 78 | + } | ||
| 79 | + if err=SendMailUsingTLS(address, auth, mail.Config.From,to, body);err!=nil{ | ||
| 80 | + return | ||
| 81 | + } | ||
| 82 | + return | ||
| 83 | +} | ||
| 84 | +//检查配置 | ||
| 85 | +func(mail *MailService)CheckConfig()error{ | ||
| 86 | + config :=mail.Config | ||
| 87 | + if len(config.Host)==0 || len(config.From)==0 || config.Port==0 || len(config.Password)==0{ | ||
| 88 | + return ErrorInvalidMailConfig | ||
| 89 | + } | ||
| 90 | + return nil | ||
| 91 | +} | ||
| 92 | + | ||
| 93 | +//邮件内容 | ||
| 94 | +type MailContent struct { | ||
| 95 | + ToMail string | ||
| 96 | + Subject string | ||
| 97 | + Body []byte | ||
| 98 | + ContentType string //html /plain | ||
| 99 | +} | ||
| 100 | +//发送邮件 | ||
| 101 | +func SendMail(content *MailContent)(err error){ | ||
| 102 | + if DefaultMail==nil{ | ||
| 103 | + return ErrorInvalidMailConfig | ||
| 104 | + } | ||
| 105 | + var to,subject,contentType string | ||
| 106 | + var body []byte | ||
| 107 | + to = content.ToMail | ||
| 108 | + subject = content.Subject | ||
| 109 | + contentType = content.ContentType | ||
| 110 | + if contentType==""{ | ||
| 111 | + contentType="text/html; charset=UTF-8" | ||
| 112 | + } | ||
| 113 | + header :=make(map[string]string) | ||
| 114 | + header["From"] = mime.BEncoding.Encode("utf-8",DefaultMail.Config.From) //from 使用其他字符串,显示xx发送 代发为 DefaultMail.Config.From | ||
| 115 | + header["To"] = to | ||
| 116 | + header["Subject"] = mime.BEncoding.Encode("utf-8",subject) | ||
| 117 | + header["Content-Type"] = contentType | ||
| 118 | + var buf bytes.Buffer | ||
| 119 | + for k, v := range header { | ||
| 120 | + buf.WriteString(fmt.Sprintf("%s: %s\r\n", k, v)) | ||
| 121 | + } | ||
| 122 | + buf.WriteString("\r\n") | ||
| 123 | + buf.Write(body) | ||
| 124 | + return DefaultMail.SendMail([]string{to},subject,buf.Bytes()) | ||
| 125 | +} | ||
| 126 | +//使用 ssl发送 端口465 | ||
| 127 | +//return a smtp client | ||
| 128 | +func Dial(addr string) (*smtp.Client, error) { | ||
| 129 | + conn, err := tls.Dial("tcp", addr, nil) | ||
| 130 | + if err != nil { | ||
| 131 | + log.Panicln("Dialing Error:", err) | ||
| 132 | + return nil, err | ||
| 133 | + } | ||
| 134 | + //分解主机端口字符串 | ||
| 135 | + host, _, _ := net.SplitHostPort(addr) | ||
| 136 | + return smtp.NewClient(conn, host) | ||
| 137 | +} | ||
| 138 | +//参考net/smtp的func SendMail() | ||
| 139 | +//使用net.Dial连接tls(ssl)端口时,smtp.NewClient()会卡住且不提示err | ||
| 140 | +//len(to)>1时,to[1]开始提示是密送 | ||
| 141 | +func SendMailUsingTLS(addr string, auth smtp.Auth, from string, | ||
| 142 | + to []string, msg []byte) (err error) { | ||
| 143 | + | ||
| 144 | + //create smtp client | ||
| 145 | + c, err := Dial(addr) | ||
| 146 | + if err != nil { | ||
| 147 | + log.Println("Create smpt client error:", err) | ||
| 148 | + return err | ||
| 149 | + } | ||
| 150 | + defer c.Close() | ||
| 151 | + | ||
| 152 | + if auth != nil { | ||
| 153 | + if ok, _ := c.Extension("AUTH"); ok { | ||
| 154 | + if err = c.Auth(auth); err != nil { | ||
| 155 | + log.Println("Error during AUTH", err) | ||
| 156 | + return err | ||
| 157 | + } | ||
| 158 | + } | ||
| 159 | + } | ||
| 160 | + | ||
| 161 | + if err = c.Mail(from); err != nil { | ||
| 162 | + return err | ||
| 163 | + } | ||
| 164 | + | ||
| 165 | + for _, addr := range to { | ||
| 166 | + if err = c.Rcpt(addr); err != nil { | ||
| 167 | + return err | ||
| 168 | + } | ||
| 169 | + } | ||
| 170 | + | ||
| 171 | + w, err := c.Data() | ||
| 172 | + if err != nil { | ||
| 173 | + return err | ||
| 174 | + } | ||
| 175 | + | ||
| 176 | + _, err = w.Write(msg) | ||
| 177 | + if err != nil { | ||
| 178 | + return err | ||
| 179 | + } | ||
| 180 | + | ||
| 181 | + err = w.Close() | ||
| 182 | + if err != nil { | ||
| 183 | + return err | ||
| 184 | + } | ||
| 185 | + return c.Quit() | ||
| 186 | +} |
common/mail_test.go
0 → 100644
| 1 | +package common | ||
| 2 | + | ||
| 3 | +import ( | ||
| 4 | + "testing" | ||
| 5 | +) | ||
| 6 | + | ||
| 7 | +//Example | ||
| 8 | +func TestSendMail(t *testing.T) { | ||
| 9 | + InitMailService(&MailConfig{ | ||
| 10 | + Host:"smtp.qq.com", | ||
| 11 | + Port:25, | ||
| 12 | + From:"785410885@qq.com", | ||
| 13 | + Password:"ibfduqhfmgypbffe", //授权码 | ||
| 14 | + IsUseSsl:false, | ||
| 15 | + }) | ||
| 16 | + //SendMail(&MailContent{ | ||
| 17 | + // ToMail:"892423867@qq.com", | ||
| 18 | + // Subject:"测试邮件", | ||
| 19 | + // Body:[]byte("邮件内容..."), | ||
| 20 | + //}) | ||
| 21 | +} | ||
| 22 | + | ||
| 23 | +func TestSendMailTls(t *testing.T) { | ||
| 24 | + InitMailService(&MailConfig{ | ||
| 25 | + Host:"smtp.qq.com", | ||
| 26 | + Port:465, | ||
| 27 | + From:"785410885@qq.com", | ||
| 28 | + Password:"ibfduqhfmgypbffe", //授权码 | ||
| 29 | + IsUseSsl:true, | ||
| 30 | + }) | ||
| 31 | + //SendMail("892423867@qq.com","测试邮件",[]byte("邮件内容...")) | ||
| 32 | + //SendMail(&MailContent{ | ||
| 33 | + // ToMail:"892423867@qq.com", | ||
| 34 | + // Subject:"测试邮件", | ||
| 35 | + // Body:[]byte("邮件内容..."), | ||
| 36 | + //}) | ||
| 37 | +} |
| @@ -15,4 +15,6 @@ func NewBeeormEngine(conf config.Mysql){ | @@ -15,4 +15,6 @@ func NewBeeormEngine(conf config.Mysql){ | ||
| 15 | } | 15 | } |
| 16 | orm.SetMaxIdleConns("default", conf.MaxIdle) | 16 | orm.SetMaxIdleConns("default", conf.MaxIdle) |
| 17 | orm.SetMaxOpenConns("default", conf.MaxOpen) | 17 | orm.SetMaxOpenConns("default", conf.MaxOpen) |
| 18 | + //orm.DefaultTimeLoc = time.Local | ||
| 19 | + //orm.Debug = true | ||
| 18 | } | 20 | } |
-
请 注册 或 登录 后发表评论