From 5797d87068de92ed61798af334c07351c88bb8d0 Mon Sep 17 00:00:00 2001
From: "as2252258@163.com" <as22522582>
Date: Sun, 28 Feb 2021 17:20:59 +0800
Subject: [PATCH] modify

---
 HttpServer/Events/OnWorkerStart.php  |  2 ++
 System/Abstracts/BaseApplication.php | 13 ++++++++++++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/HttpServer/Events/OnWorkerStart.php b/HttpServer/Events/OnWorkerStart.php
index 9f921a34..6353edaf 100644
--- a/HttpServer/Events/OnWorkerStart.php
+++ b/HttpServer/Events/OnWorkerStart.php
@@ -35,6 +35,8 @@ class OnWorkerStart extends Callback
     public function onHandler(Server $server, int $worker_id): void
     {
         Coroutine::set(['enable_deadlock_check' => false]);
+        Snowflake::app()->stateInit();
+
         if ($worker_id >= $server->setting['worker_num']) {
             $this->onTask($server, $worker_id);
         } else {
diff --git a/System/Abstracts/BaseApplication.php b/System/Abstracts/BaseApplication.php
index cff3468c..1c660d2a 100644
--- a/System/Abstracts/BaseApplication.php
+++ b/System/Abstracts/BaseApplication.php
@@ -93,13 +93,21 @@ abstract class BaseApplication extends Service
 	}
 
 
+    /**
+     * @return $this
+     */
 	public function stateInit()
 	{
 		$this->taskNumber = 0;
 		$this->state = 'SWOOLE_WORKER_IDLE';
+
+		return $this;
 	}
 
 
+    /**
+     * @return $this
+     */
 	public function decrement()
 	{
 		$this->taskNumber -= 1;
@@ -107,9 +115,13 @@ abstract class BaseApplication extends Service
 			$this->taskNumber = 0;
 			$this->state = 'SWOOLE_WORKER_IDLE';
 		}
+		return $this;
 	}
 
 
+    /**
+     * @return $this
+     */
 	public function increment()
 	{
 		$this->taskNumber += 1;
@@ -119,7 +131,6 @@ abstract class BaseApplication extends Service
 		} else {
 			$this->state = 'SWOOLE_WORKER_BUSY';
 		}
-
 		return $this;
 	}