使用PDO将数据映射到包含枚举属性的PHP对象

<?php

// 定义一个整型支持的枚举
enum UserType: int
{
    case Master = 1;
    case Admin = 2;
    case Manager = 3;
}

// 定义一个包含枚举属性的类
class User
{
    private int $id;
    private string $name;
    private UserType $userType; // 枚举属性
}

<?php
// 假设 Database::getInstance() 返回一个 PDO 实例
// 假设 fetchObject 方法内部使用了 PDOStatement::fetchObject($class_name)
$user = Database::getInstance()->fetchObject(
    sql: "SELECT id, name, userType FROM user WHERE id = 1",
    class_name: User::class
);

<?php

enum UserType: int
{
    case Master = 1;
    case Admin = 2;
    case Manager = 3;
}

class User
{
    private int $id;
    private string $name;
    private UserType $userType; // 枚举属性

    public function __construct()
    {
        // 在构造函数中 unset 掉枚举属性，
        // 这样 PDO 在尝试设置此属性时会调用 __set 魔术方法
        unset($this->userType);
    }

    // __set 魔术方法用于拦截对未定义或 unset 属性的赋值操作
    public function __set(string $key, mixed $value): void
    {
        if ($key === 'userType') {
            // 当尝试设置 userType 属性时，将整数值转换为 UserType 枚举实例
            $this->userType = UserType::from($value);
        }
        // 对于其他属性，可以根据需要进行处理或忽略
    }

    // 为了访问属性，可以添加 getter 方法
    public function getId(): int { return $this->id; }
    public function getName(): string { return $this->name; }
    public function getUserType(): UserType { return $this->userType; }
}

<?php

// 假设 $pdo 是一个 PDO 实例
$stmt = $pdo->prepare("SELECT id, name, userType FROM user WHERE id = :id");
$stmt->execute([':id' => 1]);

// 设置 fetch 模式：先构造对象，后设置属性（会触发 __set）
$stmt->setFetchMode(PDO::FETCH_CLASS | PDO::FETCH_PROPS_LATE, User::class);

$user = $stmt->fetch(); // 获取 User 类的实例

if ($user instanceof User) {
    echo "User ID: " . $user->getId() . "\n";
    echo "User Name: " . $user->getName() . "\n";
    echo "User Type: " . $user->getUserType()->name . " (Value: " . $user->getUserType()->value . ")\n";
} else {
    echo "User not found.\n";
}

<?php

enum UserType: int
{
    case Master = 1;
    case Admin = 2;
    case Manager = 3;
}

class User
{
    private UserType $userType;

    // 构造函数接受数据库中的原始整数值
    public function __construct(
        private int $id,
        private string $name,
        int $userType // 注意这里接受的是 int
    ) {
        // 在构造函数内部将整数转换为 UserType 枚举实例
        $this->userType = UserType::from($userType);
    }

    // 为了访问属性，可以添加 getter 方法
    public function getId(): int { return $this->id; }
    public function getName(): string { return $this->name; }
    public function getUserType(): UserType { return $this->userType; }
}

<?php

class Database
{
    private static ?PDO $instance = null;

    // 假设这是获取 PDO 实例的方法
    public static function getInstance(): PDO
    {
        if (self::$instance === null) {
            // 配置你的数据库连接
            $dsn = 'mysql:host=localhost;dbname=your_db;charset=utf8mb4';
            $user = 'your_user';
            $password = 'your_password';
            $options = [
                PDO::ATTR_ERRMODE            => PDO::ERRMODE_EXCEPTION,
                PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC, // 默认使用关联数组
                PDO::ATTR_EMULATE_PREPARES   => false,
            ];
            self::$instance = new PDO($dsn, $user, $password, $options);
        }
        return self::$instance;
    }

    /**
     * 从数据库获取一行数据并映射到指定类的实例。
     *
     * @param string $sql SQL查询语句。
     * @param array $args 绑定到查询的参数。
     * @param string $class_name 目标类的名称。
     * @return object|null 目标类的实例或 null 如果没有找到数据。
     */
    public function fetchObject(string $sql, array $args = [], string $class_name = "stdClass"): ?object
    {
        $stmt = self::getInstance()->prepare($sql);
        $stmt->execute($args); // 执行查询

        // 使用 PDO::FETCH_ASSOC 获取关联数组
        $row = $stmt->fetch(PDO::FETCH_ASSOC);
        $stmt->closeCursor();

        // 如果有数据，则使用数组解包创建目标类的实例
        // 注意：数组的键名必须与构造函数的参数名匹配
        return $row ? new $class_name(...$row) : null;
    }
}

// 使用新的 fetchObject 方法
$db = new Database(); // 假设 Database 是一个可实例化的类或通过静态方法获取实例
$user = $db->fetchObject(
    sql: "SELECT id, name, userType FROM user WHERE id = 1",
    class_name: User::class
);

if ($user instanceof User) {
    echo "User ID: " . $user->getId() . "\n";
    echo "User Name: " . $user->getName() . "\n";
    echo "User Type: " . $user->getUserType()->name . " (Value: " . $user->getUserType()->value . ")\n";
} else {
    echo "User not found.\n";
}