JavaScript系列（53）--内存管理与垃圾回收机制详解

JavaScript内存管理与垃圾回收机制详解 🧹

今天，让我们深入探讨JavaScript的内存管理与垃圾回收机制。理解这些机制对于编写高性能、无内存泄漏的JavaScript应用至关重要。

内存管理基础概念 🌟

💡 小知识：JavaScript的内存管理是自动的，使用垃圾回收器（Garbage Collector，GC）来自动释放不再使用的内存。主要有两种回收算法：标记-清除（Mark-Sweep）和引用计数（Reference Counting）。

垃圾回收算法实现 📊

// 1. 引用计数垃圾回收
class ReferenceCounter {
    constructor() {
        this.references = new WeakMap();
    }
    
    addReference(object) {
        const count = this.references.get(object) || 0;
        this.references.set(object, count + 1);
    }
    
    removeReference(object) {
        const count = this.references.get(object);
        if (count > 1) {
            this.references.set(object, count - 1);
        } else {
            this.references.delete(object);
        }
    }
    
    isGarbage(object) {
        return !this.references.has(object);
    }
}

// 2. 标记-清除垃圾回收
class MarkAndSweep {
    constructor() {
        this.objects = new Set();
        this.marks = new WeakSet();
    }
    
    addObject(object) {
        this.objects.add(object);
    }
    
    mark(root) {
        // 如果对象已经被标记，直接返回
        if (this.marks.has(root)) return;
        
        // 标记当前对象
        this.marks.add(root);
        
        // 递归标记所有引用
        for (const reference of this.getReferences(root)) {
            this.mark(reference);
        }
    }
    
    sweep() {
        const garbage = [];
        
        for (const object of this.objects) {
            if (!this.marks.has(object)) {
                garbage.push(object);
                this.objects.delete(object);
            }
        }
        
        // 清除标记
        this.marks = new WeakSet();
        
        return garbage;
    }
    
    getReferences(object) {
        const references = [];
        
        // 获取对象的所有属性引用
        for (const key in object) {
            if (typeof object[key] === 'object' && object[key] !== null) {
                references.push(object[key]);
            }
        }
        
        return references;
    }
}

// 3. 分代垃圾回收
class GenerationalGC {
    constructor() {
        this.youngGeneration = new Set();
        this.oldGeneration = new Set();
        this.survivorSpace = new Set();
        this.maxYoungSize = 1000000; // 1MB
        this.promotionThreshold = 3;
    }
    
    allocate(object) {
        if (this.youngGeneration.size >= this.maxYoungSize) {
            this.minorGC();
        }
        
        this.youngGeneration.add({
            object,
            age: 0
        });
    }
    
    minorGC() {
        // 标记-复制算法
        for (const item of this.youngGeneration) {
            if (this.isReachable(item.object)) {
                item.age++;
                if (item.age >= this.promotionThreshold) {
                    // 晋升到老年代
                    this.oldGeneration.add(item.object);
                } else {
                    // 移动到幸存区
                    this.survivorSpace.add(item);
                }
            }
        }
        
        // 清空年轻代
        this.youngGeneration = this.survivorSpace;
        this.survivorSpace = new Set();
    }
    
    majorGC() {
        // 对老年代进行完整的标记-清除
        const markAndSweep = new MarkAndSweep();
        
        for (const object of this.oldGeneration) {
            markAndSweep.addObject(object);
        }
        
        // 从根对象开始标记
        const roots = this.getRoots();
        for (const root of roots) {
            markAndSweep.mark(root);
        }
        
        // 清除未标记对象
        const garbage = markAndSweep.sweep();
        for (const object of garbage) {
            this.oldGeneration.delete(object);
        }
    }
    
    isReachable(object) {
        // 实现可达性分析
        const visited = new Set();
        const stack = [object];
        
        while (stack.length > 0) {
            const current = stack.pop();
            if (!visited.has(current)) {
                visited.add(current);
                
                // 检查是否有根引用
                if (this.isRoot(current)) {
                    return true;
                }
                
                // 添加所有引用到栈中
                for (const reference of this.getReferences(current)) {
                    stack.push(reference);
                }
            }
        }
        
        return false;
    }
}

内存泄漏检测 🔍

// 1. 内存泄漏检测器
class MemoryLeakDetector {
    constructor() {
        this.snapshots = [];
        this.leakThreshold = 1000000; // 1MB
    }
    
    takeSnapshot() {
        const snapshot = {
            timestamp: Date.now(),
            memory: performance.memory.usedJSHeapSize,
            objects: this.getObjectCounts()
        };
        
        this.snapshots.push(snapshot);
        return snapshot;
    }
    
