• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

redis 系列(十)java结合redis+lua 实现抢红包经典案例

原作者: [db:作者] 来自: [db:来源] 收藏 邀请

使用lua脚本来实现一个抢红包的过程,lua具有原子特性,可以避免数据并发时多线程同时操作的问题

java代码结合lua实现抢红包案例

单机版redis

package bhz.redis01;

import java.util.Random;
import java.util.concurrent.CountDownLatch;

import com.alibaba.fastjson.JSONObject;

import redis.clients.jedis.Jedis;

public class TestEval { 
	
    static String host = "192.168.1.115"; 
    static int port = 6379; 
    
    static int honBaoCount = 10000;  
      
    static int threadCount = 10;  
      
    static String hongBaoList = "hongBaoList"; 		//L1 为消费的红包 预先生成的红包 {id、money、userid(暂时未空)} 
    static String hongBaoConsumedList = "hongBaoConsumedList"; 		//L2 已消费的红包队列 
    static String hongBaoConsumedMap = "hongBaoConsumedMap";  		//去重的map
      
    static Random random = new Random();  
      
    /**
     * 		-- 函数:尝试获得红包,如果成功,则返回json字符串,如果不成功,则返回空  
	 *		-- 参数:红包队列名, 已消费的队列名,去重的Map名,用户ID  
	 *		-- 返回值:nil 或者 json字符串,包含用户ID:userId,红包ID:id,红包金额:money  
     *
	 *		-- 如果用户已抢过红包,则返回nil  
 	 *		if redis.call('hexists', KEYS[3], KEYS[4]) ~= 0 then  
	 * 		  return nil  
	 * 		else  
	 * 		  -- 先取出一个小红包  
	 *		  local hongBao = redis.call('rpop', KEYS[1]);  
	 *		  if hongBao then  
	 *		    local x = cjson.decode(hongBao);  
	 *		    -- 加入用户ID信息  
	 *		    x['userId'] = KEYS[4];  
	 *		    local re = cjson.encode(x);  
	 *		    -- 把用户ID放到去重的set里  
	 *		    redis.call('hset', KEYS[3], KEYS[4], KEYS[4]);  
	 *		    -- 把红包放到已消费队列里  
	 *		    redis.call('lpush', KEYS[2], re);  
	 *		    return re;  
	 *		  end  
	 *		end  
	 *		return nil 
     */
    static String tryGetHongBaoScript =   
//          "local bConsumed = redis.call('hexists', KEYS[3], KEYS[4]);\n"  
//          + "print('bConsumed:' ,bConsumed);\n"  
            "if redis.call('hexists', KEYS[3], KEYS[4]) ~= 0 then\n"  
            + "return nil\n"  
            + "else\n"  
            + "local hongBao = redis.call('rpop', KEYS[1]);\n"  
//          + "print('hongBao:', hongBao);\n"  
            + "if hongBao then\n"  
            + "local x = cjson.decode(hongBao);\n"  
            + "x['userId'] = KEYS[4];\n"  
            + "local re = cjson.encode(x);\n"  
            + "redis.call('hset', KEYS[3], KEYS[4], KEYS[4]);\n"  
            + "redis.call('lpush', KEYS[2], re);\n"  
            + "return re;\n"  
            + "end\n"  
            + "end\n"  
            + "return nil";  
      
    public static void main(String[] args) throws InterruptedException {  
        //generateTestData();  
        testTryGetHongBao();  
    }  
      
    static public void generateTestData() throws InterruptedException {  
        Jedis jedis = new Jedis(host, port); 
        jedis.flushAll();  
        final CountDownLatch latch = new CountDownLatch(threadCount);  
        for(int i = 0; i < threadCount; ++i) {  
            final int temp = i;  
            Thread thread = new Thread() {  
                public void run() {  
                    Jedis jedis = new Jedis(host);  
                    int per = honBaoCount/threadCount;  
                    JSONObject object = new JSONObject();  
                    for(int j = temp * per; j < (temp+1) * per; j++) {  
                        object.put("id", j);  
                        object.put("money", j);  
                        jedis.lpush(hongBaoList, object.toJSONString());  
                    }  
                    latch.countDown();  
                }  
            };  
            thread.start();  
        }  
        latch.await();  
        
        jedis.close();
    }  
      
    static public void testTryGetHongBao() throws InterruptedException {  
        final CountDownLatch latch = new CountDownLatch(threadCount);  
        
        long startTime = System.currentTimeMillis();
        System.err.println("start:" + startTime);  
        
        for(int i = 0; i < threadCount; ++i) {  
            final int temp = i;  
            Thread thread = new Thread() {  
                public void run() {  
                    Jedis jedis = new Jedis(host, port);  
                    String sha = jedis.scriptLoad(tryGetHongBaoScript);  
                    int j = honBaoCount/threadCount * temp;  
                    while(true) {  
                    	//抢红包方法
                        Object object = jedis.eval(tryGetHongBaoScript, 4, 
                        		hongBaoList/*预生成的红包队列*/, 
                        		hongBaoConsumedList, /*已经消费的红包队列*/
                        		hongBaoConsumedMap, /*去重的map*/
                        		"" + j  /*用户id*/
                        		);  
                        j++;  
                        if (object != null) {  
                        	//do something...
//                          System.out.println("get hongBao:" + object);  
                        }else {  
                            //已经取完了  
                            if(jedis.llen(hongBaoList) == 0)  
                                break;  
                        }  
                    }  
                    latch.countDown();  
                }  
            };  
            thread.start();  
        }  
          
        latch.await();  
        long costTime =  System.currentTimeMillis() - startTime;
        
        System.err.println("costTime:" + costTime);  
    }  
}  

看懂的转发!


鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
上一篇:
(一)Lua脚本语言入门发布时间:2022-07-22
下一篇:
C遍历Lua中的Table发布时间:2022-07-22
热门推荐
热门话题
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap