rconclient.go

package main

import (
	"fmt"
	"log"

	"github.com/gorcon/rcon"
	"github.com/hoshinonyaruko/palworld-go/config"
)

// RconClient 结构体，用于存储RCON连接和配置信息
type RconClient struct {
	Conn       *rcon.Conn
	BackupTask *BackupTask
	Config     *config.Config
}

// NewRconClient 创建一个新的RCON客户端
func NewRconClient(address, password string, BackupTask *BackupTask, config *config.Config) *RconClient {
	conn, err := rcon.Dial(address, password)
	if err != nil {
		log.Printf("无法连接到RCON服务器: %v", err)
		return nil
	}
	return &RconClient{
		Conn:       conn,
		BackupTask: BackupTask,
		Config:     config,
	}
}

// Close 关闭RCON连接
func (client *RconClient) Close() {
	err := client.Conn.Close()
	if err != nil {
		log.Printf("关闭RCON连接时发生错误: %v", err)
	}
}

// 重启服务器
func RestartServer(RconClient *RconClient) error {
	if _, err := RconClient.Conn.Execute("broadcast Auto_Reboot_Initialized", RconClient.Config.UseDll); err != nil {
		return fmt.Errorf("error broadcasting restart initialization: %w", err)
	}
	if _, err := RconClient.Conn.Execute("save", RconClient.Config.UseDll); err != nil {
		return fmt.Errorf("error saving game state: %w", err)
	}
	if _, err := RconClient.Conn.Execute("shutdown 300 Server_is_going_to_reboot_in_5_minutes", RconClient.Config.UseDll); err != nil {
		return fmt.Errorf("error executing shutdown: %w", err)
	}
	return nil
}

// 发广播 重启维护
func HandleMemoryUsage(threshold float64, RconClient *RconClient, config config.Config) {
	// 广播内存超阈值的警告
	if _, err := RconClient.Conn.Execute(fmt.Sprintf("broadcast Memory_Is_Above_%v%%", threshold), config.UseDll); err != nil {
		log.Printf("Error broadcasting memory threshold alert: %v", err)
	}

	// 原有的方式发送广播
	if _, err := RconClient.Conn.Execute("broadcast "+config.MaintenanceWarningMessage, config.UseDll); err != nil {
		log.Printf("Error broadcasting: %v", err)
	}

	// 保存游戏状态
	if _, err := RconClient.Conn.Execute("save", config.UseDll); err != nil {
		log.Printf("Error saving game state: %v", err)
	}

	// 安排服务器重启
	if _, err := RconClient.Conn.Execute("shutdown 60 Reboot_In_60_Seconds", config.UseDll); err != nil {
		log.Printf("Error executing shutdown: %v", err)
	}

	RconClient.BackupTask.RunBackup()
}

func Broadcast(message string, RconClient *RconClient, usedll bool) {
	// 广播
	if _, err := RconClient.Conn.Execute("broadcast "+message, usedll); err != nil {
		log.Printf("Error broadcasting : %v", err)
	}
}