133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package main
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"os/exec"
|
||
"runtime"
|
||
"time"
|
||
|
||
mqtt "github.com/eclipse/paho.mqtt.golang"
|
||
"github.com/joho/godotenv"
|
||
"github.com/shirou/gopsutil/cpu"
|
||
"github.com/shirou/gopsutil/mem"
|
||
)
|
||
|
||
// PCInfo 定义 PC 使用信息的结构。
|
||
type PCInfo struct {
|
||
CPUUsage float64 `json:"cpu_usage"` // CPU
|
||
MemUsage float64 `json:"mem_usage"` // Memory
|
||
}
|
||
|
||
// Command 定义接收命令的结构。
|
||
type Command struct {
|
||
Action string `json:"action"` // Action 待执行
|
||
}
|
||
|
||
// loadEnv 加载 .env 文件以读取环境变量。
|
||
func loadEnv() {
|
||
err := godotenv.Load(".env")
|
||
if err != nil {
|
||
fmt.Println("Error loading .env file")
|
||
}
|
||
}
|
||
|
||
// getPCInfo 检索当前 PC 使用信息。
|
||
func getPCInfo() (PCInfo, error) {
|
||
cpuPercentages, err := cpu.Percent(0, false)
|
||
if err != nil {
|
||
return PCInfo{}, err
|
||
}
|
||
|
||
memInfo, err := mem.VirtualMemory()
|
||
if err != nil {
|
||
return PCInfo{}, err
|
||
}
|
||
|
||
// 保留2位小数
|
||
return PCInfo{
|
||
CPUUsage: float64(int(cpuPercentages[0]*100)) / 100,
|
||
MemUsage: float64(int(memInfo.UsedPercent*100)) / 100,
|
||
}, nil
|
||
}
|
||
|
||
// onMessageReceived 处理从 MQTT 接收的消息。
|
||
func onMessageReceived(client mqtt.Client, msg mqtt.Message) {
|
||
var cmd Command
|
||
err := json.Unmarshal(msg.Payload(), &cmd)
|
||
if err != nil {
|
||
fmt.Println("Error decoding JSON command:", err)
|
||
return
|
||
}
|
||
|
||
if cmd.Action == "shutdown" {
|
||
fmt.Println("Shutdown command received. Shutting down...")
|
||
executeShutdown()
|
||
}
|
||
}
|
||
|
||
// executeShutdown 根据操作系统执行关机命令。
|
||
func executeShutdown() {
|
||
var cmd *exec.Cmd
|
||
switch runtime.GOOS {
|
||
case "windows":
|
||
cmd = exec.Command("shutdown", "/s", "/t", "0")
|
||
case "linux", "darwin":
|
||
cmd = exec.Command("shutdown", "-h", "now")
|
||
default:
|
||
fmt.Println("Shutdown not supported on this OS")
|
||
return
|
||
}
|
||
err := cmd.Run()
|
||
if err != nil {
|
||
fmt.Println("Failed to execute shutdown:", err)
|
||
}
|
||
}
|
||
|
||
func main() {
|
||
loadEnv()
|
||
|
||
broker := os.Getenv("MQTT_BROKER")
|
||
clientID := os.Getenv("MQTT_CLIENT_ID")
|
||
topic := os.Getenv("MQTT_TOPIC")
|
||
|
||
opts := mqtt.NewClientOptions()
|
||
opts.AddBroker(broker)
|
||
opts.SetClientID(clientID)
|
||
opts.SetDefaultPublishHandler(onMessageReceived)
|
||
|
||
client := mqtt.NewClient(opts)
|
||
if token := client.Connect(); token.Wait() && token.Error() != nil {
|
||
fmt.Println(token.Error())
|
||
return
|
||
}
|
||
|
||
// 订阅主题以接收命令
|
||
client.Subscribe(topic, 0, nil)
|
||
|
||
for {
|
||
pcInfo, err := getPCInfo()
|
||
if err != nil {
|
||
fmt.Println("Error getting PC info:", err)
|
||
continue
|
||
}
|
||
|
||
pcInfoJSON, err := json.Marshal(pcInfo)
|
||
if err != nil {
|
||
fmt.Println("Error encoding PC info to JSON:", err)
|
||
continue
|
||
}
|
||
|
||
token := client.Publish(topic, 0, false, pcInfoJSON)
|
||
token.Wait()
|
||
|
||
fmt.Println("Message published:", string(pcInfoJSON))
|
||
|
||
time.Sleep(10 * time.Second) // 控制发送频率，例如每10秒发送一次
|
||
}
|
||
|
||
// 完成后断开 MQTT 客户端连接
|
||
// client.Disconnect(250)
|
||
}
|