summaryrefslogtreecommitdiff
path: root/app/Models/DatabaseDAOPGSQL.php
blob: 75ff8be7babdcf1b3792eba6e7e0135e4e9d037b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
<?php

/**
 * This class is used to test database is well-constructed.
 */
class FreshRSS_DatabaseDAOPGSQL extends FreshRSS_DatabaseDAOSQLite {

	//PostgreSQL error codes
	const UNDEFINED_COLUMN = '42703';
	const UNDEFINED_TABLE = '42P01';

	public function tablesAreCorrect() {
		$db = FreshRSS_Context::$system_conf->db;
		$dbowner = $db['user'];
		$sql = 'SELECT * FROM pg_catalog.pg_tables where tableowner=?';
		$stm = $this->pdo->prepare($sql);
		$values = array($dbowner);
		$stm->execute($values);
		$res = $stm->fetchAll(PDO::FETCH_ASSOC);

		$tables = array(
			$this->pdo->prefix() . 'category' => false,
			$this->pdo->prefix() . 'feed' => false,
			$this->pdo->prefix() . 'entry' => false,
			$this->pdo->prefix() . 'entrytmp' => false,
			$this->pdo->prefix() . 'tag' => false,
			$this->pdo->prefix() . 'entrytag' => 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->pdo->prepare($sql);
		$stm->execute(array($this->pdo->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 = false) {
		if ($all) {
			$db = FreshRSS_Context::$system_conf->db;
			$sql = 'SELECT pg_database_size(:base)';
			$stm = $this->pdo->prepare($sql);
			$stm->bindParam(':base', $db['base']);
			$stm->execute();
		} else {
			$sql = "SELECT "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}category`') + "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}feed`') + "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}entry`') + "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}entrytmp`') + "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}tag`') + "
			     . "pg_total_relation_size('`{$this->pdo->prefix()}entrytag`')";
			$stm = $this->pdo->query($sql);
		}
		if ($stm == false) {
			return 0;
		}
		$res = $stm->fetchAll(PDO::FETCH_COLUMN, 0);
		return $res[0];
	}

	public function optimize() {
		$ok = true;
		$tables = array('category', 'feed', 'entry', 'entrytmp', 'tag', 'entrytag');

		foreach ($tables as $table) {
			$sql = 'VACUUM `_' . $table . '`';
			if ($this->pdo->exec($sql) === false) {
				$ok = false;
				$info = $this->pdo->errorInfo();
				Minz_Log::warning(__METHOD__ . ' error: ' . $sql . ' : ' . json_encode($info));
			}
		}
		return $ok;
	}
}