bl/logic/main.go

package main

import (
	"fmt"
	"log"
	"os"
	"runtime"
	"strings"
	"time"

	_ "github.com/gogf/gf/contrib/nosql/redis/v2"
	"github.com/gogf/gf/v2/os/gcmd"
	"github.com/gogf/gf/v2/os/gproc"

	"blazing/logic/service/player"

	"blazing/cool"

	"github.com/shirou/gopsutil/v4/mem"

	"blazing/modules/base/service"
	blservice "blazing/modules/player/service"
	"net/http"
	_ "net/http/pprof"
)

// PprofWeb 启动pprof性能分析web服务
func PprofWeb() {
	runtime.SetMutexProfileFraction(1) // (非必需)开启对锁调用的跟踪
	runtime.SetBlockProfileRate(1)     // (非必需)开启对阻塞操作的跟踪
	err := http.ListenAndServe(":9909", nil)
	if err != nil {
		panic(err)
	}
}

func cleanup() {
	log.Println("执行优雅清理资源...")

	player.Mainplayer.Range(func(key uint32, value *player.Player) bool {
		value.Kick(1)

		return false
	})
	//fight.Fightpool.ReleaseTimeout(0)
	log.Println("资源清理完成，程序即将退出")
}

// signalHandlerForMain 主进程信号处理函数
func signalHandlerForMain(sig os.Signal) {
	cleanup()
	fmt.Println("MainProcess is shutting down due to signal:", sig.String())
}

// main 程序主入口函数
func main() {
	// item := model.NeweggConfig()
	// item.GeneratedPetIDs = []model.GeneratedPetID{
	// 	{PetID: 1, Prob: 0.01},
	// 	{PetID: 2, Prob: 0.01},
	// 	{PetID: 3, Prob: 0.01},
	// 	{PetID: 4, Prob: 0.01},
	// 	{PetID: 5, Prob: 0.01},
	// 	{PetID: 6, Prob: 0.01},
	// }
	// item.MalePet = 1
	// item.FemalePet = 2
	// _, err := g.DB(item.GroupName()).Model(item.TableName()).FieldsEx("id").Data(item).Insert()
	// if err != nil {
	// 	panic(err)
	// }
	//loadAccounts()
	// if cool.IsRedisMode {
	// 	go cool.ListenFunc(gctx.New())
	// }
	// 解析命令行参数
	cool.Config.GameOnlineID = gcmd.GetOpt("id", "1").Uint16()
	go Start() //注入service
	// if cool.Config.PortBL == 1 || cool.Config.PortBL == 2 { //只分析1服务器的
	// 	go PprofWeb()
	// }
	go PprofWeb()
	go monitorMemAndQuit()
	fmt.Println("Process start, pid:", os.Getpid())

	gproc.AddSigHandlerShutdown(

		signalHandlerForMain,
	)

	gproc.Listen()
}

const (
	memThresholdRatio = 0.9              // 内存占用阈值70%
	checkInterval     = 10 * time.Second // 内存检测间隔，可按需调整
)

// 监控内存，超阈值则优雅退出程序
func monitorMemAndQuit() {
	var memStats runtime.MemStats
	for {
		// 1. 获取系统总内存和可用内存
		sysMem, err := mem.VirtualMemory()
		if err != nil {
			log.Printf("获取系统内存失败：%v\n", err)
			time.Sleep(checkInterval)
			continue
		}

		// 2. 获取Go进程当前堆内存占用（进程实际使用的核心内存）
		runtime.ReadMemStats(&memStats)
		procUsedMem := memStats.HeapInuse // 进程堆内存占用（字节）
		sysTotalMem := sysMem.Total       // 系统总内存（字节）

		// 3. 计算进程内存占系统总内存的比例
		usedRatio := float64(procUsedMem) / float64(sysTotalMem)
		// 格式化输出（MB），方便查看
		procUsedMB := procUsedMem / 1024 / 1024
		sysTotalMB := sysTotalMem / 1024 / 1024

		log.Printf("当前内存：进程占用%vMB / 系统总%vMB，占比%.1f%%",
			procUsedMB, sysTotalMB, usedRatio*100)

		// 4. 超70%阈值，执行优雅退出
		if usedRatio >= memThresholdRatio {
			log.Fatalf("内存占比达%.1f%%，超过70%阈值，程序开始退出", usedRatio*100)
			// ########## 可选：这里添加你的优雅清理逻辑 ##########
			// 如：关闭数据库连接、释放文件句柄、保存业务状态、推送退出告警等
			cleanup()
			// 退出程序，返回非0码（方便进程管理工具识别并重启）
			os.Exit(1)
		}

		time.Sleep(checkInterval)
	}
}

// loadAccounts 从CSV文件加载账号信息
func loadAccounts() {
	t1, _ := os.Getwd()
	data, err := os.ReadFile(t1 + "/b.csv")
	if err != nil {
		panic(err)
	}
	lines := strings.Split(string(data), "\n")
	for _, line := range lines {
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}
		t := service.NewBaseSysUserService().GetEamil(line)

		blservice.NewUserService(uint32(t.ID)).Info.Reg(t.Username, 0)
	}
	//fmt.Printf("加载 %d 个账号\n", len(accounts))
}