文档章节

Key锁

longload
 longload
发布于 2016/07/12 17:47
字数 1813
阅读 8
收藏 0

java中的几种锁:synchronized,ReentrantLock,ReentrantReadWriteLock已基本可以满足编程需求,但其粒度都太大,同一时刻只有一个线程能进入同步块,这对于某些高并发的场景并不适用。本文实现了一个基于KEY(主键)的互斥锁,具有更细的粒度,在缓存或其他基于KEY的场景中有很大的用处。下面将讲解这个锁的设计和实现

(关于这个锁的讨论贴:KeyLock讨论贴-CSDN

设想这么一个场景:转账

private int[] accounts; // 账户数组,其索引为账户ID,内容为金额
	
	public boolean transfer(int from, int to, int money) {
		if (accounts[from] < money)
			return false;
		accounts[from] -= money;
		accounts[to] += money;
		return true;
	}

 从from中转出金额到to中。可能同时会有很多个线程同时调用这个转账方法,为保证原子性,保证金额不会出错,必须为这个方法加个锁,防止对共享变量accounts的并发修改。

加锁后的代码如下:

private int[] accounts; // 账户数组,其索引为账户ID,内容为金额
	private Lock lock = new ReentrantLock();

	public boolean transfer(int from, int to, int money) {
		lock.lock();
		try {
			if (accounts[from] < money)
				return false;
			accounts[from] -= money;
			accounts[to] += money;
			return true;
		} finally {
			lock.unlock();
		}
	}

 好了,加锁后这个代码就能保证金额不出错了。但问题又出现了,一次只能执行一个转账过程!意思就是A给B转账的时候,C要给D转账也得等A给B转完了才能开始转。这就有点扯蛋了,就像只有一个柜台,所有人必须排队等前面的处理完了才能到自己,效率太低。

解决这种情况有一个方案:A给B转账的时候只锁定A和B的账户,使其转账期间不能再有其他针对A和B账户的操作,但其他账户的操作可以并行发生。类似于如下场景:

public boolean transfer(int from, int to, int money) {
		lock.lock(from, to);
		try {
			if (accounts[from] < money)
				return false;
			accounts[from] -= money;
			accounts[to] += money;
			return true;
		} finally {
			lock.unlock(from, to);
		}
	}

 但很显然,JAVA并没有为我们提供这样的锁(也有可能是我没找到。。。)

于是,就在这样的需求下我花了整一天来实现了这个锁——KeyLock(代码量很短,但多线程的东西真的很让人头疼)

不同于synchronized等锁,KeyLock是对所需处理的数据的KEY(主键)进行加锁,只要是对不同key操作,其就可以并行处理,大大提高了线程的并行度(最后有几个锁的对比测试

总结下就是:对相同KEY操作的线程互斥,对不同KEY操作的线程可以并行

KeyLock有如下几个特性

    1、细粒度,高并行性
    2、可重入
    3、公平锁
    4、加锁开销比ReentrantLock大,适用于处理耗时长、key范围大的场景

KeyLock代码如下(注释很少,因为我也不知道该怎么写清楚,能看懂就看,懒得看的直接用就行):

public class KeyLock<K> {
	// 保存所有锁定的KEY及其信号量
	private final ConcurrentMap<K, Semaphore> map = new ConcurrentHashMap<K, Semaphore>();
	// 保存每个线程锁定的KEY及其锁定计数
	private final ThreadLocal<Map<K, LockInfo>> local = new ThreadLocal<Map<K, LockInfo>>() {
		@Override
		protected Map<K, LockInfo> initialValue() {
			return new HashMap<K, LockInfo>();
		}
	};

	/**
	 * 锁定key,其他等待此key的线程将进入等待,直到调用{@link #unlock(K)}
	 * 使用hashcode和equals来判断key是否相同,因此key必须实现{@link #hashCode()}和
	 * {@link #equals(Object)}方法
	 * 
	 * @param key
	 */
	public void lock(K key) {
		if (key == null)
			return;
		LockInfo info = local.get().get(key);
		if (info == null) {
			Semaphore current = new Semaphore(1);
			current.acquireUninterruptibly();
			Semaphore previous = map.put(key, current);
			if (previous != null)
				previous.acquireUninterruptibly();
			local.get().put(key, new LockInfo(current));
		} else {
			info.lockCount++;
		}
	}
	
	/**
	 * 释放key,唤醒其他等待此key的线程
	 * @param key
	 */
	public void unlock(K key) {
		if (key == null)
			return;
		LockInfo info = local.get().get(key);
		if (info != null && --info.lockCount == 0) {
			info.current.release();
			map.remove(key, info.current);
			local.get().remove(key);
		}
	}

	/**
	 * 锁定多个key
	 * 建议在调用此方法前先对keys进行排序,使用相同的锁定顺序,防止死锁发生
	 * @param keys
	 */
	public void lock(K[] keys) {
		if (keys == null)
			return;
		for (K key : keys) {
			lock(key);
		}
	}

	/**
	 * 释放多个key
	 * @param keys
	 */
	public void unlock(K[] keys) {
		if (keys == null)
			return;
		for (K key : keys) {
			unlock(key);
		}
	}

	private static class LockInfo {
		private final Semaphore current;
		private int lockCount;

		private LockInfo(Semaphore current) {
			this.current = current;
			this.lockCount = 1;
		}
	}
}

 KeyLock使用示例

 

private int[] accounts;
	private KeyLock<Integer> lock = new KeyLock<Integer>();
	
	public boolean transfer(int from, int to, int money) {
		Integer[] keys = new Integer[] {from, to};
		Arrays.sort(keys); //对多个key进行排序,保证锁定顺序防止死锁
		lock.lock(keys);
		try {
			//处理不同的from和to的线程都可进入此同步块
			if (accounts[from] < money)
				return false;
			accounts[from] -= money;
			accounts[to] += money;
			return true;
		} finally {
			lock.unlock(keys);
		}
	}

 好,工具有了,接下来就是测试了,为了测出并行度,我把转账过程延长了,加了个sleep(2),使每个转账过程至少要花2毫秒(这只是个demo,真实环境下对数据库操作也很费时)。

 

 

测试代码如下:

//场景:多线程并发转账
public class Test {
	private final int[] account; // 账户数组,其索引为账户ID,内容为金额

	public Test(int count, int money) {
		account = new int[count];
		Arrays.fill(account, money);
	}

	boolean transfer(int from, int to, int money) {
		if (account[from] < money)
			return false;
		account[from] -= money;
		try {
			Thread.sleep(2);
		} catch (Exception e) {
		}
		account[to] += money;
		return true;
	}
	
	int getAmount() {
		int result = 0;
		for (int m : account)
			result += m;
		return result;
	}

	public static void main(String[] args) throws Exception {
		int count = 100;		//账户个数
		int money = 10000;		//账户初始金额
		int threadNum = 8;		//转账线程数
		int number = 10000;		//转账次数
		int maxMoney = 1000;	//随机转账最大金额
		Test test = new Test(count, money);
		
		//不加锁
//		Runner runner = test.new NonLockRunner(maxMoney, number);
		//加synchronized锁
//		Runner runner = test.new SynchronizedRunner(maxMoney, number);
		//加ReentrantLock锁
//		Runner runner = test.new ReentrantLockRunner(maxMoney, number);
		//加KeyLock锁
		Runner runner = test.new KeyLockRunner(maxMoney, number);
		
		Thread[] threads = new Thread[threadNum];
		for (int i = 0; i < threadNum; i++)
			threads[i] = new Thread(runner, "thread-" + i);
		long begin = System.currentTimeMillis();
		for (Thread t : threads)
			t.start();
		for (Thread t : threads)
			t.join();
		long time = System.currentTimeMillis() - begin;
		System.out.println("类型:" + runner.getClass().getSimpleName());
		System.out.printf("耗时:%dms\n", time);
		System.out.printf("初始总金额:%d\n", count * money);
		System.out.printf("终止总金额:%d\n", test.getAmount());
	}

	// 转账任务
	abstract class Runner implements Runnable {
		final int maxMoney;
		final int number;
		private final Random random = new Random();
		private final AtomicInteger count = new AtomicInteger();

		Runner(int maxMoney, int number) {
			this.maxMoney = maxMoney;
			this.number = number;
		}

		@Override
		public void run() {
			while(count.getAndIncrement() < number) {
				int from = random.nextInt(account.length);
				int to;
				while ((to = random.nextInt(account.length)) == from)
					;
				int money = random.nextInt(maxMoney);
				doTransfer(from, to, money);
			}
		}

		abstract void doTransfer(int from, int to, int money);
	}

	// 不加锁的转账
	class NonLockRunner extends Runner {
		NonLockRunner(int maxMoney, int number) {
			super(maxMoney, number);
		}

		@Override
		void doTransfer(int from, int to, int money) {
			transfer(from, to, money);
		}
	}

	// synchronized的转账
	class SynchronizedRunner extends Runner {
		SynchronizedRunner(int maxMoney, int number) {
			super(maxMoney, number);
		}

		@Override
		synchronized void doTransfer(int from, int to, int money) {
			transfer(from, to, money);
		}
	}

	// ReentrantLock的转账
	class ReentrantLockRunner extends Runner {
		private final ReentrantLock lock = new ReentrantLock();

		ReentrantLockRunner(int maxMoney, int number) {
			super(maxMoney, number);
		}

		@Override
		void doTransfer(int from, int to, int money) {
			lock.lock();
			try {
				transfer(from, to, money);
			} finally {
				lock.unlock();
			}
		}
	}

	// KeyLock的转账
	class KeyLockRunner extends Runner {
		private final KeyLock<Integer> lock = new KeyLock<Integer>();

		KeyLockRunner(int maxMoney, int number) {
			super(maxMoney, number);
		}

		@Override
		void doTransfer(int from, int to, int money) {
			Integer[] keys = new Integer[] {from, to};
			Arrays.sort(keys);
			lock.lock(keys);
			try {
				transfer(from, to, money);
			} finally {
				lock.unlock(keys);
			}
		}
	}
}

 最最重要的测试结果

 

 

(8线程对100个账户随机转账总共10000次):

       类型:NonLockRunner(不加锁)
       耗时:2482ms
       初始总金额:1000000
       终止总金额:998906(无法保证原子性)

       类型:SynchronizedRunner(加synchronized锁)
       耗时:20872ms
       初始总金额:1000000
       终止总金额:1000000

       类型:ReentrantLockRunner(加ReentrantLock锁)
       耗时:21588ms
       初始总金额:1000000
       终止总金额:1000000

       类型:KeyLockRunner(加KeyLock锁)
       耗时:2831ms
       初始总金额:1000000
       终止总金额:1000000

 

转载:http://blog.csdn.net/icebamboo_moyun/article/details/9391915

© 著作权归作者所有

longload
粉丝 3
博文 66
码字总数 104702
作品 0
成都
程序员
私信 提问
分布式锁策略

原文链接:http://www.jianshu.com/p/535efcab356d 方案一:数据库乐观锁 乐观锁通常实现基于数据版本(version)的记录机制实现的,比如有一张红包表(tbonus),有一个字段(leftcount)记录礼...

17099933344
2017/08/04
0
0
Innodb中常见SQL语句设置的锁类型

1、select … from 除了serializable隔离级别,这种SQL都是一致性非锁定读,不加锁;在serializable级别,这种SQL加next-key锁。 2、select … from … lock in share mode 这种SQL加S类型的...

未央&amp;
04/14
0
0
redis加锁的几种实现

1. redis加锁分类 redis能用的的加锁命令分表是、、 2. 第一种锁命令 这种加锁的思路是, key 不存在,那么 key 的值会先被初始化为 0 ,然后再执行 INCR 操作进行加一。 然后其它用户在执行...

宇的季节
2018/02/26
0
0
如何基于Redis实现分布式锁?

选用Redis实现分布式锁原因 Redis有很高的性能 Redis命令对此支持较好,实现起来比较方便 使用命令介绍 SETNX SETNX key val 当且仅当key不存在时,set一个key为val的字符串,返回1;若key存...

青梅煮酒论英雄
01/22
0
0
Spring-data-redis + redis 分布式锁(一)

分布式锁的解决方式 基于数据库表做乐观锁,用于分布式锁。(适用于小并发) 使用memcached的add()方法,用于分布式锁。 使用memcached的cas()方法,用于分布式锁。(不常用) 使用redis的setnx...

xiaolyuh
2017/11/15
0
0

没有更多内容

加载失败,请刷新页面

加载更多

前嗅教程:如何获取精准客源,提高销量

经常有人问嗅嗅,我是XX行业的,大数据能帮我做什么? • 可以给我带来客源吗? • 可以提高我的销量吗? • 可以增加我的利润吗? 今天嗅嗅就以生鲜供货为例,为大家讲一讲外卖平台那些事~...

forespider
25分钟前
1
0
浮窗插件

<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>移动窗口</title> <style> body { margin: 0; padding: 0; width: 100%; height: 1000px; background: #eee; } /*示......

流年那么伤
29分钟前
2
0
关于 Jenkins master 共享 JENKINS_HOME 目录的实验

本文首发于:Jenkins 中文社区 作者:翟志军 审校:王冬辉,linuxsuren Jenkins master 的高可用是个老大难的问题。和很多人一样,笔者也想过两个 Jenkins master 共享同一个 JENKINS_HOME 的...

Jenkins中文社区
37分钟前
3
0
【重构】Spring Cloud OAuth 无Token调用源码封装

背景 重构-改善既有代码的设计,重构的目的是是软件更容易被理解和修改。 书接上回Spring Security OAuth 微服务内部Token传递的源码解析,本篇主要无token 调用过程中,代码的不断完善及其重...

冷冷gg
43分钟前
26
0
watchOS更新后 Apple Watch 4心电图功能已开始支持欧洲用户

苹果在发布 Apple Watch 4 系列时也发布了 ECG(心电图)功能,但这项功能仅适用于在美版 Apple Watch。对于其他地区的用户来说,访问该功能的唯一途径是在美国购买该设备。不过当 watchOS ...

linuxCool
52分钟前
3
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部