PHP OOP和PDO

Question

我第一次尝试广泛使用PHP OOP和PDO.我终于让脚本工作,但正如你注意到的那样,我必须将PDO连接移动到登录功能 - 最初它只是在__construct().我不想打开一个新的PDO连接,这很草率.如何在整个班级中保持相同的连接？

 <?php
class user{

public $id;
public $name;
public $email;
private $password;

public function __construct() {
    $DBH = new PDO("mysql:host=HOST;dbname=DB", "USER", "PASS");  
    $DBH->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
}

public function login($email,$password,$type){

    $DBH = new PDO("mysql:host=HOST;dbname=DB", "USER", "PASS");  
    $DBH->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

    $password_hash = sha1($password);
    try{
        if ($type != "actives") {
            throw new Exception("Type Handling Error");
        }
        $STH = $DBH->query("SELECT id, email, password FROM $type WHERE email='$email' AND password='$password_hash'");
        $STH->setFetchMode(PDO::FETCH_ASSOC);  
        $row_count = $STH->rowCount();  
        $row = $STH->fetch();

        if($row_count == 1){
            session_start();
            session_regenerate_id();
            $_SESSION['id'] == $row[id];
            return true;
        }
        else{
        return false;
        }
    }
    catch (Exception $e) {
        echo $e->getMessage();
    }

}

public function loggout(){
    session_destroy();
    setcookie(session_name(), session_id(), 1, '/');
}

Answer 1

使数据库处理类中的私有成员:

class user
{    
    public $id;
    public $name;
    public $email;
    private $password;
    private $dbh;

    public function __construct(PDO $dbh)
    {
        $this->dbh = $dbh;  
    }

    public function login($email, $password, $type)
    {
        $dbh = $this->dbh;
        ...
    }

用法:

$pdo = new PDO("mysql:host=HOST;dbname=DB", "USER", "PASS");
$pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
$user = new User($pdo);

Answer 2

迟早，您可能会在代码中的其他位置（在用户对象之外）需要 PDO 连接对象。因此，我建议使用一个类，该类将提供如下所示的静态方法，以便在您想要的任何地方获取 PDO 连接对象。

             class Database{
                    private static $datasource='mysql:host=HOST dbname=DB';
                    private static $username='USER';
                    private static $password='PASS';
                    private static $db;

                    //make the constructor private and empty so that no code will create an object of this class.
                    private function __construct(){}

                     //the main public function which will return the required PDO object
                     public static function getDB(){
                      if(!isset(self::$db)){
                           try{
                         self::$db=new PDO(self::$datasoure,self::$username,self::$password);

                            }
                            catch(PDOExceptin $e)
                            {
                               $error=$e->getMessage(); //variable $error can be used in the database_error.php file 
                               //display database error file.
                               include('database_error.php');
                               exit();
                            }
                      }
                        return self::$db; 
                     }

                    }

然后您可以在任何需要 PDO 连接的时候使用如下所示的静态方法

                    $conn=Database::getDB();