RobotSubTaskConfiguration.cs 2.71 KB
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata.Builders;
using Rcs.Domain.Entities;

namespace Rcs.Infrastructure.DB.Configuration.Domain
{
    /// <summary>
    /// RobotSubTask实体的EF Core配置
    /// @author zzy
    /// </summary>
    public class RobotSubTaskConfiguration : IEntityTypeConfiguration<RobotSubTask>
    {
        public void Configure(EntityTypeBuilder<RobotSubTask> builder)
        {
            builder.ToTable("robot_sub_tasks");

            builder.HasKey(e => e.SubTaskId);

            builder.Property(e => e.SubTaskId)
                .HasColumnName("sub_task_id")
                .ValueGeneratedNever();

            builder.Property(e => e.TaskId)
                .HasColumnName("task_id")
                .IsRequired();

            builder.Property(e => e.RobotId)
                .HasColumnName("robot_id");

            builder.Property(e => e.BeginNodeId)
                .HasColumnName("begin_node_id")
                .IsRequired();

            builder.Property(e => e.EndNodeId)
                .HasColumnName("end_node_id")
                .IsRequired();

            builder.Property(e => e.Sequence)
                .HasColumnName("sequence")
                .IsRequired();

            builder.Property(e => e.Status)
                .HasColumnName("status")
                .IsRequired();

            builder.Property(e => e.CreatedAt)
                .HasColumnName("created_at")
                .HasColumnType("timestamp without time zone");

            builder.Property(e => e.UpdatedAt)
                .HasColumnName("updated_at")
                .HasColumnType("timestamp without time zone");

            // 索引
            builder.HasIndex(e => e.TaskId)
                .HasDatabaseName("idx_sub_task_task_id");

            builder.HasIndex(e => e.RobotId)
                .HasDatabaseName("idx_sub_task_robot_id");

            builder.HasIndex(e => new { e.TaskId, e.Sequence })
                .HasDatabaseName("idx_sub_task_sequence");

            // 导航属性
            builder.HasOne(e => e.Task)
                .WithMany()
                .HasForeignKey(e => e.TaskId)
                .OnDelete(DeleteBehavior.Cascade);

            builder.HasOne(e => e.Robot)
                .WithMany()
                .HasForeignKey(e => e.RobotId)
                .OnDelete(DeleteBehavior.SetNull);

            builder.HasOne(e => e.BeginNode)
                .WithMany()
                .HasForeignKey(e => e.BeginNodeId)
                .OnDelete(DeleteBehavior.Restrict);

            builder.HasOne(e => e.EndNode)
                .WithMany()
                .HasForeignKey(e => e.EndNodeId)
                .OnDelete(DeleteBehavior.Restrict);
        }
    }
}