php WPDB抽象

Posted 2021-05-05

<?php
use Exception;

/**
 * Class Connection
 *
 * Adds a layer of abstraction over WP's $wpdb global for
 * DB operations
 *
 * @package Manager\WPDatabase
 */
class Connection {

	/**
	 * @var \wpdb - WP instance of DB
	 */
	private $wpdb;

	/**
	 * Connection constructor.
	 * Accepts $wpdb global
	 *
	 * @param \wpdb $database
	 * @throws Exception
	 */
	public function __construct(\wpdb $database) {
		try {
			$connected = $database->check_connection();

			if ($connected) {
				$this->wpdb = $database;
			}
		} catch(\Throwable $e) {
			echo $e->getMessage();
		}
	}

	/**
	 * Filter SQL through WPDB prepare method
	 *
	 * @param string $sql
	 * @param array $args
	 *
	 * @return string|void
	 */
	public function prepare(string $sql, array $args) {
		return $this->wpdb->prepare($sql, $args);
	}

	/**
	 * Queries WP DB
	 *
	 * @param string $sql
	 * @throws Exception
	 */
	public function query(string $sql) {
		try {
			$this->wpdb->query($sql);
		} catch (Exception $e) {
			echo $e->getMessage();
		}
	}

	/**
	 * Returns result of WPDB query, as an associative array
	 *
	 * @param string $sql
	 *
	 * @return array|bool|null|object
	 */
	public function getResults(string $sql) {
		$result = false;

		try {
			$result = $this->wpdb->get_results($sql, ARRAY_A);
		} catch (Exception $e) {
			echo $e->getMessage();
		}

		return $result;
	}
}