告警通知-邮件短信公用方法添加

.gitignore

@@ -3,4 +3,4 @@ log
 .idea
 go.sum
 main_demo.go
-/demo/
+/example/

app/operation/controller/alarmController.go

@@ -9,7 +9,7 @@ import (
 	"net/http"
 )
 
-// alarm 告警统计
+// Alarm  告警统计
 var Alarm = new(alarmCtl)
 
 type alarmCtl struct{}

config/config.go

@@ -34,12 +34,13 @@ func Instance() *config {
 }
 
 type config struct {
-	Server   server   `yaml:"server"`
-	Database database `yaml:"database"`
-	Redis    redis    `yaml:"redis"`
-	Logger   logger   `yaml:"logger"`
-	Minio    minio    `yaml:"minio"`
-	Foreign  foreign  `yaml:"foreign"`
+	Server      server      `yaml:"server"`
+	Database    database    `yaml:"database"`
+	Redis       redis       `yaml:"redis"`
+	Logger      logger      `yaml:"logger"`
+	Minio       minio       `yaml:"minio"`
+	Foreign     foreign     `yaml:"foreign"`
+	MonitNotice monitNotice `yaml:"monit-notice"`
 }
 
 type server struct {
@@ -79,3 +80,24 @@ type foreign struct {
 	IotEdgeUrl        string `yaml:"iot_edge_url"`
 	SecurityRewindUrl string `yaml:"security_rewind_url"`
 }
+
+type email struct {
+	Host     string `yaml:"host"`
+	Port     string `yaml:"port"`
+	UserName string `yaml:"user-name"`
+	PassWord string `yaml:"pass-word"`
+}
+type tencent struct {
+	AppID             string `yaml:"app_id"`
+	SecretID          string `yaml:"secret_id"`
+	SecretKey         string `yaml:"secret_key"`
+	SmsKey            string `yaml:"sms_key"`
+	WarnSmsTemplateID string `yaml:"warn_sms_template_id"`
+}
+type sms struct {
+	Tencent tencent `yaml:"tencent"`
+}
+type monitNotice struct {
+	Email email `yaml:"email"`
+	Sms   sms   `yaml:"sms"`
+}

config/config.yaml

@@ -37,4 +37,19 @@ minio:
 #接口
 foreign:
   iot_edge_url: "http://106.52.134.22:8880"
-  security_rewind_url: "http://106.52.134.22:9099"
+  security_rewind_url: "http://106.52.134.22:9099"
+
+#告警通知相关
+monit-notice:
+  email:
+    host: "smtp.163.com"
+    port: "465"
+    user-name: "hn_longchi@163.com"
+    pass-word: "XDTEFFVZVLANOBSU"
+  sms:
+    tencent:
+      app_id: "1400729214"
+      secret_id: "AKID0tkPq1wc9TXMvEHZ5NRFo7AxVotkNltl"
+      secret_key: "g7MNBk36tuhIgplSuAE6lCRVLvFTYR5U"
+      sms_key: "湖南省龙驰照明"
+      warn_sms_template_id: "1533523"

go.mod

@@ -14,6 +14,8 @@ require (
 	github.com/robfig/cron v1.2.0
 	github.com/satori/go.uuid v1.2.0
 	github.com/sirupsen/logrus v1.9.0
+	github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common v1.0.572
+	github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/sms v1.0.572
 	gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df
 	gopkg.in/yaml.v2 v2.4.0
 	gorm.io/driver/mysql v1.4.4

util/notify/email.go

@@ -0,0 +1,55 @@
+package notify
+
+import (
+	"crypto/tls"
+	"gopkg.in/gomail.v2"
+	"iot_manager_service/config"
+	"strconv"
+)
+
+func SendEmail(html string, title string, receiver ...string) error {
+	message := html
+	config := config.Instance().MonitNotice.Email
+	// QQ 邮箱：
+	// SMTP 服务器地址：smtp.qq.com（SSL协议端口：465/994 | 非SSL协议端口：25）
+	// 163 邮箱：
+	// SMTP 服务器地址：smtp.163.com（端口：25）
+	host := config.Host
+	port := config.Port
+	userName := config.UserName
+	password := config.PassWord
+
+	m := gomail.NewMessage()
+	m.SetHeader("From", userName) // 发件人
+	// m.SetHeader("From", "alias"+"<"+userName+">") // 增加发件人别名
+
+	m.SetHeader("To", receiver...) // 收件人，可以多个收件人，但必须使用相同的 SMTP 连接
+	//m.SetHeader("Cc", "******@qq.com")                  // 抄送，可以多个
+	//m.SetHeader("Bcc", "******@qq.com")                 // 暗送，可以多个
+	m.SetHeader("Subject", title) // 邮件主题
+
+	// text/html 的意思是将文件的 content-type 设置为 text/html 的形式，浏览器在获取到这种文件时会自动调用html的解析器对文件进行相应的处理。
+	// 可以通过 text/html 处理文本格式进行特殊处理，如换行、缩进、加粗等等
+	//m.SetBody("text/html", fmt.Sprintf(message, "testUser"))
+	m.SetBody("text/html", message)
+
+	// text/plain的意思是将文件设置为纯文本的形式，浏览器在获取到这种文件时并不会对其进行处理
+	// m.SetBody("text/plain", "纯文本")
+	// m.Attach("test.sh")   // 附件文件，可以是文件，照片，视频等等
+	// m.Attach("lolcatVideo.mp4") // 视频
+	// m.Attach("lolcat.jpg") // 照片
+	portInt, _ := strconv.Atoi(port)
+	d := gomail.NewDialer(
+		host,
+		portInt,
+		userName,
+		password,
+	)
+	// 关闭SSL协议认证
+	d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
+
+	if err := d.DialAndSend(m); err != nil {
+		return err
+	}
+	return nil
+}

util/notify/tencent_sms.go

@@ -0,0 +1,129 @@
+package notify
+
+import (
+	"encoding/json"
+	errors2 "errors"
+	"fmt"
+	"iot_manager_service/config"
+	"strings"
+
+	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
+	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
+	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
+	sms "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/sms/v20210111" // 引入sms
+)
+
+// SendTencentSms 发送腾讯云短信通知
+// @param receiverPhoneNumbers 接收的手机号列表
+// @param smsTemplateId 短信模板ID
+// @param values 短信模板参数
+func SendTencentSms(receiverPhoneNumbers []string, smsTemplateId string, values []string) error {
+	mainConfig := config.Instance().MonitNotice.Sms.Tencent
+	/* 必要步骤：
+	 * 实例化一个认证对象，入参需要传入腾讯云账户密钥对secretId，secretKey。
+	 * 这里采用的是从环境变量读取的方式，需要在环境变量中先设置这两个值。
+	 * 你也可以直接在代码中写死密钥对，但是小心不要将代码复制、上传或者分享给他人，
+	 * 以免泄露密钥对危及你的财产安全。
+	 * SecretId、SecretKey 查询: https://console.cloud.tencent.com/cam/capi */
+	credential := common.NewCredential(
+		mainConfig.SecretID,
+		mainConfig.SecretKey,
+	)
+	/* 非必要步骤:
+	 * 实例化一个客户端配置对象，可以指定超时时间等配置 */
+	cpf := profile.NewClientProfile()
+
+	/* SDK默认使用POST方法。
+	 * 如果你一定要使用GET方法，可以在这里设置。GET方法无法处理一些较大的请求 */
+	cpf.HttpProfile.ReqMethod = "POST"
+
+	/* SDK有默认的超时时间，非必要请不要进行调整
+	 * 如有需要请在代码中查阅以获取最新的默认值 */
+	// cpf.HttpProfile.ReqTimeout = 5
+
+	/* 指定接入地域域名，默认就近地域接入域名为 sms.tencentcloudapi.com ，也支持指定地域域名访问，例如广州地域的域名为 sms.ap-guangzhou.tencentcloudapi.com */
+	cpf.HttpProfile.Endpoint = "sms.tencentcloudapi.com"
+
+	/* SDK默认用TC3-HMAC-SHA256进行签名，非必要请不要修改这个字段 */
+	cpf.SignMethod = "HmacSHA1"
+
+	/* 实例化要请求产品(以sms为例)的client对象
+	 * 第二个参数是地域信息，可以直接填写字符串ap-guangzhou，支持的地域列表参考 https://cloud.tencent.com/document/api/382/52071#.E5.9C.B0.E5.9F.9F.E5.88.97.E8.A1.A8 */
+	client, _ := sms.NewClient(credential, "ap-guangzhou", cpf)
+
+	/* 实例化一个请求对象，根据调用的接口和实际情况，可以进一步设置请求参数
+	 * 你可以直接查询SDK源码确定接口有哪些属性可以设置
+	 * 属性可能是基本类型，也可能引用了另一个数据结构
+	 * 推荐使用IDE进行开发，可以方便的跳转查阅各个接口和数据结构的文档说明 */
+	request := sms.NewSendSmsRequest()
+
+	/* 基本类型的设置:
+	 * SDK采用的是指针风格指定参数，即使对于基本类型你也需要用指针来对参数赋值。
+	 * SDK提供对基本类型的指针引用封装函数
+	 * 帮助链接：
+	 * 短信控制台: https://console.cloud.tencent.com/smsv2
+	 * 腾讯云短信小助手: https://cloud.tencent.com/document/product/382/3773#.E6.8A.80.E6.9C.AF.E4.BA.A4.E6.B5.81 */
+
+	/* 短信应用ID: 短信SdkAppId在 [短信控制台] 添加应用后生成的实际SdkAppId，示例如1400006666 */
+	// 应用 ID 可前往 [短信控制台](https://console.cloud.tencent.com/smsv2/app-manage) 查看
+	request.SmsSdkAppId = common.StringPtr(mainConfig.AppID)
+
+	/* 短信签名内容: 使用 UTF-8 编码，必须填写已审核通过的签名 */
+	// 签名信息可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-sign) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-sign) 的签名管理查看
+	request.SignName = common.StringPtr(mainConfig.SmsKey)
+
+	/* 模板 ID: 必须填写已审核通过的模板 ID */
+	// 模板 ID 可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-template) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-template) 的正文模板管理查看
+	request.TemplateId = common.StringPtr(smsTemplateId)
+
+	/* 模板参数: 模板参数的个数需要与 TemplateId 对应模板的变量个数保持一致，若无模板参数，则设置为空*/
+	request.TemplateParamSet = common.StringPtrs(values)
+
+	/* 下发手机号码，采用 E.164 标准，+[国家或地区码][手机号]
+	 * 示例如：+8613711112222， 其中前面有一个+号 ，86为国家码，13711112222为手机号，最多不要超过200个手机号*/
+	request.PhoneNumberSet = common.StringPtrs(receiverPhoneNumbers)
+
+	/* 用户的 session 内容（无需要可忽略）: 可以携带用户侧 ID 等上下文信息，server 会原样返回 */
+	request.SessionContext = common.StringPtr("")
+
+	/* 短信码号扩展号（无需要可忽略）: 默认未开通，如需开通请联系 [腾讯云短信小助手] */
+	request.ExtendCode = common.StringPtr("")
+
+	/* 国际/港澳台短信 SenderId（无需要可忽略）: 国内短信填空，默认未开通，如需开通请联系 [腾讯云短信小助手] */
+	request.SenderId = common.StringPtr("")
+
+	// 通过client对象调用想要访问的接口，需要传入请求对象
+	response, err := client.SendSms(request)
+	// 处理异常
+	if _, ok := err.(*errors.TencentCloudSDKError); ok {
+		fmt.Printf("An API error has returned: %s", err)
+		return err
+	}
+	// 非SDK异常，直接失败。实际代码中可以加入其他的处理。
+	if err != nil {
+		return err
+	}
+
+	b, _ := json.Marshal(response.Response)
+
+	//if strings.Contains(string(b), "insufficient balance in SMS package") {
+	//	return errors2.New("短信系统没余额了")
+	//}
+	//if strings.Contains(string(b), "the number of sms messages sent from a single mobile number every day exceeds the upper limit") {
+	//	return errors2.New("单个手机号超过上限")
+	//}
+
+	if !strings.Contains(string(b), "OK") {
+		return errors2.New(string(b))
+	}
+	// 打印返回的json字符串
+
+	return nil
+	/* 当出现以下错误码时，快速解决方案参考
+	 * [FailedOperation.SignatureIncorrectOrUnapproved](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Afailedoperation.signatureincorrectorunapproved-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
+	 * [FailedOperation.TemplateIncorrectOrUnapproved](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Afailedoperation.templateincorrectorunapproved-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
+	 * [UnauthorizedOperation.SmsSdkAppIdVerifyFail](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Aunauthorizedoperation.smssdkappidverifyfail-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
+	 * [UnsupportedOperation.ContainDomesticAndInternationalPhoneNumber](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Aunsupportedoperation.containdomesticandinternationalphonenumber-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
+	 * 更多错误，可咨询[腾讯云助手](https://tccc.qcloud.com/web/im/index.html#/chat?webAppId=8fa15978f85cb41f7e2ea36920cb3ae1&title=Sms)
+	 */
+}