package main

import (
	"fmt"
	"runtime"
	"sync"
	"time"
)

// 初始化1KB字节切片对象池
var bytePool = sync.Pool{
	New: func() interface{} {
		return make([]byte, 1024)
	},
}

// 打印GC统计，衔接你之前的知识
func printGCStatsInfo(title string) {
	var stats runtime.MemStats
	runtime.ReadMemStats(&stats)
	fmt.Printf("[%s] GC次数: %d, 存活堆内存: %d KB\n", title, stats.NumGC, stats.HeapAlloc/1024)
}

// withoutPool 不使用对象池，频繁分配新内存
// 增加参数 []byte 模拟真实业务使用，禁止编译器消除代码
func withoutPool() time.Duration {
	start := time.Now()
	var temp []byte                // 用于承接数据，避免死代码
	for i := 0; i < 1000000; i++ { // 循环次数提升10倍，让耗时可被精准统计
		buf := make([]byte, 1024)
		temp = append(temp, buf...) // 模拟真实使用切片，编译器无法优化
	}
	_ = temp
	return time.Since(start)
}

// withPool 使用对象池复用内存
func withPool() time.Duration {
	start := time.Now()
	var temp []byte                // 用于承接数据，避免死代码
	for i := 0; i < 1000000; i++ { // 同步提升循环次数
		buf := bytePool.Get().([]byte)
		temp = append(temp, buf...) // 模拟真实使用切片
		bytePool.Put(buf[:0])       // 正确放回，保留你的最佳实践
	}
	_ = temp
	return time.Since(start)
}

func main() {
	// 预热并清空GC状态
	runtime.GC()
	printGCStatsInfo("初始化")

	// 测试无对象池
	dur1 := withoutPool()
	printGCStatsInfo("无对象池后")
	runtime.GC() // 强制GC，消除前序影响

	// 测试有对象池
	dur2 := withPool()
	printGCStatsInfo("有对象池后")

	// 增加分母非0判断，避免-Inf%
	var rate float64
	if dur1 > 0 {
		rate = (1 - float64(dur2)/float64(dur1)) * 100
	}

	fmt.Printf("\n无对象池耗时: %v\n", dur1)
	fmt.Printf("有对象池耗时: %v\n", dur2)
	fmt.Printf("性能提升: %.2f%%\n", rate)
}