    getObjectCounts() {
        const counts = new Map();
        
        // 遍历全局对象
        for (const key in window) {
            const value = window[key];
            if (typeof value === 'object' && value !== null) {
                const type = value.constructor.name;
                counts.set(type, (counts.get(type) || 0) + 1);
            }
        }
        
        return counts;
    }
    
    detectLeaks() {
        if (this.snapshots.length < 2) {
            return null;
        }
        
        const latest = this.snapshots[this.snapshots.length - 1];
        const previous = this.snapshots[this.snapshots.length - 2];
        
        const memoryDiff = latest.memory - previous.memory;
        const leaks = new Map();
        
        // 检查对象数量变化
        for (const [type, count] of latest.objects) {
            const prevCount = previous.objects.get(type) || 0;
            const diff = count - prevCount;
            
            if (diff > 0) {
                leaks.set(type, diff);
            }
        }
        
        return {
            memoryDiff,
            leaks,
            isSignificant: memoryDiff > this.leakThreshold
        };
    }
}

// 2. 循环引用检测
class CircularReferenceDetector {
    detect(object, path = new Set()) {
        // 如果对象已经在路径中，发现循环引用
        if (path.has(object)) {
            return true;
        }
        
        // 只检查对象类型
        if (typeof object !== 'object' || object === null) {
            return false;
        }
        
        // 将当前对象添加到路径
        path.add(object);
        
        // 递归检查所有属性
        for (const key in object) {
            if (this.detect(object[key], path)) {
                return true;
            }
        }
        
        // 从路径中移除当前对象
        path.delete(object);
        return false;
    }
    
    findCircularPaths(object, path = [], seen = new Set()) {
        const paths = [];
        
        if (typeof object !== 'object' || object === null) {
            return paths;
        }
        
        if (seen.has(object)) {
            paths.push([...path]);
            return paths;
        }
        
        seen.add(object);
        
        for (const key in object) {
            path.push(key);
            paths.push(...this.findCircularPaths(object[key], path, seen));
            path.pop();
        }
        
        seen.delete(object);
        return paths;
    }
}

// 3. 内存使用分析器
class MemoryAnalyzer {
    constructor() {
        this.samples = [];
        this.interval = null;
    }
    
    startSampling(interval = 1000) {
        this.interval = setInterval(() => {
            this.takeSample();
        }, interval);
    }
    
    stopSampling() {
        if (this.interval) {
            clearInterval(this.interval);
            this.interval = null;
        }
    }
    
    takeSample() {
        const sample = {
            timestamp: Date.now(),
            memory: performance.memory.usedJSHeapSize,
            heapLimit: performance.memory.jsHeapSizeLimit,
            allocation: this.getRecentAllocations()
        };
        
        this.samples.push(sample);
        
        // 保持最近100个样本
        if (this.samples.length > 100) {
            this.samples.shift();
        }
        
        return sample;
    }
    
    getRecentAllocations() {
        // 获取最近的内存分配情况
        if (this.samples.length < 2) {
            return 0;
        }
        
        const latest = this.samples[this.samples.length - 1];
        const previous = this.samples[this.samples.length - 2];
        
        return latest.memory - previous.memory;
    }
    
    getAnalysis() {
        if (this.samples.length < 2) {
            return null;
        }
        
        const memoryGrowth = this.samples.reduce((acc, sample, index) => {
            if (index === 0) return acc;
            const growth = sample.memory - this.samples[index - 1].memory;
            return acc + growth;
        }, 0);
        
        const averageGrowth = memoryGrowth / (this.samples.length - 1);
        const maxMemory = Math.max(...this.samples.map(s => s.memory));
        const minMemory = Math.min(...this.samples.map(s => s.memory));
        
        return {
            averageGrowth,
            maxMemory,
            minMemory,
            memoryUtilization: maxMemory / this.samples[0].heapLimit,
            trend: averageGrowth > 1000 ? 'increasing' : 'stable'
        };
    }
}

性能优化策略 ⚡

// 1. 对象池实现
class ObjectPool {
    constructor(factory, initialSize = 10) {
        this.factory = factory;
        this.pool = [];
        this.activeObjects = new WeakSet();
        
        // 初始化对象池
        for (let i = 0; i < initialSize; i++) {
            this.pool.push(this.factory());
        }
    }
    
    acquire() {
        let object = this.pool.pop();
        
        if (!object) {
            object = this.factory();
        }
        
        this.activeObjects.add(object);
        return object;
    }
    
