golang实现抢红包

func generateRedPackage(totalAmount float64, num int32, redPackageType int32) ([]*RedPackage, error) { // 验证红包金额和个数是否合法 if totalAmount <= 0 || num <= 0 { return nil, errors.New("红包金额或个数不能小于等于0") } // 计算平均值 avgAmount := totalAmount / float64(num) // 生成红包码 redPackageCodes := make([]string, num) for i := 0; i < len(redPackageCodes); i++ { code := generateRedPackageCode() redPackageCodes[i] = code } // 分配红包金额 redPackages := make([]*RedPackage, num) for i := 0; i < len(redPackages); i++ { redPackages[i] = &RedPackage{ Code: redPackageCodes[i], Amount: avgAmount, RedPackageType: redPackageType, } totalAmount -= avgAmount if i == len(redPackages) - 1 { redPackages[i].Amount += totalAmount break } redPackages[i].Amount += avgAmount } // 存入数据库和 Redis 缓存中 return redPackages, nil }

func getRedPackage(code string, userId int64) (*RedPackage, error) { // 先从缓存中获取该红包的金额 rc := redisMgr.RedisClient() redPackageAmount, err := rc.RPop(code).Result() if err != nil { return nil, errors.New("红包已经被抢完了") } // 判断用户是否已经抢到过该红包 key := fmt.Sprintf("%s:%d", code, userId) result, err := rc.Exists(key).Result() if err != nil || result == 1 { return nil, errors.New("您已经抢过该红包了") } // 生成抢红包记录 record := &RedPackageRecord{ RedPackageCode: code, UserId: userId, Amount: redPackageAmount, CreateTime: time.Now(), } // 将抢红包记录和金额存入 MySQL 数据库中 err = dbMgr.SaveRedPackageRecord(record) if err != nil { return nil, err } // 将金额存入用户账户中 err = dbMgr.UpdateUserAmount(userId, redPackageAmount) if err != nil { return nil, err } // 返回抢到的红包金额 redPackage := &RedPackage{ Code: code, Amount: redPackageAmount, } return redPackage, nil }