BBIT-Kai
164 lines
4.4 KiB
Go
164 lines
4.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"sentinel/pkg/config"
|
|
"sentinel/pkg/device"
|
|
"sentinel/pkg/docker"
|
|
"sentinel/pkg/log"
|
|
model2 "sentinel/pkg/model"
|
|
"sentinel/pkg/utils"
|
|
"time"
|
|
)
|
|
|
|
type BusinessService struct {
|
|
mqtt *MQTTService
|
|
dockerManager docker.DockerManager
|
|
deviceID string
|
|
deptId string
|
|
cmdTopic string
|
|
deviceType string
|
|
}
|
|
|
|
func NewBusinessService(m *MQTTService, deviceID string, dm docker.DockerManager) *BusinessService {
|
|
return &BusinessService{
|
|
mqtt: m,
|
|
dockerManager: dm,
|
|
deviceID: deviceID,
|
|
}
|
|
}
|
|
|
|
func (b *BusinessService) Start() error {
|
|
if err := b.mqtt.SubscribeTopic(getInitTopic(b.deviceID), 1); err != nil {
|
|
return err
|
|
}
|
|
|
|
b.mqtt.SetMessageHandler(b.onMQTTMessage)
|
|
|
|
return nil
|
|
}
|
|
|
|
func getInitTopic(deviceID string) string {
|
|
return "+/+/+/" + deviceID + "/#"
|
|
}
|
|
|
|
func (b *BusinessService) getOwnTopic(deviceID string) string {
|
|
return b.deptId + "/cmd/" + b.deviceType + "/" + deviceID + "/#"
|
|
}
|
|
|
|
// 消息处理
|
|
func (b *BusinessService) onMQTTMessage(topic string, payload []byte) {
|
|
model := model2.FromStringToMqttTopic(topic)
|
|
payload_json, err := utils.PayloadToMap(payload)
|
|
if err != nil {
|
|
fmt.Println("解析失败:", err)
|
|
return
|
|
}
|
|
|
|
// 指令
|
|
if model.Domain == "cmd" && model.DeviceType == b.deviceType {
|
|
log.Println("收到指令：", model.Resource)
|
|
switch model.Resource {
|
|
case "shutdown":
|
|
payload_json["massage"] = b.handleShutdown()
|
|
case "restart":
|
|
payload_json["massage"] = b.handleRestart()
|
|
case "check_update":
|
|
payload_json["massage"] = b.handleCheckUpdate()
|
|
default:
|
|
log.Println("未知的命令：", model.Resource)
|
|
payload_json["massage"] = "未知的命令"
|
|
}
|
|
topic := "x/receipt/x/" + device.GetDeviceID() + "/info"
|
|
payload := utils.JSONToString(payload_json)
|
|
b.mqtt.PublicMsg(topic, payload, false) // 回执
|
|
} else if model.Domain == "status" && model.Resource == "receipt" {
|
|
b.deviceType = model.DeviceType
|
|
b.deptId = model.DeptId
|
|
// 取消订阅之前的初始化主题
|
|
if b.mqtt.UnsubscribeTopic(getInitTopic(b.deviceID)) != nil {
|
|
log.Error("无法取消初始化主题")
|
|
return
|
|
}
|
|
// 新订阅属于自己的主题
|
|
if b.mqtt.SubscribeTopic(b.getOwnTopic(b.deviceID), 1) != nil {
|
|
log.Error("无法定于属于自己的主题")
|
|
return
|
|
}
|
|
log.Println("设备初始化成功：所属项目：", model.DeptId, "\t设备类型：", model.DeviceType)
|
|
}
|
|
}
|
|
|
|
func getStatusInfoTopicPayload(isOnline bool) (string, string) {
|
|
info := map[string]interface{}{
|
|
"version": config.APP_VERSION,
|
|
"online": isOnline,
|
|
"ip": utils.GetLocalIP(),
|
|
"hostname": utils.GetHostname(),
|
|
"mac": utils.GetMacAddress(),
|
|
"os": utils.GetOSInfo(),
|
|
"cpu": utils.GetCPUInfo(),
|
|
"memory_total": utils.GetMemory(),
|
|
"disk_total": utils.GetDisk(),
|
|
"last_seen": time.Now().UTC().Format(time.RFC3339),
|
|
}
|
|
payloadBytes, _ := json.Marshal(info)
|
|
payload := string(payloadBytes) // 转成 string
|
|
return "x/status/x/" + device.GetDeviceID() + "/info", payload
|
|
}
|
|
|
|
// 关闭程序（立即退出）
|
|
func (b *BusinessService) handleShutdown() string {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
defer cancel()
|
|
b.dockerManager.StopAndRemoveContainer(ctx)
|
|
return "程序已退出"
|
|
}
|
|
|
|
// 重启程序
|
|
func (b *BusinessService) handleRestart() string {
|
|
log.Println("正在拉取镜像：", config.DOCKER_IMAGE)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
defer cancel()
|
|
if err := b.dockerManager.PullImage(ctx, config.DOCKER_IMAGE); err != nil {
|
|
log.Fatalf("镜像拉取失败：%v", err)
|
|
}
|
|
|
|
err := b.dockerManager.RunContainer(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return "程序已重启"
|
|
}
|
|
|
|
func (b *BusinessService) handleCheckUpdate() string {
|
|
cfg := config.LoadConfig()
|
|
cfg.NeedUpdate = true
|
|
if err := config.WriteLocalConfig(&cfg); err != nil {
|
|
log.Println("load config failed:", err)
|
|
}
|
|
return "程序更新机制已触发"
|
|
}
|
|
|
|
//func (b *BusinessService) handleCheckUpdate() {
|
|
// args := []string{
|
|
// "--version", strconv.Itoa(config.APP_VERSION),
|
|
// }
|
|
//
|
|
// launcher := newUpdaterLauncher()
|
|
//
|
|
// if err := launcher.Start(args); err != nil {
|
|
// log.Println("[BUS] failed to start updater:", err)
|
|
// return
|
|
// }
|
|
//
|
|
// log.Println(
|
|
// "[BUS] updater started, exiting main program",
|
|
// )
|
|
//
|
|
// time.Sleep(500 * time.Millisecond)
|
|
// os.Exit(0)
|
|
//}
|