LRU在iOS中的实现

LRU是一种缓存淘汰算法,是一种缓存淘汰机制

具体原理不做赘述,这里只提供实现示例。

*目前只实现了Swift&OC版本,后续会更新双链表等实现方法。
Swift代码示例:

//
//  LRUCache.swift
//  LRU缓存工具类
//  本类只是实现LRU思路,并没有进行真正意义的本地缓存数据。具体采用什么方式进行本地缓存,请自行添加。
//  Created by Miracle on 2022/4/15.
//


import UIKit

class LRUCache: NSObject {

    static let shared = LRUCache()
    
    
    // MARK: 私有属性
    
    
    /// 最大数量
    private var maxCount: Int = 0
    
    /// 存放key的数组
    private var keysArr: [String] = [String]()
    
    /// 存放数据的字典
    private var dataDic: [String:Any] = [String:Any]()
    
    
    // MARK: 公开方法
    
    /// 创建/重置方法
    /// - Parameter maxCount: 最大缓存数量
    func initOrReset(maxCount: Int) {
        
        //赋值最大缓存数量
        self.maxCount = maxCount
        
        //清空记录的内容
        keysArr.removeAll()
        dataDic.removeAll()
    }
    
    
    /// 增加数据缓存方法
    /// - Parameters:
    ///   - value: 数据
    ///   - key: key
    func add(value: Any, key: String) {
        
        //是否已经存在了
        let isExist = keysArr.contains(key) && dataDic.keys.contains(key)
        
        //存数据
        dataDic[key] = value
        
        //将key移动到最前面
        if isExist {
            moveToHeader(key: key)
        } else {
            keysArr.insert(key, at: 0)
        }
        
        //超出最大缓存限制,删除末位数据
        if keysArr.count > maxCount {
            let removeKey = keysArr.last ?? ""
            remove(key: removeKey)
        }
    }
    
    
    /// 根据key删除缓存数据
    func remove(key: String) {
        
        //移除数据字典中的对应数据
        if dataDic.keys.contains(key) {
            dataDic.removeValue(forKey: key)
        }
        
        //移除key数组中的key
        if keysArr.contains(key) {
            keysArr.removeAll(){
                $0 == key
            }
        }
    }
    
    
    /// 根据key获取数据
    func getData(key: String) -> Any {
        
        //根据key取数据
        let data = dataDic[key] as Any
        
        //将key移动到最前面
        moveToHeader(key: key)
        
        return data
    }
    
    
    // MARK: 私有方法
    
    
    /// 将key移动到最前面
    /// - Parameter key: 数据key
    private func moveToHeader(key: String) {
        
        //不包含直接返回
        if !keysArr.contains(key) {
            return
        }
        
        //移除对应的key
        keysArr.removeAll(){
            $0 == key
        }
        
        //将key添加到第0个
        keysArr.insert(key, at: 0)
    }
    
}

OC代码示例:
LRUCache-OC.h

//
//  LRUCache-OC.h
//  LRU缓存工具类
//  本类只是实现LRU思路,并没有进行真正意义的本地缓存数据。具体采用什么方式进行本地缓存,请自行添加。
//  Created by Miracle on 2022/4/20.
//

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@interface LRUCache_OC : NSObject

+(instancetype)shard;

/// 创建/重置方法
/// @param maxCount 最大缓存数量
-(void)initOrResetWithMaxCount:(int)maxCount;

/// 添加缓存
-(void)addWithKey:(NSString *)key value:(id)value;

/// 根据key删除数据
-(void)removeWithKey:(NSString *)key;

/// 根据key获取数据
-(id)getDataWithKey:(NSString *)key;
@end

NS_ASSUME_NONNULL_END

LRUCache-OC.m

//
//  LRUCache-OC.m
//  LRU缓存工具类
//  本类只是实现LRU思路,并没有进行真正意义的本地缓存数据。具体采用什么方式进行本地缓存,请自行添加。
//  Created by Miracle on 2022/4/20.
//

#import "LRUCache-OC.h"

@interface LRUCache_OC()

/// 最大数量
@property(nonatomic,assign) int maxCount;

/// 存放key的数组
@property(nonatomic,strong) NSMutableArray * keysArr;

/// 存放数据的字典
@property(nonatomic,strong) NSMutableDictionary * dataDic;

@end

@implementation LRUCache_OC


+(instancetype)shard{
    
    static LRUCache_OC * _shard;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        _shard = [[[self class] alloc] init];
    });
    return _shard;
}


/// 创建/重置方法
/// @param maxCount 最大缓存数量
-(void)initOrResetWithMaxCount:(int)maxCount{
    
    self.maxCount = maxCount;
    self.keysArr = [[NSMutableArray alloc]initWithCapacity:0];
    self.dataDic = [[NSMutableDictionary alloc]initWithCapacity:0];
}

/// 添加缓存
-(void)addWithKey:(NSString *)key value:(id)value{
    
    //是否已经存在了
    BOOL isExist = [self.keysArr containsObject:key] && [self.dataDic.allKeys containsObject:key];
    
    //存数据
    [self.dataDic setValue:value forKey:key];
    
    //将key移动到最前面
    if (isExist) {
        [self moveToHeaderForKey:key];
    } else {
        [self.keysArr insertObject:key atIndex:0];
    }
    
    //超出最大限制,删除末位数据
    if (self.keysArr.count > self.maxCount) {
        NSString * lastKey = self.keysArr.lastObject;
        [self removeWithKey:lastKey];
    }
}


/// 将key移动到最前面
-(void)moveToHeaderForKey:(NSString *)key{
    
    //如果不包含,直接返回
    if (![self.keysArr containsObject:key]) {
        return;
    }
    
    [self.keysArr removeObject:key];
    [self.keysArr insertObject:key atIndex:0];
}


/// 根据key删除数据
-(void)removeWithKey:(NSString *)key{
    
    if ([self.dataDic.allKeys containsObject:key]) {
        [self.dataDic removeObjectForKey:key];
    }
    
    if ([self.keysArr containsObject:key]) {
        [self.keysArr removeObject:key];
    }
}


/// 根据key获取数据
-(id)getDataWithKey:(NSString *)key{
    
    id data = [self.dataDic objectForKey:key];
    
    [self moveToHeaderForKey:key];
    
    return data;
}


@end

github链接

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1、LRU是什么? LRU(Least recently used,最近最少使用)。它是按照一个非常著名的计算机操...
    青年fw阅读 1,077评论 0 4
  • 读完本文,你可以去力扣拿下如下题目: 146.LRU缓存机制[https://leetcode-cn.com/pr...
    labuladong阅读 990评论 1 5
  • 1 LinkedHashMap的概述 public class LinkedHashMap<K,V> extend...
    程序员麦冬阅读 182评论 0 1
  • LRU就是一种缓存淘汰策略。 计算机的缓存容量有限,如果缓存满了就要删除一些内容,给新内容腾位置。但问题是,删除哪...
    随风_d6a2阅读 616评论 0 2
  • 16宿命:用概率思维提高你的胜算 以前的我是风险厌恶者,不喜欢去冒险,但是人生放弃了冒险,也就放弃了无数的可能。 ...
    yichen大刀阅读 6,122评论 0 4