相关代码已经放在github：https://github.com/nangongchengfeng/Go_gin/tree/main/Projects
GO的WEB编程（GIN实现邮件接口报警）
Gin编写邮件接口（支持多人发送）
Gin编写邮件告警接口（添加优化日志记录）
上面3个已经完成基本功能，但是所有的代码放在一个main，配置也是写死代码里面，不方便修改调整。现在在以上的基础上添加新的功能
（1）加入配置文件读写管理
（2）项目拆分
因为日常范围，我们在操作系统上，需要报警时，只能采用mailx来使用。需要配置账号，密码，和邮箱认证。如果需要多台使用的话，岂不是很麻烦，要配置多台，这个导致密码很不安全，容易泄露。所以，为了安全，有效，更方便，我们可以采用接口发送邮件。
（1）构建接口
（2）传入post的json情况
（3）把相应json转换字符
（4）发送邮件
开始下面项目规划
main.go
package main
import (
"code/mail_qq/log"
"code/mail_qq/routers"
"code/mail_qq/setting"
"fmt"
)
/*
支持多人发送
curl http://10.10.10.3:7070/send -H "Content-Type:application/json" -X POST -d '{"source":"heian","contacts":["账号@mail_qq.com","账号@mail_qq.com"],"subject":"多人测试","content":"现在进行多人测试"}'
*/
/*
zapcore.Core需要三个配置——Encoder，WriteSyncer，LogLevel
Encoder:编码器(如何写入日志)。我们将使用开箱即用的NewJSONEncoder()
WriterSyncer ：指定日志将写到哪里去。我们使用zapcore.AddSync()
Log Level：哪种级别的日志将被写入。
*/
//var sugarLogger *zap.SugaredLogger
func main() {
log.InitLogger()
defer log.SugarLogger.Sync()
port := setting.GetPort()
r := routers.SetupRouter()
r.Run(":" + fmt.Sprint(port))
log.SugarLogger.Infof("Success! Port is start")
//r.Run(":8080")
}
conf/config.ini
port = 8080
[mail]
user = 账号@qq.com
password = 密码
host = smtp.qq.com:25
source = heianutil/ GetIP.go
工具类，获取客户端的IP
package util
import "github.com/gin-gonic/gin"
//获取ip
func GetRequestIP(c *gin.Context) string {
reqIP := c.ClientIP()
if reqIP == "::1" {
reqIP = "127.0.0.1"
}
return reqIP
}
setting/setting.go
加入配置文件，给代码返回相应的参数
package setting
import (
"fmt"
"os"
"github.com/go-ini/ini"
)
func GetMail() (user, password, host, source string) {
//读取.ini里面的数据库配置
config, err := ini.Load("conf/config.ini")
if err != nil {
//失败
fmt.Printf("Fail to read file: %v", err)
os.Exit(1)
}
host = config.Section("mail").Key("host").String()
//port = config.Section("mail").Key("port").String()
user = config.Section("mail").Key("user").String()
password = config.Section("mail").Key("password").String()
source = config.Section("mail").Key("source").String()
//fmt.Println(user, password, host, source)
return
}
func GetPort() (prot string) {
config, err := ini.Load("conf/config.ini")
if err != nil {
//失败
fmt.Printf("Fail to read file: %v", err)
os.Exit(1)
}
prot = config.Section("").Key("port").String()
return
}
log/ log.go
日志切割和分割类，主要记录日志
package log
import (
"github.com/natefinch/lumberjack"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
var SugarLogger *zap.SugaredLogger
func InitLogger() {
writeSyncer := getLogWriter()
encoder := getEncoder()
core := zapcore.NewCore(encoder, writeSyncer, zapcore.DebugLevel)
logger := zap.New(core, zap.AddCaller())
SugarLogger = logger.Sugar()
}
func getEncoder() zapcore.Encoder {
encoderConfig := zap.NewProductionEncoderConfig()
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
return zapcore.NewConsoleEncoder(encoderConfig)
}
func getLogWriter() zapcore.WriteSyncer {
/*
Lumberjack Logger采用以下属性作为输入:
Filename: 日志文件的位置
MaxSize：在进行切割之前，日志文件的最大大小（以MB为单位）
MaxBackups：保留旧文件的最大个数
MaxAges：保留旧文件的最大天数
Compress：是否压缩/归档旧文件
*/
lumberJackLogger := &lumberjack.Logger{
Filename: "./logs/info.log",
MaxSize: 10,
MaxBackups: 5,
MaxAge: 30,
Compress: false,
}
return zapcore.AddSync(lumberJackLogger)
}
routers/router.go
package routers
import (
"code/mail_qq/app"
"github.com/gin-gonic/gin"
)
func SetupRouter() *gin.Engine {
r := gin.Default()
//v1
v1Group := r.Group("v1")
{
//待办事项
//添加
v1Group.POST("/send", app.PostMail)
}
return r
}
app/ send.go
发送邮件的代码
package app
import (
"code/mail_qq/log"
"code/mail_qq/setting"
"code/mail_qq/util"
"fmt"
"net/http"
"net/smtp"
"strings"
"github.com/gin-gonic/gin"
)
//var sugarLogger *zap.SugaredLogger
// 定义接收数据的结构体
type User struct {
// binding:"required"修饰的字段，若接收为空值，则报错，是必须字段
Source string `form:"source" json:"source" uri:"source" xml:"source" binding:"required"`
Contacts []string `form:"contacts" json:"contacts" uri:"contacts" xml:"contacts" binding:"required"`
Subject string `form:"subject" json:"subject" uri:"subject" xml:"subject" binding:"required"`
Content string `form:"content" json:"content" uri:"content" xml:"content" binding:"required"`
}
func SendToMail(user, sendUserName, password, host, to, subject, body, mailtype string) error {
hp := strings.Split(host, ":")
//fmt.Println(hp)
auth := smtp.PlainAuth("", user, password, hp[0])
var content_type string
if mailtype == "html" {
content_type = "Content-Type: text/" + mailtype + "; charset=UTF-8"
} else {
content_type = "Content-Type: text/plain" + "; charset=UTF-8"
}
msg := []byte("To: " + to + "\r\nFrom: " + sendUserName + "<" + user + ">" + "\r\nSubject: " + subject + "\r\n" + content_type + "\r\n\r\n" + body)
send_to := strings.Split(to, ";")
err := smtp.SendMail(host, auth, user, send_to, msg)
//fmt.Println(err)
return err
}
func PostMail(c *gin.Context) {
c_ip := util.GetRequestIP(c)
fmt.Println(c_ip)
log.SugarLogger.Debugf("调用 PostMail 接口Api，调用者IP： %s ", c_ip)
声明接收的变量
var json User
将request的body中的数据，自动按照json格式解析到结构体
//
if err := c.ShouldBindJSON(&json); err != nil {
// // 返回错误信息
// // gin.H封装了生成json数据的工具
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
log.SugarLogger.Errorf("Error: %s", err.Error())
return
}
//fmt.Println(json.Content, json.Contacts)
//c.JSON(http.StatusOK, gin.H{"status": &json})
//sugarLogger.Infof("info: %s", json)
sources := json.Source
user, password, host, source := setting.GetMail()
if sources != source {
fmt.Println("Send mail error!,source 认证失败")
log.SugarLogger.Errorf("Send mail error!,source 认证失败")
c.JSON(http.StatusOK, gin.H{
"error": "Send mail error!,source 认证失败",
})
return
}
//println(json.Contacts)
to := json.Contacts
if to[0] == "" {
fmt.Println("Send mail error!,发送人为空")
log.SugarLogger.Errorf("Send mail error!,发送人为空")
c.JSON(http.StatusOK, gin.H{
"error": "Send mail error!,发送人为空",
})
return
}
subject := json.Subject
if strings.TrimSpace(subject) == "" {
fmt.Println("Send mail error!标题为空")
log.SugarLogger.Errorf("Send mail error!标题为空")
c.JSON(http.StatusOK, gin.H{
"error": "Send mail error!,标题为空",
})
return
}
body := `
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="iso-8859-15">
<title>MMOGA POWER</title>
</head>
<body>
` + fmt.Sprintf(json.Content) +
`</body>
</html>`
sendUserName := "告警平台" //发送邮件的人名称
fmt.Println("send email")
for _, s := range to {
//fmt.Println(i, s)
err := SendToMail(user, sendUserName, password, host, s, subject, body, "html")
//log.Printf("接收人：", s+"\n"+"标题:", json.Subject+"\n", "发送内容：", json.Content+"\n")
fmt.Printf("接收人:%s \n 标题: %s \n 内容: %s \n", s, json.Subject, json.Content)
log.SugarLogger.Infof("接收人: %s ,标题: %s, 内容: %s", s, json.Subject, json.Content)
fmt.Println(err)
if err != nil {
fmt.Println("Send mail error!\n")
log.SugarLogger.Errorf("Error 调用者IP: %s ,Send mail error! !", c_ip)
c.JSON(http.StatusOK, gin.H{
"error": "Send mail error! !\n",
})
//fmt.Println(err)
} else {
fmt.Println("Send mail success!\n")
log.SugarLogger.Infof("success 调用者IP: %s ,Send mail success! !", c_ip)
c.JSON(http.StatusOK, gin.H{
"success": "Send mail success! !\n",
})
}
}
}
功能实现
日志记录
