boinc/html/inc/db_conn.inc

252 lines
7.9 KiB
PHP

<?php
// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
// Should not be enabled for public code
//ini_set('display_errors', 'stdout');
//error_reporting(E_ALL);
/// represents a connection to a database.
// Intended to be subclassed (e.g., BoincDb, BossaDb)
if (class_exists("mysqli")) {
define("MYSQLI", true);
} else {
define("MYSQLI", false);
}
class DbConn {
var $db_conn;
var $db_name;
function init_conn($user, $passwd, $host, $name) {
if (MYSQLI) {
if (version_compare(PHP_VERSION, '5.3.0') < 0) {
$this->db_conn = new mysqli($host, $user, $passwd);
} else {
$this->db_conn = new mysqli("p:".$host, $user, $passwd);
}
} else {
$this->db_conn = mysql_pconnect($host, $user, $passwd);
}
if (!$this->db_conn) {
return false;
}
$this->db_name = $name;
return true;
}
// in keeping with PHP/MySQL convention, return true (nonzero) on success.
// (This is the opposite of the BOINC convention)
//
function do_query($q) {
global $generating_xml;
$q = str_replace('DBNAME', $this->db_name, $q);
//echo "query: $q<br>\n";
if (MYSQLI) {
$ret = $this->db_conn->query($q);
} else {
$ret = mysql_query($q, $this->db_conn);
}
if (!$ret) {
if (!$generating_xml) {
echo "Database Error<br>\n";
}
//echo ": ", $this->base_error(), "\n<pre>";
//var_dump(debug_backtrace());
//echo "</pre>query: $q\n";
return null;
}
return $ret;
}
// # rows affected by last query
//
function affected_rows() {
if (MYSQLI) {
$this->db_conn->affected_rows;
} else {
return mysql_affected_rows($this->db_conn);
}
}
function lookup_fields($table, $classname, $fields, $clause) {
$query = "select $fields from DBNAME.$table where $clause";
$result = $this->do_query($query);
if (!$result) {
return null;
}
if (MYSQLI) {
$obj = $result->fetch_object($classname);
$result->free();
} else {
$obj = mysql_fetch_object($result, $classname);
mysql_free_result($result);
}
return $obj;
}
function lookup($table, $classname, $clause) {
return $this->lookup_fields($table, $classname, "*", $clause);
}
function lookup_id($id, $table, $classname) {
return $this->lookup($table, $classname, "id=$id");
}
function enum_general($classname, $query) {
$result = $this->do_query($query);
if (!$result) return null;
$x = array();
if (MYSQLI) {
while ($obj = $result->fetch_object($classname)) {
$x[] = $obj;
}
$result->free();
} else {
while ($obj = mysql_fetch_object($result, $classname)) {
$x[] = $obj;
}
mysql_free_result($result);
}
return $x;
}
function enum_fields(
$table, $classname, $fields, $where_clause, $order_clause
) {
$x = array();
if ($where_clause) {
$where_clause = "where $where_clause";
}
$query = "select $fields from DBNAME.$table $where_clause $order_clause";
return $this->enum_general($classname,$query);
}
function enum($table, $classname, $where_clause=null, $order_clause=null) {
return self::enum_fields(
$table, $classname, '*', $where_clause, $order_clause
);
}
function update($obj, $table, $clause) {
$query = "update DBNAME.$table set $clause where id=$obj->id";
return $this->do_query($query);
}
function update_aux($table, $clause) {
$query = "update DBNAME.$table set $clause";
return $this->do_query($query);
}
function insert($table, $clause) {
$query = "insert into DBNAME.$table $clause";
return $this->do_query($query);
}
function delete($obj, $table) {
$query = "delete from DBNAME.$table where id=$obj->id";
return $this->do_query($query);
}
function delete_aux($table, $clause) {
$query = "delete from DBNAME.$table where $clause";
return $this->do_query($query);
}
function insert_id() {
if (MYSQLI) {
return $this->db_conn->insert_id;
} else {
return mysql_insert_id($this->db_conn);
}
}
function get_int($query, $field) {
$result = $this->do_query($query);
if (!$result) error_page("database error on query $query");
if (MYSQLI) {
$x = $result->fetch_object("StdClass");
$result->free();
} else {
$x = mysql_fetch_object($result);
mysql_free_result($result);
}
if ($x) return $x->$field;
return false;
}
function get_double($query, $field) {
$result = $this->do_query($query);
if (!$result) error_page("database error on query $query");
if (MYSQLI) {
$x = $result->fetch_object("StdClass");
$result->free();
} else {
$x = mysql_fetch_object($result);
mysql_free_result($result);
}
if ($x) return (double)$x->$field;
return false;
}
function count($table, $clause="TRUE") {
$query = "select count(*) as total from DBNAME.$table where $clause";
return $this->get_int($query, 'total');
}
function sum($table, $field, $clause="") {
$query = "select sum($field) as total from DBNAME.$table $clause";
return $this->get_double($query, 'total');
}
function max($table, $field, $clause="") {
$query = "select max($field) as total from DBNAME.$table $clause";
return $this->get_double($query, 'total');
}
function percentile($table, $field, $clause, $pct) {
$n = $this->count($table, $clause);
if (!$n) return false;
$m = (int)($n*$pct/100);
$query = "select $field from DBNAME.$table where $clause order by $field limit $m,1";
return $this->get_double($query, $field);
}
function replace($table, $clause) {
$query = "replace into DBNAME.$table set $clause";
return $this->do_query($query);
}
function base_escape_string($string) {
if (MYSQLI) {
return $this->db_conn->escape_string($string);
} else {
return mysql_real_escape_string($string);
}
}
function base_error() {
if (MYSQLI) {
return $this->db_conn->error;
} else {
return mysql_error($this->db_conn);
}
}
function table_exists($table_name) {
$result = $this->do_query("show tables from DBNAME like '$table_name'");
if (!$result) error_page("database error on query $query");
if (MYSQLI) {
$t = $result->fetch_array();
$result->free();
} else {
$t = mysql_fetch_array($result);
mysql_free_result($result);
}
if ($t[0] == "$table_name") return true;
return false;
}
}
?>