增加对语音播报的支持“

Signed-off-by: jikun.zhang <[email protected]>

Author: jikun.zhang

PrometheusAlertVoice/PrometheusAlertVoice.exe

@@ -0,0 +1,4 @@
+[SERVER]
+PORT=9999
+#设置语速的快慢，该参数的范围是从-10到10之间
+SPEED=1

PrometheusAlertVoice/setup.ini

@@ -249,3 +249,10 @@ BARK_COPY=1
 BARK_ARCHIVE=1
 # 消息分组
 BARK_GROUP=PrometheusAlert
+
+#---------------------↓语音播报-----------------------
+#语音播报需要配合语音播报插件才能使用
+#是否开启语音播报通道,0为关闭,1为开启
+open-voice=1
+VOICE_IP=127.0.0.1
+VOICE_PORT=9999

conf/app-example.conf

@@ -79,6 +79,10 @@ func (c *MainController) AlertTest() {
 		TgMessage := "PrometheusAlertCenter测试告警"
 		ret := SendBark(TgMessage, logsign)
 		c.Data["json"] = ret
+	case "voice":
+		vMessage := "Prometheus Alert Center 测试告警"
+		ret := SendVoice(vMessage, logsign)
+		c.Data["json"] = ret
 	default:
 		c.Data["json"] = "hahaha!"
 	}

controllers/WebTest.go

@@ -24,6 +24,7 @@ type DashboardJson struct {
 	Webhook         int `json:"webhook"`
 	Weixin          int `json:"weixin"`
 	Workwechat      int `json:"workwechat"`
+	Voice           int `json:"voice"`
 	Zabbix          int `json:"zabbix"`
 	Grafana         int `json:"grafana"`
 	Graylog         int `json:"graylog"`

controllers/default.go

@@ -325,6 +325,7 @@ func SendMessageR(message Prometheus, rwxurl, rddurl, rfsurl, rphone, remail, rg
 		SendTG(PhoneCallMessage, logsign)
 		// 发送消息到Bark
 		SendBark(PhoneCallMessage, logsign)
+		SendVoice(PhoneCallMessage, logsign)
 		// 推送消息到企业微信
 		SendWorkWechat(beego.AppConfig.String("WorkWechat_ToUser"), beego.AppConfig.String("WorkWechat_ToParty"), beego.AppConfig.String("WorkWechat_ToTag"), wxtext, logsign)
 

controllers/prometheus.go

@@ -514,6 +514,9 @@ func SendMessagePrometheusAlert(message string, pmsg *PrometheusAlertMsg, logsig
 	// Bark
 	case "bark":
 		ReturnMsg += SendBark(message, logsign)
+	// Bark
+	case "voice":
+		ReturnMsg += SendVoice(message, logsign)
 	//异常参数
 	default:
 		ReturnMsg = "参数错误"

controllers/prometheusalert.go

@@ -0,0 +1,40 @@
+package controllers
+
+import (
+	"PrometheusAlert/models"
+	"github.com/astaxie/beego"
+	"github.com/astaxie/beego/logs"
+	"net"
+)
+
+// SendVoice
+func SendVoice(message, logsign string) string {
+	open := beego.AppConfig.String("open-voice")
+	if open != "1" {
+		logs.Info(logsign, "[voicemessage]", "语音播报配置未开启,请先配置open-voice为1")
+		return "语音播报配置未开启,请先配置open-voice为1"
+	}
+	v_ip := beego.AppConfig.String("VOICE_IP")
+	v_port := beego.AppConfig.String("VOICE_PORT")
+	//发送tcp语音消息文本
+	v_addr, err := net.ResolveTCPAddr("tcp", v_ip+":"+v_port)
+	if err != nil {
+		logs.Error(logsign, "[voicemessage]", err.Error())
+		return "语音组件连接初始化失败：" + err.Error()
+	}
+	conn, err := net.DialTCP("tcp", nil, v_addr)
+	if err != nil {
+		logs.Error(logsign, "[voicemessage]", err.Error())
+		return "语音组件连接失败：" + err.Error()
+	}
+	_, err = conn.Write([]byte(message))
+	if err != nil {
+		logs.Error(logsign, "[voicemessage]", err.Error())
+		return "语音组件发送消息失败：" + err.Error()
+	}
+	logs.Info(logsign, "[voicemessage]", message+"  语音播报消息发送成功")
+	conn.Close()
+	models.AlertToCounter.WithLabelValues("voice").Add(1)
+	ChartsJson.Voice += 1
+	return message + "  语音播报消息发送成功"
+}