OL.FrameCore.Infrastructure/UnitOfWork/EFUnitOfWork.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Text;
using System.Linq;

namespace OL.FrameCore.Infrastructure.UnitOfWork
{
    public class EFUnitOfWork<TContext> : IUnitOfWork<TContext> where TContext : DbContext
    {
        TContext _dbContext;
        Dictionary<Type, object> dictRepository;

        public EFUnitOfWork(TContext dbContext)
        {
            _dbContext = dbContext;
        }

        public TContext DbContext => _dbContext;

        public IRepositoryBase<TEntity> GetRepository<TEntity>() where TEntity : class
        {
            //var repository = new RepositoryBase<TEntity>(_dbContext);
            //return repository;
            if (dictRepository == null)
            {
                dictRepository = new Dictionary<Type, object>();
            }
            Type type = typeof(TEntity);
            if (!this.dictRepository.ContainsKey(type))
            {
                this.dictRepository[type] = new RepositoryBase<TEntity>(this._dbContext);
            }
            return (IRepositoryBase<TEntity>)this.dictRepository[type];
            
        }

        public void Save()
        {
            _dbContext.SaveChanges();
        }

        //public void ExecuteSqlQuery(string sql)
        //{
        //    _dbContext.Database.SqlQuery<UserInfo>("select* from UserInfoes ").ToList();
        //}
        public IList<T> SqlQuery<T>(string sql, params object[] parameters)
            where T : new()
        {
            //注意：不要对GetDbConnection获取到的conn进行using或者调用Dispose，否则DbContext后续不能再进行使用了，会抛异常
            var conn = _dbContext.Database.GetDbConnection();
            try
            {
                conn.Open();
                using (var command = conn.CreateCommand())
                {
                    command.CommandText = sql;
                    command.Parameters.AddRange(parameters);
                    var propts = typeof(T).GetProperties();
                    var rtnList = new List<T>();