mirror of
https://github.com/pmmp/PocketMine-MP.git
synced 2025-07-06 01:51:51 +00:00
127 lines
3.3 KiB
PHP
127 lines
3.3 KiB
PHP
<?php
|
|
|
|
/*
|
|
*
|
|
* ____ _ _ __ __ _ __ __ ____
|
|
* | _ \ ___ ___| | _____| |_| \/ (_)_ __ ___ | \/ | _ \
|
|
* | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
|
|
* | __/ (_) | (__| < __/ |_| | | | | | | | __/_____| | | | __/
|
|
* |_| \___/ \___|_|\_\___|\__|_| |_|_|_| |_|\___| |_| |_|_|
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* @author PocketMine Team
|
|
* @link http://www.pocketmine.net/
|
|
*
|
|
*
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace pocketmine\scheduler;
|
|
|
|
use pocketmine\utils\MainLogger;
|
|
use pocketmine\utils\Utils;
|
|
use pocketmine\Worker;
|
|
use function error_reporting;
|
|
use function gc_enable;
|
|
use function ini_set;
|
|
use function set_error_handler;
|
|
|
|
class AsyncWorker extends Worker{
|
|
/** @var mixed[] */
|
|
private static $store = [];
|
|
|
|
/** @var \ThreadedLogger */
|
|
private $logger;
|
|
/** @var int */
|
|
private $id;
|
|
|
|
/** @var int */
|
|
private $memoryLimit;
|
|
|
|
public function __construct(\ThreadedLogger $logger, int $id, int $memoryLimit){
|
|
$this->logger = $logger;
|
|
$this->id = $id;
|
|
$this->memoryLimit = $memoryLimit;
|
|
}
|
|
|
|
public function run(){
|
|
error_reporting(-1);
|
|
|
|
$this->registerClassLoader();
|
|
|
|
//set this after the autoloader is registered
|
|
set_error_handler([Utils::class, 'errorExceptionHandler']);
|
|
|
|
if($this->logger instanceof MainLogger){
|
|
$this->logger->registerStatic();
|
|
}
|
|
|
|
gc_enable();
|
|
|
|
if($this->memoryLimit > 0){
|
|
ini_set('memory_limit', $this->memoryLimit . 'M');
|
|
$this->logger->debug("Set memory limit to " . $this->memoryLimit . " MB");
|
|
}else{
|
|
ini_set('memory_limit', '-1');
|
|
$this->logger->debug("No memory limit set");
|
|
}
|
|
}
|
|
|
|
public function getLogger() : \ThreadedLogger{
|
|
return $this->logger;
|
|
}
|
|
|
|
public function handleException(\Throwable $e){
|
|
$this->logger->logException($e);
|
|
}
|
|
|
|
public function getThreadName() : string{
|
|
return "Asynchronous Worker #" . $this->id;
|
|
}
|
|
|
|
public function getAsyncWorkerId() : int{
|
|
return $this->id;
|
|
}
|
|
|
|
/**
|
|
* Saves mixed data into the worker's thread-local object store. This can be used to store objects which you
|
|
* want to use on this worker thread from multiple AsyncTasks.
|
|
*
|
|
* @param string $identifier
|
|
* @param mixed $value
|
|
*/
|
|
public function saveToThreadStore(string $identifier, $value) : void{
|
|
self::$store[$identifier] = $value;
|
|
}
|
|
|
|
/**
|
|
* Retrieves mixed data from the worker's thread-local object store.
|
|
*
|
|
* Note that the thread-local object store could be cleared and your data might not exist, so your code should
|
|
* account for the possibility that what you're trying to retrieve might not exist.
|
|
*
|
|
* Objects stored in this storage may ONLY be retrieved while the task is running.
|
|
*
|
|
* @param string $identifier
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function getFromThreadStore(string $identifier){
|
|
return self::$store[$identifier] ?? null;
|
|
}
|
|
|
|
/**
|
|
* Removes previously-stored mixed data from the worker's thread-local object store.
|
|
*
|
|
* @param string $identifier
|
|
*/
|
|
public function removeFromThreadStore(string $identifier) : void{
|
|
unset(self::$store[$identifier]);
|
|
}
|
|
}
|