TargetRepository.cs 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. using Abp.Collections.Extensions;
  2. using SnapshotWinFormsApp.Application.Dtos;
  3. using SnapshotWinFormsApp.Entities.NewHotline;
  4. using SnapshotWinFormsApp.Repository.Interfaces;
  5. using SqlSugar;
  6. using System;
  7. using System.Collections.Concurrent;
  8. using System.Collections.Generic;
  9. using System.Data;
  10. using System.Linq;
  11. using System.Text;
  12. using System.Threading.Tasks;
  13. namespace SnapshotWinFormsApp.Repository;
  14. public class ConcurrentQueueRepository<TEntity> where TEntity : Entity, new()
  15. {
  16. private ConcurrentQueue<TEntity> _queue = new ConcurrentQueue<TEntity>();
  17. public void Insert(TEntity entity, ISqlSugarClient sugarClient, bool isEnd)
  18. {
  19. _queue.Enqueue(entity);
  20. if (_queue.Count != 100 && isEnd == false) return;
  21. sugarClient.Ado.BeginTran();
  22. sugarClient.Insertable(_queue.ToList()).ExecuteCommand();
  23. sugarClient.Ado.CommitTran();
  24. _queue.Clear();
  25. }
  26. }
  27. public class TargetRepository<T> : ITargetRepository<T> where T : OldIdEntity, new()
  28. {
  29. private readonly SqlSugarClient _db;
  30. public readonly ConcurrentQueueRepository<T> _queue;
  31. public SqlSugarClient db => this._db;
  32. public TargetRepository(CreateInstanceInDto inDto)
  33. {
  34. var context = inDto.DbSqlServer;
  35. _db = context.DbItems.GetValueOrDefault(inDto.Key + "PGSQLDB");
  36. _queue = new ConcurrentQueueRepository<T>();
  37. }
  38. public async Task<IList<T>> GetAllAsync(CancellationToken token)
  39. {
  40. return await _db.Queryable<T>().ToListAsync(token);
  41. }
  42. public T GetById(int id)
  43. {
  44. return _db.Queryable<T>().InSingle(id);
  45. }
  46. public async Task<string> InsertAsync(T entity, CancellationToken token)
  47. {
  48. if (entity.Id.IsNullOrEmpty())
  49. entity.InitId();
  50. await _db.Insertable(entity).ExecuteCommandAsync(token);
  51. return entity.Id;
  52. }
  53. public ISugarQueryable<T> Queryable()
  54. {
  55. return _db.Queryable<T>();
  56. }
  57. public void Update(T entity)
  58. {
  59. _db.Updateable<T>(entity).ExecuteCommand();
  60. }
  61. public IUpdateable<T> Updateable()
  62. => _db.Updateable<T>();
  63. public async Task<int> ExecuteSqlAsync(string sql)
  64. {
  65. return await _db.Ado.ExecuteCommandAsync(sql, new List<SugarParameter>());
  66. }
  67. public DataTable GetDataTable(string sql)
  68. => _db.Ado.GetDataTable(sql);
  69. public string InsertBulk(T entity, bool isEnd)
  70. {
  71. if (entity.Id.IsNullOrEmpty())
  72. entity.InitId();
  73. _queue.Insert(entity,_db, isEnd);
  74. return entity.Id;
  75. }
  76. public IList<T> GetAll()
  77. {
  78. return _db.Queryable<T>().ToList();
  79. }
  80. }