ERP_user/Nats/Nats.go

package Nats

import (
	"ERP_user/conf"
	"ERP_user/models/Account"
	"ERP_user/models/System"
	"fmt"
	powerlibs "git.baozhida.cn/ERP_libs/Power"
	"github.com/nats-io/nats.go"
	"github.com/vmihailenco/msgpack/v5"
)

var Nats *nats.Conn

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

	go NatsInit()

}

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

		type T_User struct {
			T_uuid  string `xml:"T_uuid"`
			T_power string `xml:"T_power"`
			T_name  string `xml:"T_name"`
			T_user  string `xml:"T_user"`
		}
		type T_R struct {
			Code int16  `xml:"Code"`
			Msg  string `xml:"Msg"`
			Data T_User `xml:"Data"` // 泛型
		}

		var t_R T_R

		// 验证登录
		b_, admin_r := Account.Verification(string(m.Data), "")
		if !b_ {
			t_R.Code = 201
			t_R.Msg = "请重新登陆!"
			b, _ := msgpack.Marshal(&t_R)
			_ = Nats.Publish(m.Reply, b)
			return
		}

		t_R.Code = 200
		t_R.Msg = "ok"
		t_R.Data.T_uuid = admin_r.T_uuid
		t_R.Data.T_power = admin_r.T_power
		t_R.Data.T_name = admin_r.T_name
		t_R.Data.T_user = admin_r.T_user

		b, _ := msgpack.Marshal(&t_R)
		_ = Nats.Publish(m.Reply, b)
	})
	// 请求-响应 获取所有用户列表
	_, _ = Nats.Subscribe("ERP_User_Read_User_List_All", func(m *nats.Msg) {

		type T_R struct {
			Code int16            `xml:"Code"`
			Msg  string           `xml:"Msg"`
			Data []Account.User_R `xml:"Data"` // 泛型
		}

		var t_R T_R
		t_R.Code = 200
		t_R.Msg = "ok"
		t_R.Data = Account.Read_User_List_All()

		b, _ := msgpack.Marshal(&t_R)
		_ = Nats.Publish(m.Reply, b)
	})
	// 请求-响应 获取所有权限列表
	_, _ = Nats.Subscribe("ERP_User_Read_Power_List_All", func(m *nats.Msg) {

		type T_R struct {
			Code  int16               `xml:"Code"`
			Msg   string              `xml:"Msg"`
			Count int64               `xml:"Count"`
			Data  []powerlibs.Power_R `xml:"Data"` // 泛型
		}

		var t_R T_R
		t_R.Code = 200
		t_R.Msg = "ok"
		t_R.Data, t_R.Count = Account.Read_Power_List("", 0, 9999)

		b, _ := msgpack.Marshal(&t_R)
		_ = Nats.Publish(m.Reply, b)
	})
	// 发布-订阅 模式，异步订阅 系统日志
	_, _ = Nats.Subscribe("ERP_AddSysLogs", func(m *nats.Msg) {
		fmt.Printf("ERP_AddSysLogs message: %v\n", string(m.Data))

		type T_S struct {
			T_class string
			T_title string
			T_txt   string
		}

		var t_S T_S

		err := msgpack.Unmarshal(m.Data, &t_S)
		if err != nil {

			System.Add_SysLogs("Nats", "Nats AddSysLogs 解析失败", string(m.Data))
			return
		}

		System.Add_SysLogs(t_S.T_class, t_S.T_title, t_S.T_txt)

	})
	// 发布-订阅 模式，异步订阅 用户日志
	_, _ = Nats.Subscribe("ERP_AddUserLogs", func(m *nats.Msg) {
		fmt.Printf("ERP_AddUserLogs message: %v\n", string(m.Data))

		type T_S struct {
			T_uuid  string
			T_class string
			T_title string
			T_txt   string
		}

		var t_S T_S

		err := msgpack.Unmarshal(m.Data, &t_S)
		if err != nil {
			System.Add_SysLogs("Nats", "Nats AddUserLogs 解析失败", string(m.Data))
			return
		}

		System.Add_UserLogs(t_S.T_uuid, t_S.T_class, t_S.T_title, t_S.T_txt)

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

		type T_S struct {
			T_uuid  string
			T_Title string
			T_Url   string
		}

		var t_S T_S

		err := msgpack.Unmarshal(m.Data, &t_S)
		if err != nil {
			System.Add_SysLogs("Nats", "Nats AddNews 解析失败", string(m.Data))
			return
		}

		_, err = System.Add_News(System.News{T_uuid: t_S.T_uuid, T_Title: t_S.T_Title, T_Url: t_S.T_Url})
		if err != nil {
			System.Add_SysLogs("Nats", "Nats AddNews 添加失败", string(m.Data))
			return
		}
	})
}