FressRSS/app/Models/DatabaseDAOPGSQL.php

<?php

/**
 * This class is used to test database is well-constructed.
 */
class FreshRSS_DatabaseDAOPGSQL extends FreshRSS_DatabaseDAO {
	public function tablesAreCorrect() {
		$db = FreshRSS_Context::$system_conf->db;
		$dbowner = $db['user'];
		$sql = 'SELECT * FROM pg_catalog.pg_tables where tableowner=?';
		$stm = $this->bd->prepare($sql);
		$values = array($dbowner);
		$stm->execute($values);
		$res = $stm->fetchAll(PDO::FETCH_ASSOC);

		$tables = array(
			$this->prefix . 'category' => false,
			$this->prefix . 'feed' => false,
			$this->prefix . 'entry' => false,
		);
		foreach ($res as $value) {
			$tables[array_pop($value)] = true;
		}

		return count(array_keys($tables, true, true)) == count($tables);
	}

	public function getSchema($table) {
		$sql = 'select column_name as field, data_type as type, column_default as default, is_nullable as null from INFORMATION_SCHEMA.COLUMNS where table_name = ?';
		$stm = $this->bd->prepare($sql);
		$stm->execute(array($this->prefix . $table));
		return $this->listDaoToSchema($stm->fetchAll(PDO::FETCH_ASSOC));
	}

	public function daoToSchema($dao) {
		return array(
			'name' => $dao['field'],
			'type' => strtolower($dao['type']),
			'notnull' => (bool)$dao['null'],
			'default' => $dao['default'],
		);
	}

	public function size($all = true) {
		$db = FreshRSS_Context::$system_conf->db;
		$sql = 'SELECT pg_size_pretty(pg_database_size(?))';
		$values = array($db['base']);
		$stm = $this->bd->prepare($sql);
		$stm->execute($values);
		$res = $stm->fetchAll(PDO::FETCH_COLUMN, 0);
		return $res[0];
	}

	public function optimize() {
		$ok = true;

		$sql = 'VACUUM `' . $this->prefix . 'entry`';
		$stm = $this->bd->prepare($sql);
		$ok &= $stm != false;
		if ($stm) {
			$ok &= $stm->execute();
		}

		$sql = 'VACUUM `' . $this->prefix . 'feed`';
		$stm = $this->bd->prepare($sql);
		$ok &= $stm != false;
		if ($stm) {
			$ok &= $stm->execute();
		}

		$sql = 'VACUUM `' . $this->prefix . 'category`';
		$stm = $this->bd->prepare($sql);
		$ok &= $stm != false;
		if ($stm) {
			$ok &= $stm->execute();
		}

		return $ok;
	}
}