go redis 奖池递减

4 篇文章 0 订阅

今天看一个老师傅讲redis,关于redis高并发的情况下实现发奖的设计,中间讲到把奖池的减放在判断的前面可以避免奖品的超发,顿时觉得说的有道理,但是自己敲下来,又不是那么回事,可能我没理解他将的方法。自己写了一下:

package main

import (
	"../config"
	"../pool"
	"flag"
	"fmt"
	"github.com/garyburd/redigo/redis"
	"sync"
	"utils"
)

const (
	moneyPool = "10000" //奖金池现金
)

var (
	wg    sync.WaitGroup
	redPo *redis.Pool
	mx    sync.Mutex
)

func init() {
	flag.Parse()
	redPo = pool.NewRedisPool(*config.RedisServer, *config.RedisPassword)
	redCon := redPo.Get()
	defer redCon.Close()
	redCon.Do("set", "moneyPool", moneyPool) //将奖金池设置到redis
}

func main() {
	defer func() {
		redCon := redPo.Get()
		redCon.Close()
		i, _ := redis.Int(redCon.Do("get", "commander"))
		fmt.Printf("moneyPool 剩余现金 %v", i)
	}()

	wg.Add(10)
	for i := 0; i < 10; i++ {
		go handleMoney()
	}
	wg.Wait()
}

func handleMoney() {
	redCon := redPo.Get()
	defer redCon.Close()

	for {
		b := decrMoneyPool(redCon)
		if b {
			break
		}
	}
	wg.Done()
}

func decrMoneyPool(conn redis.Conn) bool {
	mx.Lock()
	reply, err := redis.Int(conn.Do("decrby", "moneyPool", 1))
	if reply < 1 {
		mx.Unlock() //这里一定要释放锁,否则调用的线程无法结束
		return true
	}
	mx.Unlock()
	utils.HandleError(err, "Do get")
	fmt.Println("reply", reply)
	return false
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值