使用Map实现redis的功能(适用于单机版)

背景:某个项目需要用到类似redis的功能(存储键值对,并且希望键值对是有过期时间的)。如果只是为了一个小功能,就安装一个redis服务器,那明显太浪费性能。所以开发了这个类,使用MAP来模仿和实现redis的处理。

原理:利用两个Map,一个map用来存储值,一个map用来存储过期时间。另外有一个线程,定期去扫描那些key过期,如果过期,则移除这个值。

注:使用ConcurrentHashMap是为了线程安全。


代码如下:

/**
 * 
 */
package com.yihu.util.mapRedis;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;



/**
 * 使用Map实现redis的功能(只支持单机)
 * @author lch 201705
 *
 */
public class MapForRedisUtils extends Thread {
	private static MapForRedisUtils mapForRedisUtils=null;
	private static Map<String,String> map=new ConcurrentHashMap<String,String>();
	private static Map<String,Long> expireMap=new ConcurrentHashMap<String,Long>();
	
	/**
	 * 存储值
	 * @param key 健
	 * @param val 值
	 * @param minute  过期时间(分钟)
	 */
	public static void setValue(String key,String val,int minute){
		try{
			map.put(key,val);
			expireMap.put(key,System.currentTimeMillis()+minute*60L*1000);
		}catch(Exception e){
			e.printStackTrace();
		}
	}
	
	/**
	 * 获取值
	 * @param key 健
	 */
	public static String getValue(String key){
		try{
			return map.get(key);
		}catch(Exception e){
			e.printStackTrace();
			return null;
		}
	}	

	
	/**
	 * 获取过期时间(相对于1970-1-1的毫秒数)
	 * @param key 健
	 */
	public static long getExpireTime(String key){
		try{
			return expireMap.get(key);
		}catch(Exception e){
			e.printStackTrace();
			return 0;
		}
	}

	/**
	 * 设置过期时间(毫秒数)
	 * @param key 健
	 * @param expireMills 多少毫秒
	 */
	public static long setExpireTime(String key,int minute){
		try{
			return expireMap.put(key,System.currentTimeMillis()+minute*60L*1000);
		}catch(Exception e){
			e.printStackTrace();
			return 0;
		}
	}
	
	@Override
	public void run() {
		while(true){
			try{
				
				//移除过期的数据
				List<String> needRemoveKey=new ArrayList<String>();
				for (Map.Entry<String, Long> entry : MapForRedisUtils.expireMap.entrySet()) {
					if(entry.getValue()!=null&&entry.getValue()<=System.currentTimeMillis()){
						needRemoveKey.add(entry.getKey());
					}
				}
				for(String key:needRemoveKey){
					map.remove(key);
					expireMap.remove(key);
				}
				
				try {
					Thread.sleep(10000);//暂停10秒钟,变成死循环
				} catch (Exception e1) {
					e1.printStackTrace();
				}
			}catch(Exception e){
				try {
					Thread.sleep(1000);//防止故障的时候,变成死循环
				} catch (Exception e1) {
					e1.printStackTrace();
				}
			}
		}
	}
	
	
	public static void init(){
		if(mapForRedisUtils==null){
			mapForRedisUtils=new MapForRedisUtils();
			mapForRedisUtils.start();//启动监视线程(自动删除过期的数据)
		}
	}
	
	public static void main(String[] args) throws InterruptedException {
		MapForRedisUtils.init();
		MapForRedisUtils.setValue("test", "1111111", 1);//设置值
		System.out.println(MapForRedisUtils.getValue("test"));//读取值
		Thread.sleep(2*60*1000);
		System.out.println("2 min after,val="+MapForRedisUtils.getValue("test"));//数据过期之后,再去读取,就发现没有了
	}
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值