PHP项目Symfony datafixtures与填充

wen PHP项目 1

本文目录导读:

PHP项目Symfony datafixtures与填充

  1. 安装DataFixtures Bundle
  2. 创建实体类
  3. 创建User实体(带关联关系)
  4. 创建Fixture类
  5. 配置文件和命令
  6. 带组的Fixtures
  7. 使用依赖注入
  8. 最佳实践

我来详细介绍Symfony项目中DataFixtures的使用和填充数据的方法。

安装DataFixtures Bundle

composer require --dev doctrine/doctrine-fixtures-bundle

创建实体类

// src/Entity/Product.php
namespace App\Entity;
use Doctrine\ORM\Mapping as ORM;
#[ORM\Entity]
class Product
{
    #[ORM\Id]
    #[ORM\GeneratedValue]
    #[ORM\Column(type: 'integer')]
    private ?int $id = null;
    #[ORM\Column(type: 'string', length: 255)]
    private ?string $name = null;
    #[ORM\Column(type: 'decimal', precision: 10, scale: 2)]
    private ?float $price = null;
    #[ORM\Column(type: 'text', nullable: true)]
    private ?string $description = null;
    // Getters and Setters
    public function getId(): ?int
    {
        return $this->id;
    }
    public function getName(): ?string
    {
        return $this->name;
    }
    public function setName(string $name): self
    {
        $this->name = $name;
        return $this;
    }
    public function getPrice(): ?float
    {
        return $this->price;
    }
    public function setPrice(float $price): self
    {
        $this->price = $price;
        return $this;
    }
    public function getDescription(): ?string
    {
        return $this->description;
    }
    public function setDescription(?string $description): self
    {
        $this->description = $description;
        return $this;
    }
}

创建User实体(带关联关系)

// src/Entity/User.php
namespace App\Entity;
use Doctrine\ORM\Mapping as ORM;
#[ORM\Entity]
class User
{
    #[ORM\Id]
    #[ORM\GeneratedValue]
    #[ORM\Column(type: 'integer')]
    private ?int $id = null;
    #[ORM\Column(type: 'string', length: 180, unique: true)]
    private ?string $email = null;
    #[ORM\Column(type: 'string', length: 180)]
    private ?string $name = null;
    #[ORM\Column(type: 'json')]
    private array $roles = ['ROLE_USER'];
    public function getId(): ?int
    {
        return $this->id;
    }
    public function getEmail(): ?string
    {
        return $this->email;
    }
    public function setEmail(string $email): self
    {
        $this->email = $email;
        return $this;
    }
    public function getName(): ?string
    {
        return $this->name;
    }
    public function setName(string $name): self
    {
        $this->name = $name;
        return $this;
    }
    public function getRoles(): array
    {
        $roles = $this->roles;
        $roles[] = 'ROLE_USER';
        return array_unique($roles);
    }
    public function setRoles(array $roles): self
    {
        $this->roles = $roles;
        return $this;
    }
}

创建Fixture类

基础Fixture(单表)

// src/DataFixtures/ProductFixtures.php
namespace App\DataFixtures;
use App\Entity\Product;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Persistence\ObjectManager;
class ProductFixtures extends Fixture
{
    public function load(ObjectManager $manager): void
    {
        $products = [
            ['name' => 'Laptop', 'price' => 999.99, 'description' => 'High performance laptop'],
            ['name' => 'Mouse', 'price' => 29.99, 'description' => 'Wireless mouse'],
            ['name' => 'Keyboard', 'price' => 79.99, 'description' => 'Mechanical keyboard'],
            ['name' => 'Monitor', 'price' => 399.99, 'description' => '27 inch 4K monitor'],
            ['name' => 'Headphones', 'price' => 149.99, 'description' => 'Noise cancelling headphones'],
        ];
        foreach ($products as $productData) {
            $product = new Product();
            $product->setName($productData['name']);
            $product->setPrice($productData['price']);
            $product->setDescription($productData['description']);
            $manager->persist($product);
        }
        $manager->flush();
    }
}

