Yi.Admin/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/InviteCodeAggregateRoot.cs

using SqlSugar;
using Volo.Abp.Domain.Entities.Auditing;

namespace Yi.Framework.AiHub.Domain.Entities;

/// <summary>
/// 用户邀请码
/// </summary>
[SugarTable("Ai_InviteCode")]
[SugarIndex($"index_{nameof(UserId)}", nameof(UserId), OrderByType.Asc, true)]
[SugarIndex($"index_{nameof(Code)}", nameof(Code), OrderByType.Asc, true)]
public class InviteCodeAggregateRoot : FullAuditedAggregateRoot<Guid>
{
    public InviteCodeAggregateRoot()
    {
    }

    public InviteCodeAggregateRoot(Guid userId, string code)
    {
        UserId = userId;
        Code = code;
        IsUsed = false;
        IsUserInvited = false;
        UsedCount = 0;
    }

    /// <summary>
    /// 用户ID（邀请码拥有者）
    /// </summary>
    public Guid UserId { get; set; }

    /// <summary>
    /// 邀请码（唯一）
    /// </summary>
    [SugarColumn(Length = 50)]
    public string Code { get; set; } = string.Empty;

    /// <summary>
    /// 是否已被使用（一个邀请码只能被使用一次）
    /// </summary>
    public bool IsUsed { get; set; }

    /// <summary>
    /// 邀请码拥有者是否已被他人邀请（被邀请后不可再提供邀请码）
    /// </summary>
    public bool IsUserInvited { get; set; }

    /// <summary>
    /// 被使用次数（统计用）
    /// </summary>
    public int UsedCount { get; set; }

    /// <summary>
    /// 使用时间
    /// </summary>
    public DateTime? UsedTime { get; set; }

    /// <summary>
    /// 使用人ID
    /// </summary>
    public Guid? UsedByUserId { get; set; }

    /// <summary>
    /// 备注信息
    /// </summary>
    [SugarColumn(Length = 500, IsNullable = true)]
    public string? Remark { get; set; }

    /// <summary>
    /// 标记邀请码已被使用
    /// </summary>
    /// <param name="usedByUserId">使用者ID</param>
    public void MarkAsUsed(Guid usedByUserId)
    {
        IsUsed = true;
        UsedTime = DateTime.Now;
        UsedByUserId = usedByUserId;
        UsedCount++;
    }

    /// <summary>
    /// 标记用户已被邀请
    /// </summary>
    public void MarkUserAsInvited()
    {
        IsUserInvited = true;
    }
}