This commit is contained in:
2021-04-16 15:11:52 +08:00
parent 525b149e02
commit cd41397f36
+5 -11
View File
@@ -36,11 +36,12 @@ class OnWorkerStart extends Callback
$annotation = Snowflake::app()->getAnnotation(); $annotation = Snowflake::app()->getAnnotation();
$annotation->setLoader(unserialize($content)); $annotation->setLoader(unserialize($content));
$annotation->runtime(directory('app'));
if ($worker_id < $server->setting['worker_num']) { if ($worker_id < $server->setting['worker_num']) {
$this->onWorker($server, $annotation); $this->onWorker($server);
} else { } else {
$this->onTask($server, $annotation); $this->onTask($server);
} }
} }
@@ -58,14 +59,11 @@ class OnWorkerStart extends Callback
/** /**
* @param Server $server * @param Server $server
* @param Annotation $annotation
* @throws ConfigException * @throws ConfigException
* @throws Exception * @throws Exception
*/ */
public function onTask(Server $server, Annotation $annotation) public function onTask(Server $server)
{ {
$annotation->runtime(directory('app'));
putenv('environmental=' . Snowflake::TASK); putenv('environmental=' . Snowflake::TASK);
name($server->worker_pid, 'Task#' . $server->worker_id); name($server->worker_pid, 'Task#' . $server->worker_id);
@@ -78,15 +76,11 @@ class OnWorkerStart extends Callback
/** /**
* @param Server $server * @param Server $server
* @param Annotation $annotation
* @throws Exception * @throws Exception
*/ */
public function onWorker(Server $server, Annotation $annotation) public function onWorker(Server $server)
{ {
try { try {
$time = microtime(true);
$annotation->runtime(directory('app'));
name($server->worker_pid, 'Worker#' . $server->worker_id); name($server->worker_pid, 'Worker#' . $server->worker_id);
Snowflake::setWorkerId($server->worker_pid); Snowflake::setWorkerId($server->worker_pid);