Cold_Api/Nats/Nats.go

package Nats

import (
	"Cold_Api/conf"
	"Cold_Api/controllers/lib"
	"Cold_Api/logs"
	"Cold_Api/models/Device"
	"encoding/xml"
	"fmt"
	"github.com/astaxie/beego/cache"
	"github.com/nats-io/nats.go"
	"strconv"
	"strings"
	"time"
)

var redisCache_NatsServer cache.Cache

func init() {

	logs.Println("============Nats init============")
	var err error
	// 连接Nats服务器
	lib.Nats, err = nats.Connect("nats://" + conf.NatsServer_Url)
	if err != nil {
		fmt.Println("nats 连接失败！")
		panic(err)
	}
	logs.Println("nats OK！")

	go NatsInit()

	config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
		"redisCache_NatsServer", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
	fmt.Println(config)
	redisCache_NatsServer, err = cache.NewCache("redis", config)
	if err != nil || redisCache_NatsServer == nil {
		errMsg := "failed to init redis"
		fmt.Println(errMsg, err)
		panic(errMsg)
	}

}

func NatsInit() {

	// 请求-响应， 响应 test3 消息。
	_, _ = lib.Nats.Subscribe("Wx_BasicMessage_Event_QRCode", func(m *nats.Msg) {
		fmt.Printf(" => Nats   Wx_BasicMessage_Event_QRCode message: %s\n", string(m.Data))

		type Person_QRCode struct {
			ToUserName   string `xml:"ToUserName"`   //注意这里有个反引号
			FromUserName string `xml:"FromUserName"` //注意这里有个反引号
			CreateTime   string `xml:"CreateTime"`   //注意这里有个反引号
			EventKey     string `xml:"EventKey"`
		}
		var person_QRCode Person_QRCode
		err1 := xml.Unmarshal(m.Data, &person_QRCode)
		if err1 != nil {
			fmt.Println("Unmarshal error")
			_ = lib.Nats.Publish(m.Reply, []byte(""))
			return
		}
		// 进入 二维码配对
		fmt.Println("FromUserName-", person_QRCode.FromUserName)
		fmt.Println("EventKey-", person_QRCode.EventKey)

		// 开始  处理消息
		if strings.Contains(person_QRCode.EventKey, "@宝智达 微信公众号通知") {
			//"请将本内容发送到 深圳市宝智达科技有限公司 微信公众号-|"+lib.AesEncryptCBC(T_calss_id, "0123456789012345")+"|-",
			Content_r := lib.GetBetweenStr(person_QRCode.EventKey, "-|", "|-")
			decryptCode := lib.AesDecryptCBC(Content_r, "0123456789012345")
			decryptCode_int, err := strconv.Atoi(decryptCode)
			fmt.Println("解密结果：", decryptCode, "  decryptCode_int", decryptCode_int, "  Content_r:", Content_r)
			R_DeviceClass, err := Device.Read_Class_ById(decryptCode_int)
			if err != nil {
				_ = lib.Nats.Publish(m.Reply, []byte(""))
				return
			}

			if strings.Contains(R_DeviceClass.T_Notice_wx, person_QRCode.FromUserName) {

				_ = lib.Nats.Publish(m.Reply, []byte(R_DeviceClass.T_name+" 已绑定，无需重复绑定！"))
				return
			}

			R_DeviceClass.T_Notice_wx = R_DeviceClass.T_Notice_wx + person_QRCode.FromUserName + "/重令名|"
			Device.Update_Class_ById(R_DeviceClass)

			_ = lib.Nats.Publish(m.Reply, []byte(R_DeviceClass.T_name+" 绑定成功！"))
			return
		}
		_ = lib.Nats.Publish(m.Reply, []byte(""))
	})

	// 请求-响应， 响应 test3 消息。
	_, _ = lib.Nats.Subscribe("Wx2_BasicMessage_Event_QRCode", func(m *nats.Msg) {
		fmt.Printf(" => Nats   Wx2_BasicMessage_Event_QRCode message: %s\n", string(m.Data))

		type Person_QRCode struct {
			ToUserName   string `xml:"ToUserName"`   //注意这里有个反引号
			FromUserName string `xml:"FromUserName"` //注意这里有个反引号
			CreateTime   string `xml:"CreateTime"`   //注意这里有个反引号
			EventKey     string `xml:"EventKey"`
		}
		var person_QRCode Person_QRCode
		err1 := xml.Unmarshal(m.Data, &person_QRCode)
		if err1 != nil {
			fmt.Println("Unmarshal error")
			_ = lib.Nats.Publish(m.Reply, []byte(""))
			return
		}
		// 进入 二维码配对
		fmt.Println("FromUserName-", person_QRCode.FromUserName)
		fmt.Println("EventKey-", person_QRCode.EventKey)

		// 开始  处理消息
		if strings.Contains(person_QRCode.EventKey, "@宝智达 微信公众号通知") {
			//"请将本内容发送到 深圳市宝智达科技有限公司 微信公众号-|"+lib.AesEncryptCBC(T_calss_id, "0123456789012345")+"|-",
			Content_r := lib.GetBetweenStr(person_QRCode.EventKey, "-|", "|-")
			decryptCode := lib.AesDecryptCBC(Content_r, "0123456789012345")
			decryptCode_int, err := strconv.Atoi(decryptCode)
			fmt.Println("解密结果：", decryptCode, "  decryptCode_int", decryptCode_int, "  Content_r:", Content_r)
			R_DeviceClass, err := Device.Read_Class_ById(decryptCode_int)
			if err != nil {
				_ = lib.Nats.Publish(m.Reply, []byte(""))
				return
			}

			if strings.Contains(R_DeviceClass.T_Notice_wx2, person_QRCode.FromUserName+"/重令名|") {
				_ = lib.Nats.Publish(m.Reply, []byte(R_DeviceClass.T_name+" 重复扫码！请在 5分钟内 回复您的名字，否则将 无法收到报警消息！如回复：张三"))
				return
			}
			if strings.Contains(R_DeviceClass.T_Notice_wx2, person_QRCode.FromUserName) {
				_ = lib.Nats.Publish(m.Reply, []byte(R_DeviceClass.T_name+" 已绑定，无需重复绑定！"))
				return
			}

			R_DeviceClass.T_Notice_wx2 = R_DeviceClass.T_Notice_wx2 + person_QRCode.FromUserName + "/重令名|"
			Device.Update_Class_ById(R_DeviceClass)
			redisCache_NatsServer.Put(person_QRCode.FromUserName, decryptCode_int, 5*time.Minute)
			_ = lib.Nats.Publish(m.Reply, []byte(R_DeviceClass.T_name+" 扫码成功！请在 5分钟内 回复您的名字，否则将 无法收到报警消息！如回复：张三"))
			return
		}
		_ = lib.Nats.Publish(m.Reply, []byte(""))
	})

	// 请求-响应， 响应 test3 消息。
	_, _ = lib.Nats.Subscribe("Wx2_BasicMessage_Text", func(m *nats.Msg) {
		logs.Println(" => Nats   Wx2_BasicMessage_Text message: %s\n", string(m.Data))

		type Person_Text struct {
			ToUserName   string `xml:"ToUserName"`   //注意这里有个反引号
			FromUserName string `xml:"FromUserName"` //注意这里有个反引号
			CreateTime   string `xml:"CreateTime"`   //注意这里有个反引号
			MsgType      string `xml:"MsgType"`
			Content      string `xml:"Content"`
		}

		var person_Text Person_Text
		err1 := xml.Unmarshal(m.Data, &person_Text)
		if err1 != nil {
			fmt.Println("Unmarshal error")
			_ = lib.Nats.Publish(m.Reply, []byte(""))
			return
		}
		// 进入 二维码配对
		fmt.Println("FromUserName-", person_Text.FromUserName)
		fmt.Println("Content-", person_Text.Content)

		if len(person_Text.Content) > 8*3 || len(person_Text.Content) <= 1*3 {
			_ = lib.Nats.Publish(m.Reply, []byte("请正确输入您的名字！"))
			return
		}

		if !redisCache_NatsServer.IsExist(person_Text.FromUserName) {
			return
		}
		Class_ById := lib.To_int(redisCache_NatsServer.Get(person_Text.FromUserName))
		R_DeviceClass, err := Device.Read_Class_ById(Class_ById)
		if err != nil {
			_ = lib.Nats.Publish(m.Reply, []byte(""))
			return
		}

		R_DeviceClass.T_Notice_wx2 = strings.Replace(R_DeviceClass.T_Notice_wx2,
			person_Text.FromUserName+"/重令名|",
			person_Text.FromUserName+"/"+person_Text.Content+"|",
			-1)
		Device.Update_Class_ById(R_DeviceClass)

		// 删除 缓存
		redisCache_NatsServer.Delete(person_Text.FromUserName)

		_ = lib.Nats.Publish(m.Reply, []byte("尊敬的 "+person_Text.Content+",您以成功绑定 "+R_DeviceClass.T_name))
	})

	//// 发布-订阅 模式，异步订阅 test1
	//_, _ = Nats.Subscribe("test1", func(m *nats.Msg) {
	//	fmt.Printf("Received a message: %s\n", string(m.Data))
	//})

	//// 队列 模式，订阅 test2， 队列为queue, test2 发向所有队列，同一队列只有一个能收到消息
	//_, _ = Nats.QueueSubscribe("test2", "queue", func(msg *nats.Msg) {
	//	fmt.Printf("Queue a message: %s\n", string(msg.Data))
	//})

	//// 请求-响应， 响应 test3 消息。
	//_, _ = Nats.Subscribe("test3", func(m *nats.Msg) {
	//	fmt.Printf("Reply a message: %s\n", string(m.Data))
	//	_ = Nats.Publish(m.Reply, []byte("I can help!!"))
	//})

}