import jakarta.annotation.Resource;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.time.format.DateTimeFormatterBuilder;
import java.time.temporal.ChronoField;
import java.util.Collections;

@RestController
public class distr_lock {
    @Resource
    private RedisTemplate<String,String> redisTemplate;


    @GetMapping("/order")
    public void order() {
        for (int i = 0; i < 50; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for (int i = 0; i < 20; i++) {
                        String s = generateOrderId();
                        System.out.println(s);
                    }
                }
            }).start();
        }
    }

    /**
     * 我們可以采用這樣一種策略，
     * 時(shí)間精確到秒+7位數(shù)的自增數(shù)字
     * 這樣可以實(shí)現(xiàn)同一個(gè)用戶在同一秒下單的不超過(guò)1000萬(wàn)單
     * 然后使用時(shí)間在redis創(chuàng)建一個(gè)key,進(jìn)行自增，然后30秒過(guò)期
     */
    public String generateOrderId(){
        DateTimeFormatter formatter = new DateTimeFormatterBuilder()
                .appendValue(ChronoField.YEAR, 4) // 年份，4位數(shù)字
                .appendValue(ChronoField.MONTH_OF_YEAR, 2) // 月份，2位數(shù)字
                .appendValue(ChronoField.DAY_OF_MONTH, 2) // 日期，2位數(shù)字
                .appendValue(ChronoField.HOUR_OF_DAY, 2) // 小時(shí)，2位數(shù)字
                .appendValue(ChronoField.MINUTE_OF_HOUR, 2) // 分鐘，2位數(shù)字
                .appendValue(ChronoField.SECOND_OF_MINUTE, 2) // 秒，2位數(shù)字
                .toFormatter();

        // 獲取當(dāng)前時(shí)間的字符串表示，不使用-或:作為分隔符
        String currentDateTime = LocalDateTime.now().format(formatter);
//        Long count = getAndExpireCounter(currentDateTime);
        Long count = getAndExpireCounterByLua(currentDateTime);
        String format = String.format("%07d", count);
        return currentDateTime + format;
    }



    public Long getAndExpireCounter(String key) {
        // 獲取Redis連接，以便執(zhí)行事務(wù)
        Long increment = 0L;
        increment = redisTemplate.opsForValue().increment(key);
        if (increment!=null && increment == 1) {
            // 設(shè)置過(guò)期時(shí)間,這里并沒(méi)有使用事務(wù)或者lua腳本
            // 因?yàn)槲矣X(jué)得上面的自增操作是原子性的，也就是肯定會(huì)得到一個(gè)值為1
            // 如果說(shuō)redis服務(wù)崩了,到這一步?jīng)]有設(shè)置好過(guò)期時(shí)間，那么我們整個(gè)
            // 服務(wù)也會(huì)崩潰，這個(gè)key就會(huì)一直存在于服務(wù)器，會(huì)出現(xiàn)這個(gè)問(wèn)題，為了嚴(yán)謹(jǐn)
            // 最好使用lua腳本編寫(xiě)，項(xiàng)目上線會(huì)保證更可靠的效果
            redisTemplate.expire(key, 30, java.util.concurrent.TimeUnit.SECONDS);
        }
        // 獲取自增后的值
        return increment;
    }
	
	/**
     * 整個(gè)腳本的作用是：
     * 1.檢查一個(gè)鍵是否存在于Redis中。
     * 2.如果鍵不存在，則設(shè)置該鍵的值為1，并為其設(shè)置60秒的過(guò)期時(shí)間，然后返回1。
     * 3.如果鍵已經(jīng)存在，則將其值增加1，并返回增加后的新值。
     * Redis執(zhí)行Lua腳本時(shí)是原子的，這意味著在執(zhí)行這個(gè)腳本的過(guò)程中，
     * 不會(huì)有其他腳本或命令同時(shí)修改這個(gè)鍵，從而保證了操作的原子性和一致性。
     * @param key
     * @return
     */
    public Long getAndExpireCounterByLua(String key) {
        // 定義Lua腳本
        String script =
                "if redis.call('EXISTS', KEYS[1]) == 0 then " +
                        "   redis.call('SET', KEYS[1], 1) " +
                        "   redis.call('EXPIRE', KEYS[1], 30) " +
                        "   return 1 " +
                        "else " +
                        "   return redis.call('INCR', KEYS[1]) " +
                        "end";

        // 使用Redis的腳本執(zhí)行功能
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptText(script);
        redisScript.setResultType(Long.class);

        // 執(zhí)行腳本
        return redisTemplate.execute(redisScript, Collections.singletonList(key));
    }