<?php

namespace traits;

use bw\Redis;


/**
 * 缓存
 */
trait CacheTrait
{


    private $cache_lock_error_msg = "获取锁失败！";

    private $cache_lock_thorws_excption = true;

    public function setCacheLockThorwsExcption($cache_lock_thorws_excption){
        $this->cache_lock_thorws_excption = $cache_lock_thorws_excption;
        return $this;
    }


    public function getCacheLockThorwsExcption(){
        return $this->cache_lock_thorws_excption;
    }


    public function hasRedis($is_interrupt = false) {
        $error_msg = '';
        try {
            $redis = $this->getRedis();

            // 检测连接是否正常
            $redis->ping();
        } catch (\BadFunctionCallException $e) {
            // 缺少扩展
            $error_msg = $e->getMessage() ? $e->getMessage() : "缺少 redis 扩展";
        } catch (\RedisException $e) {
            // 连接拒绝
            \think\Log::write('redis connection redisException fail: ' . $e->getMessage());
            $error_msg = $e->getMessage() ? $e->getMessage() : "redis 连接失败";
        } catch (\Exception $e) {
            // 异常
            \think\Log::write('redis connection fail: ' . $e->getMessage());
            $error_msg = $e->getMessage() ? $e->getMessage() : "redis 连接异常";
        }

        if ($error_msg) {
            if ($is_interrupt || $this->cache_lock_thorws_excption) {
                throw new \Exception($error_msg);
            } else {
                $this->setCacheLockErrorMsg($error_msg);
                return false;
            }
        }

        return true;
    }


    public function getRedis() {
        if (!isset($GLOBALS['SPREDIS'])) {
            $GLOBALS['SPREDIS'] = (new Redis())->getRedis();
        }

        return $GLOBALS['SPREDIS'];
    }


    public function getLock($key,$suffix = "-lock-suffix", $timeout = 120){
        $this->hasRedis(true);
        $redis = $this->getRedis();
        $hashKey = $key.$suffix;
//        $redis->del($hashKey);
        //判断锁是否存在，如果存在，返回错误
        if ($redis->EXISTS($hashKey)){
            if($this->cache_lock_thorws_excption){
                throw new \Exception($this->cache_lock_error_msg);
            }else{
                return false;
            }

        }
        //如果不存在创建锁并返回
        $redis->set($hashKey, 1,$timeout);
        return true;
    }


    public function setCacheLockErrorMsg($cache_lock_error_msg){
        $this->cache_lock_error_msg = $cache_lock_error_msg;
        return $this;
    }


    public function getCacheLockErrorMsg(){
        return $this->cache_lock_error_msg;
    }


    public function freeLock($key,$suffix = "-lock-suffix"){
        $this->hasRedis(true);
        $redis = $this->getRedis();
        if(!$key || !$suffix) return true;
        $hashKey = $key.$suffix;
        if (!$redis->EXISTS($hashKey)) return true;
        $redis->del($hashKey);
        return true;
    }


}