lc-fangdaosha/eventServer/eventServer.go

package eventServer

import (
	"encoding/xml"
	"fmt"
	"github.com/sirupsen/logrus"
	"io/ioutil"
	"net/http"
	"strings"
	"time"
)

// MQDEvent (MQTT Data Event) 用于mqtt传输事件
type MQDEvent struct {
	EventCode string `json:"eventCode"`
	EId       int    `json:"eId"`
	Ip        string `json:"ip"`
	Type      string `json:"type"`
	StartTime string `json:"startTime"`
	Name      string `json:"name"`
	Data      []byte `json:"data"`
}

func StartEventServer() {
	http.HandleFunc("/event", handler)
	logrus.Fatal("事件监听服务启动失败", http.ListenAndServe(":8850", nil))
}

func handler(w http.ResponseWriter, r *http.Request) {
	go func() {
		//监听主机应答固定,直接先应答
		w.WriteHeader(200)
		w.Header().Add("Date", time.Now().String())
		w.Header().Add("Connection", "close")
	}()
	//logRequest(r)
	contentType := r.Header.Get("Content-Type")
	if strings.Contains(contentType, "application/xml") {
		bytes, err := ioutil.ReadAll(r.Body)
		if err != nil {
			logrus.Error("事件处理-读取错误:", err)
			return
		}
		var event EventNotificationAlert
		err = xml.Unmarshal(bytes, &event)
		if err != nil {
			logrus.Error("事件处理-解析错误:", err)
			return
		}
		//处理事件 todo 邮箱
		handleEvent_(event)
	} else if strings.Contains(contentType, "multipart/form-data") {
		fmt.Println("multipart/form-data 事件")
		//handleMultipart(r)
	} else {
		logrus.WithField("Content-Type", contentType).Error("该Content-Type没有写处理逻辑")
		return
	}
}

////TODO 限制事件重复触发,一段时间内不重复发送邮件,但保持报警状态
//
//var eventService = service.ServiceGroupApp.AppServiceGroup.EventService
//
//// 处理多文件事件
//func handleMultipart(r *http.Request) {
//	// todo  远程联动-ip音柱
//	multipartReader := multipart.NewReader(r.Body, "boundary")
//	var msg string //邮件消息
//	// 循环读取每个 part
//	var eventAlert EventNotificationAlert
//	for {
//		part, err := multipartReader.NextPart()
//		//defer part.Close()
//		if err == io.EOF {
//			break
//		}
//		if err != nil {
//			log.Println("Failed to read part:", err)
//			continue
//		}
//		// 检查 part 的 Content-Disposition
//		formName := part.FormName()
//		fmt.Println("formName", formName)
//		if formName != "intrusionImage" {
//			//不含图片的xml部分数据
//			xmlData, err := ioutil.ReadAll(part)
//			if err != nil {
//				return
//			}
//			xml.Unmarshal(xmlData, &eventAlert)
//			//event.Ip = eventAlert.IpAddress
//			//event.Type = eventAlert.EventType
//			//event.StartTime = eventAlert.DateTime
//			msg = handleEvent_(eventAlert)
//			continue
//		}
//		fmt.Printf("eventAlert:%+v\n", eventAlert)
//		//处理图片部分数据
//		contentType := part.Header.Get("Content-Type")
//		eventCode := part.Header.Get("Content-ID")
//		picName := timeFmt(eventAlert.DateTime) + ".jpeg"
//		//event.Name = picName
//		data, _ := ioutil.ReadAll(part)
//		f := &mail.File{
//			Name:     picName,
//			MimeType: contentType,
//			Data:     data,
//			Inline:   true,
//		}
//		picture := &app.Picture{
//			Name: picName,
//			Time: time.Now(),
//			Mime: contentType,
//			Size: len(data),
//		}
//		pictureData := &app.PictureData{
//			Data: data,
//		}
//		event := &app.Event{
//			EventCode:  eventCode,
//			MacAddress: eventAlert.MacAddress,
//			EventType:  eventAlert.EventType,
//		}
//		//保存图片
//		go eventService.Save(event, picture, pictureData)
//		//邮件通知
//		SendAlarmEmail(eventAlert.MacAddress, msg, f)
//		//utils.EmailPicture("1104038181@qq.com",
//		//	"防溺水通知",
//		//	"",
//		//	f)
//	}
//}

func handleEvent_(event EventNotificationAlert) string {
	var eType string
	if event.EventType == "duration" {
		eType = event.DurationList.Duration[0].RelationEvent
	} else {
		eType = event.EventType
	}
	return fmt.Sprintf("事件类型:%s,时间:%s", sMap[eType], timeFmt(event.DateTime))
}

func timeFmt(str string) string {
	s := strings.Split(str, "T")
	t := s[0]
	s1 := strings.Split(s[1], ".")
	s2 := strings.Split(s1[0], "+")
	return t + " " + s2[0]
}

func logRequest(req *http.Request) {
	// 打印请求行
	fmt.Printf("%s %s %s\n", req.Method, req.URL.Path, req.Proto)

	// 打印请求头
	for name, headers := range req.Header {
		for _, h := range headers {
			fmt.Printf("%v: %v\n", name, h)
		}
	}

	// 打印请求体
	body, _ := ioutil.ReadAll(req.Body)
	fmt.Printf("%s\n", body)
}