    release(object) {
        if (this.activeObjects.has(object)) {
            this.activeObjects.delete(object);
            
            // 重置对象状态
            if (object.reset) {
                object.reset();
            }
            
            this.pool.push(object);
        }
    }
    
    get size() {
        return this.pool.length;
    }
}

// 2. 内存限制器
class MemoryLimiter {
    constructor(maxMemory = 100 * 1024 * 1024) { // 100MB
        this.maxMemory = maxMemory;
        this.currentMemory = 0;
        this.allocations = new WeakMap();
    }
    
    allocate(size) {
        if (this.currentMemory + size > this.maxMemory) {
            throw new Error('Memory limit exceeded');
        }
        
        const object = new ArrayBuffer(size);
        this.allocations.set(object, size);
        this.currentMemory += size;
        
        return object;
    }
    
    free(object) {
        const size = this.allocations.get(object);
        if (size) {
            this.currentMemory -= size;
            this.allocations.delete(object);
        }
    }
    
    get availableMemory() {
        return this.maxMemory - this.currentMemory;
    }
}

// 3. 缓存管理器
class CacheManager {
    constructor(maxSize = 1000) {
        this.maxSize = maxSize;
        this.cache = new Map();
        this.accessCount = new Map();
    }
    
    set(key, value, ttl = 3600000) { // 默认1小时
        if (this.cache.size >= this.maxSize) {
            this.evictLeastUsed();
        }
        
        this.cache.set(key, {
            value,
            expires: Date.now() + ttl
        });
        this.accessCount.set(key, 0);
    }
    
    get(key) {
        const entry = this.cache.get(key);
        
        if (!entry) {
            return null;
        }
        
        if (entry.expires < Date.now()) {
            this.cache.delete(key);
            this.accessCount.delete(key);
            return null;
        }
        
        this.accessCount.set(key, this.accessCount.get(key) + 1);
        return entry.value;
    }
    
    evictLeastUsed() {
        let leastUsedKey = null;
        let leastUsedCount = Infinity;
        
        for (const [key, count] of this.accessCount) {
            if (count < leastUsedCount) {
                leastUsedKey = key;
                leastUsedCount = count;
            }
        }
        
        if (leastUsedKey) {
            this.cache.delete(leastUsedKey);
            this.accessCount.delete(leastUsedKey);
        }
    }
    
    cleanup() {
        const now = Date.now();
        for (const [key, entry] of this.cache) {
            if (entry.expires < now) {
                this.cache.delete(key);
                this.accessCount.delete(key);
            }
        }
    }
}

最佳实践建议 💡

1. 内存优化模式

// 1. 弱引用处理
class WeakRefHandler {
    constructor() {
        this.refs = new WeakMap();
        this.registry = new FinalizationRegistry(key => {
            console.log(`Object with key ${key} has been garbage collected`);
        });
    }
    
    addRef(key, object) {
        const ref = new WeakRef(object);
        this.refs.set(key, ref);
        this.registry.register(object, key);
    }
    
    getRef(key) {
        const ref = this.refs.get(key);
        return ref ? ref.deref() : null;
    }
}

// 2. 大数据处理
class LargeDataHandler {
    constructor() {
        this.chunkSize = 1000;
    }
    
    *processInChunks(data) {
        for (let i = 0; i < data.length; i += this.chunkSize) {
            const chunk = data.slice(i, i + this.chunkSize);
            yield this.processChunk(chunk);
        }
    }
    
    processChunk(chunk) {
        // 处理数据块的逻辑
        return chunk.map(item => item * 2);
    }
}

// 3. 资源释放管理
class ResourceManager {
    constructor() {
        this.resources = new Set();
    }
    
    acquire(resource) {
        this.resources.add(resource);
        return resource;
    }
    
    release(resource) {
        if (this.resources.has(resource)) {
            if (resource.dispose) {
                resource.dispose();
            }
            this.resources.delete(resource);
        }
    }
    
    releaseAll() {
        for (const resource of this.resources) {
            this.release(resource);
        }
    }
}

结语 📝

JavaScript的内存管理和垃圾回收机制是确保应用性能和稳定性的关键。通过本文，我们学习了：

1. 内存管理的基本原理
2. 垃圾回收算法的实现
3. 内存泄漏的检测和防范
4. 性能优化策略
5. 最佳实践和设计模式

💡 学习建议：在开发中要时刻注意内存使用情况，特别是在处理大量数据或长期运行的应用时。合理使用内存优化工具和模式，可以显著提升应用的性能和稳定性。

如果你觉得这篇文章有帮助，欢迎点赞收藏，也期待在评论区看到你的想法和建议！👇

终身学习，共同成长。

咱们下一期见