Fork/Fork.Net/Y.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 bool 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 false;
        }
        public bool Del<T>(T EntityObj, bool isSave) where T : class
        {
            try
            {
                this.db.Set<T>().Remove(EntityObj);
                if (isSave)
                {
                    return Save();
                }
                return false;
            }
            catch (Exception e) { }
            return false;
        }
        public bool Update<T>(T EntityObj, bool isSave) where T : class
        {
            try
            {
                this.db.Entry(EntityObj).State = EntityState.Modified;
                if (isSave)
                {
                    return Save();
                }
                return false;
            }
            catch (Exception e) { }
            return false;
        }
        public bool Save()
        {
            return db.SaveChanges() > 0;
        }

        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 void Dispose()
        {
            db.Dispose();
        }
        public IEnumerable<T> ExecuteSqlCom<T, U>(string sql, U paramObjs)
            where U : class
            where T : class
        {
            return db.Set<T>().SqlQuery(sql, paramObjs);
        }
    }
}