解决PDO中lastInsertId()失效问题：深入解析连接管理与解决方案

class Customer extends Dbh {
    protected function setAddCustomer($c_name, $c_address){
      // 第一次连接：用于准备和执行INSERT语句
      $stmt = $this->connect()->prepare('INSERT INTO customer (c_name, c_address) VALUES (?, ?);');

      if(!$stmt->execute(array($c_name, $c_address))) {
        $stmt = null;
        header("location:../index.php?error=connectionfaild!");
        exit();      
       }

      // 第二次连接：尝试获取lastInsertId()
      // 如果connect()每次都返回新的PDO实例，这里的lastInsertId()将无法获取到前一个连接的ID
      $last_id = $this->connect()->lastInsertId(); 
      return $last_id;
    }
}

class Dbh {
    private $connection; // 存储PDO连接实例

    // 假设connect()方法负责实际建立连接
    protected function establishConnection(): \PDO {
        // 这里放置实际的数据库连接逻辑，例如：
        // $dsn = 'mysql:host=localhost;dbname=yourdb;charset=utf8';
        // $user = 'youruser';
        // $pass = 'yourpass';
        // return new \PDO($dsn, $user, $pass);
        // 为简化示例，此处仅作示意
        return new \PDO('mysql:host=localhost;dbname=testdb', 'root', ''); 
    }

    // 新增的getConnection方法，用于获取并复用PDO连接
    public function getConnection(): \PDO {
        if (!$this->connection) {
            $this->connection = $this->establishConnection(); // 首次调用时建立连接
            $this->connection->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION); // 设置错误模式
            // 其他PDO属性设置...
        }
        return $this->connection; // 返回已存在的连接实例
    }
}

class Customer extends Dbh {
    protected function setAddCustomer($c_name, $c_address){
      // 使用getConnection()获取并复用PDO连接
      $pdo = $this->getConnection(); 
      $stmt = $pdo->prepare('INSERT INTO customer (c_name, c_address) VALUES (?, ?);');

      if(!$stmt->execute(array($c_name, $c_address))) {
        $stmt = null;
        // 错误处理...
        header("location:../index.php?error=connectionfaild!");
        exit();      
       }

      // 现在lastInsertId()将作用于同一个PDO连接实例
      $last_id = $pdo->lastInsertId(); 
      return $last_id;
    }
}

class Customer {
    private \PDO $connection;

    // 通过构造函数注入PDO连接实例
    public function __construct(\PDO $connection) {
        $this->connection = $connection;
    }

    protected function setAddCustomer($c_name, $c_address){
      $stmt = $this->connection->prepare('INSERT INTO customer (c_name, c_address) VALUES (?, ?);');

      if(!$stmt->execute(array($c_name, $c_address))) {
        $stmt = null;
        // 错误处理...
        header("location:../index.php?error=connectionfaild!");
        exit();      
       }

      // 直接使用注入的PDO连接实例
      $last_id = $this->connection->lastInsertId(); 
      return $last_id;
    }
}

// 在应用入口点或服务容器中创建并管理PDO连接
// $pdo = new \PDO('mysql:host=localhost;dbname=testdb', 'root', '');
// $pdo->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);

// 然后将同一个PDO实例传递给Customer类
// $customer = new Customer($pdo);
// $customer->setAddCustomer('John Doe', '123 Main St');