<?php


namespace ZM\Store\Lock;


use Swoole\Coroutine;
use Swoole\Coroutine\System;
use Swoole\Table;

class SpinLock
{
    /** @var null|Table */
    private static $kv_lock = null;

    private static $delay = 1;

    public static function init($key_cnt, $delay = 1) {
        self::$kv_lock = new Table($key_cnt, 0.7);
        self::$delay = $delay;
        self::$kv_lock->column('lock_num', Table::TYPE_INT, 8);
        return self::$kv_lock->create();
    }

    public static function lock(string $key) {
        while (($r = self::$kv_lock->incr($key, 'lock_num')) > 1) { //此资源已经被锁上了
            if (Coroutine::getCid() != -1) System::sleep(self::$delay / 1000);
            else usleep(self::$delay * 1000);
        }
    }

    public static function tryLock(string $key) {
        if (($r = self::$kv_lock->incr($key, 'lock_num')) > 1) {
            return false;
        }
        return true;
    }

    public static function unlock(string $key) {
        return self::$kv_lock->set($key, ['lock_num' => 0]);
    }

    public static function transaction(string $key, callable $function) {
        SpinLock::lock($key);
        $function();
        SpinLock::unlock($key);
    }
}