在Redis的事務中,WATCH命令可用於提供CAS(check-and-set)功能。假設我們通過WATCH命令在事務執行之前監控了多個Keys,倘若在WATCH之后有任何Key的值發生了變化,EXEC命令執行的事務都將被放棄,同時返回Null multi-bulk應答以通知調用者事務執行失敗。例如,我們再次假設Redis中並未提供incr命令來完成鍵值的原子性遞增,如果要實現該功能,我們只能自行編寫相應的代碼。其偽碼如下:
val = GET mykey
val = val + 1
SET mykey $val
以上代碼只有在單連接的情況下才可以保證執行結果是正確的,因為如果在同一時刻有多個客戶端在同時執行該段代碼,那么就會出現多線程程序中經常出現的一種錯誤場景--競態爭用(race condition)。比如,客戶端A和B都在同一時刻讀取了mykey的原有值,假設該值為10,此后兩個客戶端又均將該值加一后set回Redis服務器,這樣就會導致mykey的結果為11,而不是我們認為的12。為了解決類似的問題,我們需要借助WATCH命令的幫助,見如下代碼:
WATCH mykey
val = GET mykey
val = val + 1
MULTI
SET mykey $val
EXEC
和此前代碼不同的是,新代碼在獲取mykey的值之前先通過WATCH命令監控了該鍵,此后又將set命令包圍在事務中,這樣就可以有效的保證每個連接在執行EXEC之前,如果當前連接獲取的mykey的值被其它連接的客戶端修改,那么當前連接的EXEC命令將執行失敗。這樣調用者在判斷返回值后就可以獲悉val是否被重新設置成功。
根據這樣的思路,我們在JAVA下進行實現:
新建一個項目,首先引入JAVA的redis操作庫:Jedis,這里用的是jedis-2.9.0.jar
新建一個類:MyRedistest.class做線程操作
package com.myredistest;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import redis.clients.jedis.Jedis;
/**
* redis
*
* @author 10255_000
*
*/
public class MyRedistest {
public static void main(String[] args) {
final String watchkeys = "watchkeys";
ExecutorService executor = Executors.newFixedThreadPool(20); //20個線程池並發數
final Jedis jedis = new Jedis("192.168.56.101", 6379);
jedis.set(watchkeys, "100");//設置起始的搶購數
// jedis.del("setsucc", "setfail");
jedis.close();
for (int i = 0; i < 1000; i++) {//設置1000個人來發起搶購
executor.execute(new MyRunnable("user"+getRandomString(6)));
}
executor.shutdown();
}
public static String getRandomString(int length) { //length是隨機字符串長度
String base = "abcdefghijklmnopqrstuvwxyz0123456789";
Random random = new Random();
StringBuffer sb = new StringBuffer();
for (int i = 0; i < length; i++) {
int number = random.nextInt(base.length());
sb.append(base.charAt(number));
}
return sb.toString();
}
}
建一個類:MyRunnable.class 實現Runnable做線程操作:
package com.myredistest;
import java.util.List;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.Transaction;
public class MyRunnable implements Runnable {
String watchkeys = "watchkeys";// 監視keys
Jedis jedis = new Jedis("192.168.56.101", 6379);
String userinfo;
public MyRunnable() {
}
public MyRunnable(String uinfo) {
this.userinfo=uinfo;
}
@Override
public void run() {
try {
jedis.watch(watchkeys);// watchkeys
String val = jedis.get(watchkeys);
int valint = Integer.valueOf(val);
if (valint <= 100 && valint>=1) {
Transaction tx = jedis.multi();// 開啟事務
// tx.incr("watchkeys");
tx.incrBy("watchkeys", -1);
List<Object> list = tx.exec();// 提交事務,如果此時watchkeys被改動了,則返回null
if (list == null ||list.size()==0) {
String failuserifo = "fail"+userinfo;
String failinfo="用戶:" + failuserifo + "商品爭搶失敗,搶購失敗";
System.out.println(failinfo);
/* 搶購失敗業務邏輯 */
jedis.setnx(failuserifo, failinfo);
} else {
for(Object succ : list){
String succuserifo ="succ"+succ.toString() +userinfo ;
String succinfo="用戶:" + succuserifo + "搶購成功,當前搶購成功人數:"
+ (1-(valint-100));
System.out.println(succinfo);
/* 搶購成功業務邏輯 */
jedis.setnx(succuserifo, succinfo);
}
}
} else {
String failuserifo ="kcfail" + userinfo;
String failinfo1="用戶:" + failuserifo + "商品被搶購完畢,搶購失敗";
System.out.println(failinfo1);
jedis.setnx(failuserifo, failinfo1);
// Thread.sleep(500);
return;
}
} catch (Exception e) {
e.printStackTrace();
} finally {
jedis.close();
}
}
}
執行MyRedistest ,查看redis中插入的key值
