4 files changed, 37 insertions, 6 deletions
diff --git a/BackEnd/Timeline/Entities/DatabaseContext.cs b/BackEnd/Timeline/Entities/DatabaseContext.cs
index cb776e72..7984f0a3 100644
--- a/BackEnd/Timeline/Entities/DatabaseContext.cs
+++ b/BackEnd/Timeline/Entities/DatabaseContext.cs
@@ -1,4 +1,4 @@
-using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore;
 
 namespace Timeline.Entities
 {
@@ -36,6 +36,9 @@ namespace Timeline.Entities
 
         public DbSet<UserConfigurationEntity> UserConfigurations { get; set; } = default!;
 
+        public DbSet<RegisterCode> RegisterCodes { get; set; } = default!;
+        public DbSet<UserRegisterInfo> UserRegisterInfos { get; set; } = default!;
+
         public DbSet<JwtTokenEntity> JwtToken { get; set; } = default!;
         public DbSet<UserTokenEntity> UserTokens { get; set; } = default!;
         public DbSet<DataEntity> Data { get; set; } = default!;
diff --git a/BackEnd/Timeline/Entities/RegisterCode.cs b/BackEnd/Timeline/Entities/RegisterCode.cs
index 945d61e9..034dde98 100644
--- a/BackEnd/Timeline/Entities/RegisterCode.cs
+++ b/BackEnd/Timeline/Entities/RegisterCode.cs
@@ -10,10 +10,10 @@ namespace Timeline.Entities
         public long Id { get; set; }
 
         [Column("owner_id")]
-        public long OwnerId { get; set; }
+        public long? OwnerId { get; set; }
 
         [ForeignKey(nameof(OwnerId))]
-        public UserEntity Owner { get; set; } = default!;
+        public UserEntity? Owner { get; set; } = default!;
 
         [Column("code")]
         public string Code { get; set; } = default!;
diff --git a/BackEnd/Timeline/Entities/UserEntity.cs b/BackEnd/Timeline/Entities/UserEntity.cs
index 267d0ef2..d8015ccf 100644
--- a/BackEnd/Timeline/Entities/UserEntity.cs
+++ b/BackEnd/Timeline/Entities/UserEntity.cs
@@ -37,9 +37,6 @@ namespace Timeline.Entities
         [Column("last_modified")]
         public DateTime LastModified { get; set; }
 
-        [Column("register_code")]
-        public string? RegisterCode { get; set; }
-
         public UserAvatarEntity? Avatar { get; set; }
 
 #pragma warning disable CA2227 // Collection properties should be read only
diff --git a/BackEnd/Timeline/Entities/UserRegisterInfo.cs b/BackEnd/Timeline/Entities/UserRegisterInfo.cs
new file mode 100644
index 00000000..1d62aadb
--- /dev/null
+++ b/BackEnd/Timeline/Entities/UserRegisterInfo.cs
@@ -0,0 +1,31 @@
+using System;
+using System.ComponentModel.DataAnnotations;
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace Timeline.Entities
+{
+    [Table("user_register_info")]
+    public class UserRegisterInfo
+    {
+        [Column("id"), Key, DatabaseGenerated(DatabaseGeneratedOption.Identity)]
+        public long Id { get; set; }
+
+        [Column("user_id")]
+        public long UserId { get; set; }
+
+        [ForeignKey(nameof(UserId))]
+        public UserEntity User { get; set; } = default!;
+
+        [Column("register_code")]
+        public string RegisterCode { get; set; } = default!;
+
+        [Column("introducer_id")]
+        public long? IntroducerId { get; set; }
+
+        [ForeignKey(nameof(IntroducerId))]
+        public UserEntity? Introducer { get; set; }
+
+        [Column("register_time")]
+        public DateTime RegisterTime { get; set; }
+    }
+}