public function __construct(NodeConnectionInterface $connection) { if ($connection instanceof ConnectionWrapper) { /** @var ConnectionWrapper $connection */ $connection = $connection->getConnection(); } $this->connection = $connection; }