From a42fcecbfd0183ab84b4b68680bb8aa1a7b72669 Mon Sep 17 00:00:00 2001
From: Young <iamlovedit@163.com>
Date: Tue, 15 Oct 2024 22:42:44 +0800
Subject: [PATCH] added insert methods

---
 src/Infrastructure/Repository/IRepositoryBase.cs |  4 ++++
 src/Infrastructure/Repository/IServiceBase.cs    |  4 ++++
 src/Infrastructure/Repository/RepositoryBase.cs  | 10 ++++++++++
 src/Infrastructure/Repository/ServiceBase.cs     | 10 ++++++++++
 4 files changed, 28 insertions(+)

diff --git a/src/Infrastructure/Repository/IRepositoryBase.cs b/src/Infrastructure/Repository/IRepositoryBase.cs
index 78f47f6..b93880d 100644
--- a/src/Infrastructure/Repository/IRepositoryBase.cs
+++ b/src/Infrastructure/Repository/IRepositoryBase.cs
@@ -15,6 +15,10 @@ public interface IRepositoryBase<T> where T : class, new()
 
     Task<long> AddSnowflakeAsync(T entity);
 
+    Task<T> AddAsync(T entity);
+
+    Task<List<T>> AddRangeAsync(IList<T> entities);
+
     Task<IList<long>> AddSnowflakesAsync(IList<T> entities);
 
     Task<PageData<T?>> QueryPageAsync(Expression<Func<T, bool>>? whereExpression, int pageIndex = 1, int pageSize = 20,
diff --git a/src/Infrastructure/Repository/IServiceBase.cs b/src/Infrastructure/Repository/IServiceBase.cs
index 538ff81..4b4abf7 100644
--- a/src/Infrastructure/Repository/IServiceBase.cs
+++ b/src/Infrastructure/Repository/IServiceBase.cs
@@ -15,6 +15,10 @@ public interface IServiceBase<T> where T : class, new()
 
     Task<long> AddSnowflakeAsync(T entity);
 
+    Task<T> AddAsync(T entity);
+    
+    Task<List<T>> AddRangeAsync(IList<T> entities);
+    
     Task<IList<long>> AddSnowflakesAsync(IList<T> entities);
 
     Task<PageData<T?>> QueryPageAsync(Expression<Func<T, bool>>? whereExpression, int pageIndex = 1, int pageSize = 20,
diff --git a/src/Infrastructure/Repository/RepositoryBase.cs b/src/Infrastructure/Repository/RepositoryBase.cs
index 38a1b86..3c71fa5 100644
--- a/src/Infrastructure/Repository/RepositoryBase.cs
+++ b/src/Infrastructure/Repository/RepositoryBase.cs
@@ -18,6 +18,16 @@ public class RepositoryBase<T>(ISqlSugarClient context) : IRepositoryBase<T> whe
         return await context.Insertable<T>(entity).ExecuteReturnSnowflakeIdAsync();
     }
 
+    public async Task<T> AddAsync(T entity)
+    {
+        return await context.Insertable<T>(entity).ExecuteReturnEntityAsync();
+    }
+
+    public async Task<List<T>> AddRangeAsync(IList<T> entities)
+    {
+        return await context.Insertable<T>(entities).ExecuteReturnPkListAsync<T>();
+    }
+
     public async Task<IList<long>> AddSnowflakesAsync(IList<T> entities)
     {
         return await context.Insertable<T>(entities).ExecuteReturnSnowflakeIdListAsync();
diff --git a/src/Infrastructure/Repository/ServiceBase.cs b/src/Infrastructure/Repository/ServiceBase.cs
index 93f9a18..3946393 100644
--- a/src/Infrastructure/Repository/ServiceBase.cs
+++ b/src/Infrastructure/Repository/ServiceBase.cs
@@ -17,6 +17,16 @@ public abstract class ServiceBase<T>(IRepositoryBase<T> dbContext) : IServiceBas
         return await DAL.AddSnowflakeAsync(entity);
     }
 
+    public async Task<T> AddAsync(T entity)
+    {
+        return await DAL.AddAsync(entity);
+    }
+
+    public async Task<List<T>> AddRangeAsync(IList<T> entities)
+    {
+        return await DAL.AddRangeAsync(entities);
+    }
+
     public async Task<IList<long>> AddSnowflakesAsync(IList<T> entities)
     {
         return await DAL.AddSnowflakesAsync(entities);