Go语言中限制内存分配的方法

package main

import (
    "fmt"
    "os"
    "strconv"
)

func main() {
    // 1. 获取用户指定的内存大小 (以MB为单位)
    maxMemoryMBStr := os.Getenv("MAX_MEMORY_MB") // 从环境变量获取
    if maxMemoryMBStr == "" {
        fmt.Println("MAX_MEMORY_MB 环境变量未设置，使用默认值 512MB")
        maxMemoryMBStr = "512" // 默认值
    }

    maxMemoryMB, err := strconv.Atoi(maxMemoryMBStr)
    if err != nil {
        fmt.Println("MAX_MEMORY_MB 环境变量无效，必须是整数")
        os.Exit(1)
    }

    // 将MB转换为字节
    maxMemoryBytes := int64(maxMemoryMB) * 1024 * 1024

    fmt.Printf("允许的最大内存使用量: %d MB\n", maxMemoryMB)

    // 2. 模拟大数据加载 (实际情况需要根据你的数据源进行调整)
    dataSize := int64(2 * 1024 * 1024 * 1024) // 2GB 的数据
    chunkSize := maxMemoryBytes / 4 // 将数据分成 4 个 chunk (示例)

    for i := int64(0); i < dataSize; i += chunkSize {
        // 计算当前 chunk 的大小
        currentChunkSize := chunkSize
        if i+chunkSize > dataSize {
            currentChunkSize = dataSize - i
        }

        // 模拟加载数据块 (实际情况需要从文件或网络加载)
        dataChunk := make([]byte, currentChunkSize)
        fmt.Printf("加载数据块 %d: 大小 %d 字节\n", i/chunkSize+1, currentChunkSize)

        // TODO: 处理数据块 (dataChunk)

        // 释放内存 (如果需要，可以手动调用 runtime.GC()，但通常不建议)
        dataChunk = nil
        //runtime.GC() // 谨慎使用
    }

    fmt.Println("数据处理完成")
}