Fork/Fork.Net/Azylee.Utils/Azylee.DB/DAO/Muse.cs

using System;
using System.Collections.Generic;
using System.Data.Entity;
using System.Data.Entity.Infrastructure;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
using System.Threading.Tasks;

namespace Y.DB.DAO
{
    public class Muse : IDbContext
    {
        DbTable db;
        public Muse()
        {
            db = new DbTable();
        }

        public int Add<T>(T EntityObj, bool isSave = true) where T : class
        {
            try
            {
                this.db.Set<T>().Add(EntityObj);
                if (isSave)
                {
                    return Save();
                }
            }
            catch (Exception e) { }
            return 0;
        }
        public int Adds<T>(IEnumerable<T> EntityObjs) where T : class
        {
            try
            {
                db.Set<T>().AddRange(EntityObjs);
                return Save();
            }
            catch (Exception e) { return 0; }
        }
        public int Del<T>(T EntityObj, bool isSave) where T : class
        {
            try
            {
                this.db.Set<T>().Remove(EntityObj);
                if (isSave)
                {
                    return Save();
                }
            }
            catch (Exception e) { }
            return 0;
        }
        public int Dels<T>(IEnumerable<T> EntityObjs) where T : class
        {
            try
            {
                this.db.Set<T>().RemoveRange(EntityObjs);
                return Save();
            }
            catch (Exception e) { }
            return 0;
        }
        public int Update<T>(T EntityObj, bool isSave) where T : class
        {
            try
            {
                this.db.Entry(EntityObj).State = EntityState.Modified;
                if (isSave)
                {
                    return Save();
                }
            }
            catch (Exception e) { }
            return 0;
        }
        public int Save()
        {
            return db.SaveChanges();
        }

        public T Get<T>(Expression<Func<T, bool>> expression, string[] include) where T : class
        {
            try
            {
                if (include != null && include.Count() > 0)
                {
                    DbQuery<T> query = GetInclude<T>(include);
                    if (query != null)
                        return query.FirstOrDefault(expression);
                }
                return this.db.Set<T>().FirstOrDefault(expression);
            }
            catch (Exception e)
            {
            }
            return null;
        }
        public IEnumerable<T> Gets<T>(Expression<Func<T, bool>> expression, string[] include) where T : class
        {
            try
            {
                if (include != null && include.Count() > 0)
                {
                    DbQuery<T> query = GetInclude<T>(include);
                    if (query != null)
                        return query.Where(expression).ToList();
                }
            }
            catch (Exception)
            {
                throw;
            }
            return db.Set<T>().Where(expression).ToList();
        }
        public IEnumerable<T> GetAll<T>(string[] include, bool track) where T : class
        {
            if (include != null && include.Count() > 0)
            {
                DbQuery<T> query = GetInclude<T>(include);
                if (query != null)
                    if (track)
                        return query.ToList();
                    else
                        return query.AsNoTracking().ToList();
            }
            if (!track)
                db.Set<T>().AsNoTracking().ToList();
            return db.Set<T>().ToList();
        }
        private DbQuery<T> GetInclude<T>(string[] include) where T : class
        {
            DbQuery<T> searchCondition = null;
            foreach (var item in include)
            {
                if (searchCondition == null)
                    searchCondition = this.db.Set<T>().Include(item);
                else
                    searchCondition = searchCondition.Include(item);
            }
            return searchCondition;
        }

        public bool Any<T>(Expression<Func<T, bool>> expression, string[] include) where T : class
        {
            try
            {
                if (include != null && include.Count() > 0)
                {
                    DbQuery<T> query = GetInclude<T>(include);
                    if (query != null)
                        return query.Any(expression);
                }
                return this.db.Set<T>().Any(expression);
            }
            catch (Exception e)
            {
            }
            return false;
        }
        public DbSet<T> Do<T>() where T : class
        {
            return db.Set<T>();
        }
        public IEnumerable<T> ExecuteSqlCom<T, U>(string sql, U paramObjs)
            where U : class
            where T : class
        {
            return db.Set<T>().SqlQuery(sql, paramObjs);
        }
        public void Dispose()
        {
            db.Dispose();
        }
    }
}