modify

System/Crontab/Zookeeper.php

@@ -0,0 +1,108 @@
+<?php
+
+
+namespace Snowflake\Crontab;
+
+
+use Exception;
+use Snowflake\Process\Process;
+use Snowflake\Snowflake;
+use Swoole\Coroutine\WaitGroup;
+use Swoole\Coroutine\Channel;
+use Swoole\Timer;
+
+/**
+ * Class Zookeeper
+ * @package Snowflake\Process
+ */
+class Zookeeper extends Process
+{
+
+
+    private Channel $channel;
+    private WaitGroup $waitGroup;
+
+
+    /** @var Crontab[] $names */
+    public array $names = [];
+
+
+    public array $scores = [];
+    public array $timers = [];
+
+
+    /**
+     * @param \Swoole\Process $process
+     * @throws Exception
+     */
+    public function onHandler(\Swoole\Process $process): void
+    {
+        $crontab = Snowflake::app()->get('crontab');
+        $crontab->clearAll();
+        if (Snowflake::getPlatform()->isLinux()) {
+            name($this->pid, 'Crontab zookeeper.');
+        }
+        Timer::tick(1000, function () {
+            [$range, $redis] = $this->loadCrotabTask();
+            $server = Snowflake::app()->getSwoole();
+            $setting = $server->setting['worker_num'];
+            foreach ($range as $value) {
+                $this->dispatch($server, $redis, $setting, $value);
+            }
+            $redis->release();
+        });
+    }
+
+
+    /**
+     * @param $server
+     * @param $redis
+     * @throws \Exception
+     */
+    private function dispatch($server, $redis, $setting, $value)
+    {
+        $server->sendMessage(swoole_serialize([
+            'action' => 'crontab', 'handler' => swoole_unserialize($redis->get('crontab:' . md5($value)))
+        ]), random_int(0, $setting - 1));
+        $redis->del('crontab:' . md5($value));
+    }
+
+
+    /**
+     * @return array
+     * @throws \Exception
+     */
+    private function loadCrotabTask()
+    {
+        $redis = Snowflake::app()->getRedis();
+
+        $startTime = time();
+
+        $redis->multi();
+        $range = $redis->zRangeByScore(Producer::CRONTAB_KEY, '0', (string)$startTime);
+        $redis->zRemRangeByScore(Producer::CRONTAB_KEY, '0', (string)$startTime);
+        $redis->exec();
+
+        return [$range, $redis];
+    }
+
+
+    /**
+     * @param string $name
+     */
+    public function clear(string $name)
+    {
+        if (!isset($this->names[$name])) {
+            return;
+        }
+        $timers = $this->timers[$name];
+
+        $search = array_search($name, $this->scores[$timers]);
+        if ($search !== false) {
+            unset($this->scores[$timers][$search]);
+        }
+        unset($this->timers[$name], $this->names[$name]);
+    }
+
+
+}