带关联关系的Fixture

// src/DataFixtures/UserFixtures.php
namespace App\DataFixtures;
use App\Entity\User;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Persistence\ObjectManager;
use Symfony\Component\PasswordHasher\Hasher\UserPasswordHasherInterface;
class UserFixtures extends Fixture
{
    private UserPasswordHasherInterface $passwordHasher;
    public function __construct(UserPasswordHasherInterface $passwordHasher)
    {
        $this->passwordHasher = $passwordHasher;
    }
    public function load(ObjectManager $manager): void
    {
        // 创建管理员
        $admin = new User();
        $admin->setEmail('admin@example.com');
        $admin->setName('Admin User');
        $admin->setRoles(['ROLE_ADMIN', 'ROLE_USER']);
        $admin->setPassword(
            $this->passwordHasher->hashPassword($admin, 'admin123')
        );
        $manager->persist($admin);
        // 创建普通用户
        $users = [
            ['email' => 'john@example.com', 'name' => 'John Doe'],
            ['email' => 'jane@example.com', 'name' => 'Jane Smith'],
            ['email' => 'bob@example.com', 'name' => 'Bob Johnson'],
        ];
        foreach ($users as $index => $userData) {
            $user = new User();
            $user->setEmail($userData['email']);
            $user->setName($userData['name']);
            $user->setPassword(
                $this->passwordHasher->hashPassword($user, 'password123')
            );
            $manager->persist($user);
            // 保存引用供其他Fixture使用
            $this->addReference('user_' . ($index + 1), $user);
        }
        $manager->flush();
    }
}

高级Fixture(带依赖关系和大量数据)

// src/DataFixtures/OrderFixtures.php
namespace App\DataFixtures;
use App\Entity\Order;
use App\Entity\OrderItem;
use App\Entity\Product;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Common\DataFixtures\DependentFixtureInterface;
use Doctrine\Persistence\ObjectManager;
use Faker\Factory;
class OrderFixtures extends Fixture implements DependentFixtureInterface
{
    public function load(ObjectManager $manager): void
    {
        $faker = Factory::create();
        // 获取所有产品(实际项目中可以查询数据库)
        $products = $manager->getRepository(Product::class)->findAll();
        for ($i = 0; $i < 20; $i++) {
            $order = new Order();
            $order->setOrderNumber('ORD-' . date('Ymd') . '-' . str_pad($i + 1, 4, '0', STR_PAD_LEFT));
            $order->setCustomerName($faker->name);
            $order->setCustomerEmail($faker->email);
            $order->setShippingAddress($faker->address);
            $order->setStatus($faker->randomElement(['pending', 'processing', 'shipped', 'delivered']));
            $order->setCreatedAt($faker->dateTimeBetween('-1 year', 'now'));
            $order->setUpdatedAt(new \DateTime());
            $totalAmount = 0;
            // 添加订单项
            $itemCount = rand(1, 5);
            for ($j = 0; $j < $itemCount; $j++) {
                $product = $faker->randomElement($products);
                $quantity = rand(1, 3);
                $unitPrice = $product->getPrice();
                $subtotal = $quantity * $unitPrice;
                $totalAmount += $subtotal;
                $orderItem = new OrderItem();
                $orderItem->setProduct($product);
                $orderItem->setQuantity($quantity);
                $orderItem->setUnitPrice($unitPrice);
                $orderItem->setSubtotal($subtotal);
                $orderItem->setOrder($order);
                $manager->persist($orderItem);
            }
            $order->setTotalAmount($totalAmount);
            $manager->persist($order);
            // 每10条记录flush一次
            if ($i % 10 === 0) {
                $manager->flush();
                $manager->clear();
            }
        }
        $manager->flush();
    }
    public function getDependencies(): array
    {
        return [
            ProductFixtures::class,
            UserFixtures::class,
        ];
    }
}

