You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

55 lines
1.7 KiB

3 years ago
  1. <?php
  2. /**
  3. * @copyright Copyright (c) 2018 Robin Appelman <robin@icewind.nl>
  4. *
  5. * @author Christoph Wurst <christoph@winzerhof-wurst.at>
  6. * @author Robin Appelman <robin@icewind.nl>
  7. *
  8. * @license GNU AGPL version 3 or any later version
  9. *
  10. * This program is free software: you can redistribute it and/or modify
  11. * it under the terms of the GNU Affero General Public License as
  12. * published by the Free Software Foundation, either version 3 of the
  13. * License, or (at your option) any later version.
  14. *
  15. * This program is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. * GNU Affero General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Affero General Public License
  21. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. *
  23. */
  24. namespace OC\DB;
  25. use Doctrine\Common\EventManager;
  26. use Doctrine\DBAL\Configuration;
  27. use Doctrine\DBAL\Driver;
  28. class ReconnectWrapper extends \Doctrine\DBAL\Connection {
  29. public const CHECK_CONNECTION_INTERVAL = 60;
  30. private $lastConnectionCheck = null;
  31. public function __construct(array $params, Driver $driver, Configuration $config = null, EventManager $eventManager = null) {
  32. parent::__construct($params, $driver, $config, $eventManager);
  33. $this->lastConnectionCheck = time();
  34. }
  35. public function connect() {
  36. $now = time();
  37. $checkTime = $now - self::CHECK_CONNECTION_INTERVAL;
  38. if ($this->lastConnectionCheck > $checkTime || $this->isTransactionActive()) {
  39. return parent::connect();
  40. } else {
  41. $this->lastConnectionCheck = $now;
  42. if (!$this->ping()) {
  43. $this->close();
  44. }
  45. return parent::connect();
  46. }
  47. }
  48. }