package com.example.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.util.ArrayList;
import java.util.List;

/**
 * @Auther: 長頸鹿
 * @Date: 2021/08/21/14:09
 * @Description:
 */
@Service
@Slf4j
public class SpopRandomSetService {

    @Autowired
    private RedisTemplate redisTemplate;

    private static final String SPOP_USER_SETS = "pop:user:set";

    // 把所有員工全部添加到集合列表中
    @PostConstruct
    public void initData(){
        log.info("初始化獎品等級信息...");
        // 判斷集合是否已經(jīng)存在
        boolean flag = this.redisTemplate.hasKey(SPOP_USER_SETS);
        // 防止作弊
        if (!flag) {
            // 獲取所有員工的信息
            List<Integer> initDataList = initDataList();
            // 把員工信息寫入到redis中 sadd key data
            initDataList.forEach(data -> this.redisTemplate.opsForSet().add(SPOP_USER_SETS, data));
        }
    }

    // 模擬100用戶抽獎
    private List<Integer> initDataList() {
        // todo : 從數(shù)據(jù)庫里面來，把公司里面所有的員工從數(shù)據(jù)表中全部查詢出來
        List<Integer> listData = new ArrayList<>();
        for (int i = 0; i < 100; i++) {
            listData.add(i + 1);
        }
        return listData;
    }

}

// 隨機(jī)抽取用戶
    public int start(){
        return (int)redisTemplate.opsForSet().pop(SPOP_USER_SETS);
    }

package com.example.controller;

import com.example.service.SpopRandomSetService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @Auther: 長頸鹿
 * @Date: 2021/08/21/14:13
 * @Description: 抽獎接口測試
 */
@RestController
public class SpopRandomSetController {

    @Autowired
    private SpopRandomSetService spopRandomSetService;

    @PostMapping("/sPop/random/user")
    public int start() {
        return spopRandomSetService.start();
    }

}

# 查詢集合成員
smembers pop:user:Set
# 查詢集合的長度變化
scard pop:user:Set