使用Faker生成大量数据

// src/DataFixtures/AppFixtures.php
namespace App\DataFixtures;
use App\Entity\Product;
use App\Entity\User;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Persistence\ObjectManager;
use Faker\Factory;
class AppFixtures extends Fixture
{
    public function load(ObjectManager $manager): void
    {
        $faker = Factory::create('zh_CN'); // 使用中文
        // 生成1000个产品
        for ($i = 0; $i < 1000; $i++) {
            $product = new Product();
            $product->setName($faker->unique()->word() . ' ' . $faker->word());
            $product->setPrice($faker->randomFloat(2, 10, 1000));
            $product->setDescription($faker->sentence());
            $manager->persist($product);
        }
        // 生成100个用户
        for ($i = 0; $i < 100; $i++) {
            $user = new User();
            $user->setName($faker->name());
            $user->setEmail($faker->unique()->email());
            $manager->persist($user);
        }
        $manager->flush();
    }
}

配置文件和命令

Fixture顺序控制

# config/services.yaml
services:
    App\DataFixtures\:
        resource: '../src/DataFixtures'
        tags: ['doctrine.fixture.orm']

常用命令

# 加载所有fixtures
php bin/console doctrine:fixtures:load
# 加载指定组的fixtures
php bin/console doctrine:fixtures:load --group=dev
# 追加数据(不删除现有数据)
php bin/console doctrine:fixtures:load --append
# 使用特定的entity manager
php bin/console doctrine:fixtures:load --em=default
# 加载时显示SQL语句
php bin/console doctrine:fixtures:load --show-sql

带组的Fixtures

// src/DataFixtures/DevFixtures.php
namespace App\DataFixtures;
use Doctrine\Bundle\FixturesBundle\FixtureGroupInterface;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Persistence\ObjectManager;
class DevFixtures extends Fixture implements FixtureGroupInterface
{
    public function load(ObjectManager $manager): void
    {
        // 开发环境的测试数据
    }
    public static function getGroups(): array
    {
        return ['dev', 'test'];
    }
}

使用依赖注入

// src/DataFixtures/AdvancedFixtures.php
namespace App\DataFixtures;
use Doctrine\Bundle\FixturesBundle\Fixture;
use Doctrine\Persistence\ObjectManager;
use Symfony\Component\Serializer\SerializerInterface;
use App\Service\DataProcessor;
class AdvancedFixtures extends Fixture
{
    private SerializerInterface $serializer;
    private DataProcessor $processor;
    public function __construct(
        SerializerInterface $serializer,
        DataProcessor $processor
    ) {
        $this->serializer = $serializer;
        $this->processor = $processor;
    }
    public function load(ObjectManager $manager): void
    {
        // 从JSON文件加载数据
        $jsonData = file_get_contents(__DIR__ . '/data/products.json');
        $products = $this->serializer->deserialize($jsonData, Product::class . '[]', 'json');
        foreach ($products as $product) {
            // 处理数据
            $this->processor->process($product);
            $manager->persist($product);
        }
        $manager->flush();
    }
}

最佳实践

使用Purge模式控制

# config/packages/doctrine_fixtures.yaml
doctrine_fixtures:
    # 删除模式: delete | truncate
    purge_mode: truncate
    # 是否在加载前清空数据库
    purge_with_truncate: true

优化性能

// 批量插入优化
public function load(ObjectManager $manager): void
{
    // 批量写入
    $batchSize = 20;
    for ($i = 0; $i < 10000; $i++) {
        $entity = new Entity();
        // ... 设置属性
        $manager->persist($entity);
        if (($i % $batchSize) === 0) {
            $manager->flush();
            $manager->clear(); // 释放内存
        }
    }
    $manager->flush();
}

这些方法可以帮助你在Symfony项目中高效地使用DataFixtures进行数据填充和测试。

抱歉,评论功能暂时关闭!