秒殺功能

秒殺場景現在已經非常常見了，各種電商平臺都有秒殺的產品，接下來我們模擬一個秒殺的專案，最終能夠確保高併發下的執行緒安全。介面比較簡單，但是功能基本實現。

介面

點選“秒殺點我”按鈕後臺就會輸出秒殺結果。

第一版

使用Redis快取資料庫，使用一個key-value儲存秒殺商品數量，使用set集合儲存秒殺成功的使用者。我們以商品0101為示例，設定商品的初始數量為200件。不考慮併發問題，實現功能。

html、jsp、servlet檔案不重要省略。

package com.redis.secondskill;
 
import java.util.List;
 
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.Transaction;
 
public class SS0 {
 public static boolean doSecKill(String uid,String prodid) {
 JedisPool jedisPool = JedisPollTool.getInstance();
 Jedis jedis = jedisPool.getResource();
 String productCountStr = "sec:"+prodid+":count";
 String productUserStr = "sec:"+prodid+":user";
 String productCount = jedis.get(productCountStr);
 if(null == productCount) {
 System.out.println("秒殺還沒有開始");
 JedisPollTool.distroy(jedisPool,jedis);
 return false;
 }
 if(jedis.sismember(productUserStr,uid)) {
 System.out.println(uid + "使用者已經秒殺成功");
 JedisPollTool.distroy(jedisPool,jedis);
 return false;
 }
 int prodCount = Integer.parseInt(productCount);
 if(prodCount <= 0) {
 System.out.println("秒殺結束");
 JedisPollTool.distroy(jedisPool,jedis);
 return false;
 }
 jedis.decr(productCountStr);
 jedis.sadd(productUserStr,uid);
 
 JedisPollTool.distroy(jedisPool,jedis);
 System.out.println(uid + "秒殺成功");
 return true;
 }
}

使用linux httpd-tools工具進行併發測試。

ab -n 1000 -c 200 -p /test/file.txt -T "application/x-www-form-urlencoded" 192.168.0.101:8080/redis-demo/ss

結果

從結果大致來看，沒有什麼問題，來檢視一個後臺Redis的資料

秒殺的結果裡面居然有負數，證明賣超了。

第二版

使用Redis的事務，保證沒有超賣的情況發生。

package com.redis.secondskill;
 
import java.util.List;
 
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.Transaction;
 
public class SS1 {
 public static boolean doSecKill(String uid,String prodid) {
 JedisPool jedisPool = JedisPollTool.getInstance();
 Jedis jedis = jedisPool.getResource();
 String productCountStr = "sec:"+prodid+":count";
 String productUserStr = "sec:"+prodid+":user";
 jedis.watch(productCountStr); //開始監視
 String productCount = jedis.get(productCountStr);
 if(null == productCount) {
 System.out.println("秒殺還沒有開始");
 JedisPollTool.distroy(jedisPool,jedis);
 return false;
 }
 Transaction transaction = jedis.multi();
 transaction.decr(productCountStr);
 transaction.sadd(productUserStr,uid);
 List<Object> exec = transaction.exec();
 
 
 if(exec == null || exec.size() == 0) {
 System.out.println("秒殺失敗，稍後重試");
 JedisPollTool.distroy(jedisPool,jedis);
 return false;
 }
 JedisPollTool.distroy(jedisPool,jedis);
 System.out.println(uid + "秒殺成功");
 return true;
 }
}

結果

由於使用了watch和事務，每次的併發執行緒訪問中只有一個執行緒能夠提交成功，可以保證不出現超賣的現象，但是對於一些使用者來說是極其不公平的。

第三版

使用Lua指令碼來實現，因為Redis是單執行緒的，又是C語言編寫的，可以使用Lua呼叫Redis的命令，Lua會具有排他性，所以能夠保證安全。

package com.redis.secondskill;
 
import java.util.HashSet;
import java.util.Set;
 
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
 
public class SS2 {
 
 static String luaScript ="local userid=KEYS[1];\r\n" + 
 "local prodid=KEYS[2];\r\n" + 
 "local qtkey='sec:'..prodid..\":count\";\r\n" + 
 "local usersKey='sec:'..prodid..\":user\";\r\n" + 
 "local userExists=redis.call(\"sismember\",usersKey,userid);\r\n" + 
 "if tonumber(userExists)==1 then \r\n" + 
 " return 2;\r\n" + 
 "end\r\n" + 
 "local num = redis.call(\"get\",qtkey);\r\n" + 
 "if tonumber(num)<=0 then \r\n" + 
 " return 0;\r\n" + 
 "else \r\n" + 
 " redis.call(\"decr\",qtkey);\r\n" + 
 " redis.call(\"sadd\",userid);\r\n" + 
 "end\r\n" + 
 "return 1" ;
 
 public static boolean doSecKill(String uid,String prodid) {
 JedisPool jedisPool = JedisPollTool.getInstance();
 Jedis jedis = jedisPool.getResource();
 String sha1 = jedis.scriptLoad(luaScript);
 
 Object result= jedis.evalsha(sha1,2,uid,prodid);
 
 String reString=String.valueOf(result);
 if ("0".equals( reString ) ) {
 System.err.println("已搶空！！");
 }else if("1".equals( reString ) ) {
 System.out.println(uid + "搶購成功！！！！");
 }else if("2".equals( reString ) ) {
 System.err.println("該使用者已搶過！！");
 }else{
 System.err.println("搶購異常！！");
 }
 JedisPollTool.distroy(jedisPool,jedis);
 return true;
 
 }
}

結果

這才是我們最希望看到的結果！

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。